driver.get_screenshot_as_png用法
时间: 2023-10-10 15:08:04 浏览: 164
Python Selenium截图功能实现代码
`driver.get_screenshot_as_png()`是Selenium WebDriver中的一个方法,用于获取当前浏览器窗口的屏幕截图。
使用该方法需要先创建一个WebDriver实例。下面是一个示例:
```python
from selenium import webdriver
# 创建一个Chrome浏览器实例
driver = webdriver.Chrome()
# 访问某个网页
driver.get("http://www.google.com")
# 获取屏幕截图并保存为PNG格式
screenshot = driver.get_screenshot_as_png()
with open('screenshot.png', 'wb') as f:
f.write(screenshot)
# 关闭浏览器
driver.quit()
```
该方法的返回值是一个二进制数据流,可以将其保存为PNG格式的图片文件。如果要使用其他格式,可以使用Pillow等图像处理库进行转换。
阅读全文