selenium网页自动截图

网页截图核心：

driver.get_screenshot_as_file(r'E:\UnittestProject\selenium\百度.png')

案例：

打开百度首页，停留2秒，截图并保存。

代码：

from selenium import webdriver
from time import sleep

driver=webdriver.Firefox()
driver.get("http://www.baidu.com")
sleep(2)
#网页截图并保存
driver.get_screenshot_as_file(r'E:\UnittestProject\selenium\百度.png')
print("over!")

注意：文件路径必须转义，这里在文件路径前面加了r

结果：
控制台正常打印内容，且e盘有相关图片