python自动化测试——截图（截取页面特定元素）

from PIL import Image
from selenium import webdriver
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.get("https://www.***.cn")

filename = "image.png"
#截屏
driver.save_screenshot(filename)

#需要截取的元素定位
element = driver.find_element(By.ID, "***")
#获取元素上下左右的位置
left = element.location[‘x‘]
top = element.location[‘y‘]
right = element.location[‘x‘] + element.size[‘width‘]
bottom = element.location[‘y‘] + element.size[‘height‘]

#打开刚才的截图
im = Image.open(filename)
#截取对应位置
im = im.crop((left, top, right, bottom))
#保存覆盖原有截图
im.save(filename)

driver.quit()