selenium python截图函数

webdriver的get_screenshot_as_file()函数截图当前页面并保存到指定的路径下面

#coding=utf-8

from selenium import webdriver
driver=webdriver.Firefox()
driver.get('http://www.baidu.com')
try:
    driver.get_element_by_id('kwss').send_keys('selenium')
    driver.get_element_by_id('su').click()
except:
    driver.get_screenshot_as_file('C:\\Program Files\\Python36\\Scripts\\baiduerror.png')   #路径可以用绝对路径，且路径中不能包含没有的文件夹名（需要是已存在的路径）

get_screenshot_as_file()函数截图当前页面并保存到指定的路径下面