selenium之截取当前窗口

webdriver 提供了 get_screenshot_as_file(filename)函数来截取当前窗口

截取单个元素使用screenshot(filename)

下面代码以百度为例演示

from selenium import webdriver
import time
driver = webdriver.Chrome()
driver.get("http://www.baidu.com")
driver.maximize_window()  #最大化窗口
driver.get_screenshot_as_file("./picture.png")    #截取整个当前窗口，里面参数是保存图片的路径
time.sleep(2)
input_text=driver.find_element_by_id("su")  #定位到“百度一下”赋值给变量
input_text.screenshot("./su.png")    #截取单个元素的窗口，也就是只截取到百度一下元素图片
driver.quit()