使用Python和Chrome对网页屏幕截图

2017年，Chrome发布了无头（无GUI）功能，能够从指定的视口截取单个网页的屏幕截图。这有助于保留存档以进行版本比较、监视和面向客户端的可交付成果。因为它是一个无头功能，所以非常适合与Python一起使用。在短短几行中，我们将运行无头Chrome，截取屏幕截图，对其进行优化，然后我们准备对其进行其他操作。它有助于检测更改和寻找性能机会。让我们开始吧！

要求和假设

• 已安装 Python 3 并理解基本的 Python 语法

• 访问 Linux 安装（我推荐 Ubuntu)

• 已安装的浏览器

启动脚本

首先，让我们导入优化图像（命令行）模块，将屏幕截图压缩到优化的 PNG 文件中。

pip3 install optimize-images

接下来，我们导入所需的模块。在核心 Python 之外，我们只需要时间、日期时间和操作系统模块来运行优化图像模块。请记住，它是一个命令行模块，因此我们需要将其作为可执行文件运行。

import os

from datetime import datetime

import time

让我们设置一些我们需要的变量。创建屏幕截图文件名时将使用 name 变量。将 name 变量重新分配给您的网站名称，不要有空格。将 URL 重新分配给要截取屏幕截图的页面的地址。

name = "importsem"

url = "https://importsem.com"

getdate = datetime.now().strftime("%m-%d-%y")

截取屏幕截图

现在我们可以继续使用 os 模块在无头模式下运行 Chrome，隐藏滚动条以获得更好的图像，并设置窗口大小。窗口大小是您必须根据您的布局进行调整的东西。请务必交换屏幕截图目标路径。我们将使用 15 秒的延迟来确保文件已准备就绪，然后再继续。因此，我们可以将原始大小与优化大小进行比较，我们将使用 os.stat（） 函数来获取文件大小。

try:

stream = os.popen("chromium-browser --headless --hide-scrollbars --screenshot='/PATH_TO_DESTINATION/" + name + "_org_" + getdate + ".png' --window-size=1920,1200 " + url)

time.sleep(15)

org_png = os.stat('/PATH_TO_DESTINATION/" + name + "_org_" + getdate + ".png').st_size

优化截图图片

现在我们有了屏幕截图PNG，我们应该对其进行优化以节省一些文件大小。我们将使用优化图像模块。在文档中，您可以使用一些配置来获得更好的压缩。我只是将它们保留为默认值。我通常保存大约 10-15% 的文件大小。再次确保更改代码中的目标路径。我们将使用 20 秒的延迟来确保文件已准备就绪，然后再继续。

stream = os.popen("optimize-images /PATH_TO_DESTINATION/" + name + "_op_" + getdate + ".png")

time.sleep(20)

op_png = os.stat('/PATH_TO_DESTINATION/" + name + "_op_" + getdate + ".png').st_size

现在我们关闭尝试/除非捕获任何错误。

except:

print("Screenshot failed")

比较两个图像

最后，让我们比较文件大小并报告压缩节省的字节数。首先，我们检查图像是否存在于我们期望的位置。然后我们打印出每个图像尺寸并计算差异。

if os.path.isfile() == true and os.path.isfile() == true:

print("Original Image: " + org_png)

print("Optimized Image" + op_png)

print("Saved: " + org_png-op_png)

else:

print("One of the files doesn't exist")