窗口截图.py

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("C:\我的代码\selenium自动化测试\selenium2自动化测试实战基于python语言\窗口截图\baidu_img.png")