1、截屏,截取整个页面
get_screenshot_as_file(img_path)
2、截屏,截取单个元素
screenshot(img_path)
3、案例
import time
from selenium import webdriver
# 创建浏览器驱动对象
driver = webdriver.Chrome(r"D:\chromedriver\chromedriver_win32-V103.0.5060.53\chromedriver.exe")
# 访问网址
driver.get("http://www.baidu.com")
# 截屏,截取整个页面
time.sleep(3)
driver.get_screenshot_as_file("./all.png")
# 截屏,截取单个元素
time.sleep(3)
driver.find_element_by_id("kw").screenshot("./ele.png")