Selenium+chrome 全屏、区域截图

1. 全屏截图

# -*- coding:utf-8 -*-
from selenium import webdriver
from time import sleep
import time
 
driver = webdriver.Chrome()
driver.get('http://www.baidu.com/')
driver.get_screenshot_as_file("截图\\screen.png")
 
sleep(2)
driver.quit()

2. 区域截图

# -*- coding:utf-8 -*-
from selenium import webdriver
from time import sleep
import time
from PIL import Image

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

# 要截屏的目标元素
element = driver.find_element_by_id("su")

# 获取element的顶点坐标
xPiont = element.location['x']
yPiont = element.location['y']
# 获取element的宽、高
element_width = xPiont + element.size['width']
element_height = yPiont + element.size['height']
 
picture = Image.open('截图\\element.png')
picture = picture.crop((xPiont, yPiont, element_width, element_height))
picture.save("截图\\element.png")
 
sleep(2)
driver.quit()