selenium - webdriver - 截圖方法get_screenshot_as_file()

WebDriver提供了截圖函數get_screenshot_as_file()來截取當前窗口。

from selenium import webdriver
from time import sleep

driver = webdriver.Chrome()
driver.get("http://www.baidu.com")

driver.find_element_by_id('kw').send_keys('selenium')
driver.find_element_by_id('su').click()
sleep(5)

# 截取當前窗口，並指定截圖圖片的保存位置
driver.get_screenshot_as_file("D:\selenium_img.png")
sleep(5)

driver.quit()

 

參考資料：http://www.testclass.net/selenium_python/get-screenshot/