selenium自动化测试浏览器截屏操作

from selenium  import webdriver
from time import sleep

driver=webdriver.Firefox()

driver.get("http://baidu.com")
#将浏览器的截图保存在下个地址中，并取名
driver.get_screenshot_as_file(r"C:\Users\Administrator\Desktop\临时文件\020301.png")
sleep(3)

driver.quit()