自动化用例是由程序去执行的，因此有时候打印的错误信息并不十分明确。如果在脚本执行出错的时候能对当前窗口截图保存，那么通过图片就可以非常直观地看出出错的原因。WebDriver提供了截图函数get_screenshot_as_file()来截取当前窗口。 ```python from selenium import webdriver from time import sleep

driver = webdriver.Firefox() driver.get(‘http://www.baidu.com’)

driver.find_element_by_id(‘kw’).send_keys(‘selenium’) driver.find_element_by_id(‘su’).click() sleep(2)

截取当前窗口，并指定截图图片的保存位置

driver.get_screenshot_as_file(“D:\baidu_img.jpg”)

driver.quit()

脚本运行完成后打开D盘，就可以找到baidu_img.jpg图片文件了。




原始封面

![课程图片](https://images.unsplash.com/photo-1513836279014-a89f7a76ae86?w=300)