selenium截取当前窗口,并保存为图片

# -*- coding:utf-8 -*-from selenium import webdriverfrom time import sleepdriver = 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(3)driver.get_screenshot_as_file('c:/users/os/Desktop/baidu.jpg')driver.quit()