迹忆客 专注技术分享

我们可以使用 Selenium 在自动化 Web 浏览器上使用 Python 脚本执行任务。 我们可以使用不同的方法将浏览器的屏幕捕获为屏幕截图。

本篇文章将介绍如何在 Python 中使用 selenium 捕获自动浏览器的屏幕截图。

使用 save_screenshot() 函数在 Python 中使用 selenium 捕获屏幕截图

selenium 包提供了 save_screenshot() 函数来截取在自动浏览器中打开的网页的屏幕截图。

我们可以在函数中指定屏幕截图的文件名（PNG 格式）和路径。 该函数与 webdriver 对象一起使用。

请参阅下面的代码。

from selenium import webdriver
driver = webdriver.Chrome(r'C:/driverpath/chromedriver.exe')
driver.get("https://www.delftstack.com/")
driver.save_screenshot("filename.png")

在上面的示例中，我们创建一个 webdriver 对象并使用 get() 函数创建到所需 URL 的连接。 然后使用 save_screenshot() 函数捕获屏幕截图并保存。

使用 get_screenshot_as_file() 函数在 Python 中使用 selenium 捕获屏幕截图

get_screenshot_as_file() 函数还会创建一个包含 webdriver 对象屏幕截图的文件。 我们可以通过在函数中提及它来将文件保存在所需的路径中。

如果文件创建成功，则返回 True。 如果有任何异常，该函数将返回 False。

例如：

from selenium import webdriver
driver = webdriver.Chrome(r'C:/driverpath/chromedriver.exe')
driver.get("https://www.delftstack.com/")
driver.get_screenshot_as_file("filename1.png")

使用 Screenshot-Selenium 包在 Python 中使用 selenium 捕获屏幕截图

Selenium-Screenshot 包可用于捕获 webdriver 的 HTML 元素的屏幕截图。

该软件包与 Python 3 兼容，可以使用 pip install Selenium-Screenshot 进行安装。 我们还可以使用它来单独捕获 HTML 元素的剪辑。

我们可以使用 full_screenshot() 函数来捕获整个浏览器窗口。 我们需要首先创建一个 Screenshot 类的对象才能使用它。

webdriver 对象和文件名在 full_screenshot() 函数中指定。

请参阅下面的代码。

from selenium import webdriver
from Screenshot import Screenshot_Clipping
s = Screenshot_Clipping.Screenshot()
driver = webdriver.Chrome(r'C:/driverpath/chromedriver.exe')
driver.get("https://www.delftstack.com/")
s.full_Screenshot(driver, save_path=r'.' , image_name='name.png')

我们在上面的示例中使用 full_screenshot() 函数来捕获屏幕截图。 我们使用 image_name 参数和 save_path 参数提供文件的名称和路径。

总结

我们讨论了如何在 Python 中使用 selenium 截取屏幕截图。 我们可以使用 save_screenshot() 和 get_screenshot_as_file() 函数将屏幕截图保存为 PNG 文件。

或者，我们也可以安装一个名为 Selenium-Screenshot 的子包，并使用它来捕获浏览器上的网页。