十四 selenium页面截图和元素截图

截图

• 页面整个截图
  driver.get_screenshot_as_file("test.png")
  driver.save_screenshot("tests.png")
• 元素截图
  先定位到元素
  ele.screenshot("xxx.png")

from selenium import webdriver

driver = webdriver.Chrome()

driver.get("https://www.baidu.com")
driver.maximize_window()

# 截取整个页面
driver.get_screenshot_as_file("test.png")
driver.save_screenshot("tests.png")

# 找到搜索框
inputElement = driver.find_element_by_id("kw")

# 截取搜索框元素
inputElement.screenshot("inputElement.png")

知识点
get_screenshot_as_file()、save_screenshot() 效果一样， save_screenshot()最终调用的函数就是前者
后缀名建议写 .png ，否则会有Warning提示