Selenium：屏幕截图并自定义裁剪大小

一、正常流程，打开浏览器跳转到百度

# 导入 Pillow 的 Image ：pip install Pillow
from PIL import Image
# 导入浏览器驱动
from selenium import webdriver

# 创建一个驱动 
driver = webdriver.Chrome()
# 打开url,跳转到指定页面
driver.get("https://www.baidu.com")

二、截取当前屏幕大图，另存为指定路径的文件

# 对当前屏幕进行截图，并保存为指定路径文件
driver.get_screenshot_as_file(r'D:\screen_big.png')

三、定位元素，获取 Xpath 或 CSS 路径

四、获取定位元素的位置、宽高参数

# 定位需要打印的元素
pic_ele = self.driver.find_element_by_xpath('//*[@id="s_lg_img"]')
# 元素位置、宽高参数获取
left = pic_ele.location.get('x')
top = pic_ele.location.get('y')
right = pic_ele.size.get('width') + left
bottom = pic_ele.size.get('height') + top

五、读取刚刚截取的大图文件，进行裁剪后保存

# 读取图片
img = Image.open(r'D:\screen_big.png')
# 根据元素的 Location和size 图片裁剪
pic_ele = img.crop((left, top, right, bottom))
# 保存裁剪好的文件图片
pic_ele.save(r'D:\screen_small.png')
# 退出浏览器
self.driver.quit()