Python+selenium之窗口截图

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

 1 from selenium import webdriver
 2 from time import sleep
 3 driver = webdriver.Firefox()
 4 driver.get("http://www.baidu.com")
 5 
 6 driver.find_element_by_id('kw').send_keys('selenium')
 7 driver.find_element_by_id('su').click()
 8 sleep(2)
 9 
10 #截取当前窗口,指定截图图片的保存位置
11 driver.get_screenshot_as_file(r"C:\Users\fyr\Desktop\123.jpg")
12 driver.quit()

 

posted @ 2017-09-12 17:43  木棉花的漂泊  阅读(775)  评论(0编辑  收藏  举报