利用selenium自动实现对验证码截图并保存

一、简介

我们日常在做自动化测试时，需要对网页进行截图保存来进行评估；本文讲的是对验证码进行截图，其实就是对某一个元素进行定位截图，因不好用某个网站的验证码进行演示，因此这里模仿对百度首页并对“百度一下”进行截图，然后保存在本地。

二、具体用法

其实验证码截图与“百度一下”截图原理都是一样，就是定位某个图片元素进行截图。
代码如下：

import time
from io import BytesIO
from PIL import Image
from selenium import webdriver


driver = webdriver.Chrome()
driver.maximize_window()
driver.get('https://baidu.com')
time.sleep(2)
# 执行js，对页面进行缩放，100 占 125 的 80%
# driver.execute_script('document.body.style.zoom="0.8"')
# 要截图的元素
element = driver.find_element_by_css_selector('#su')
# 坐标
x, y = element.location.values()
# 宽高
h, w = element.size.values()
# 把截图以二进制形式的数据返回
image_data = driver.get_screenshot_as_png()
# 以新图片打开返回的数据
screenshot = Image.open(BytesIO(image_data))
# 对截图进行裁剪
result = screenshot.crop((x, y, x + w, y + h))
# 显示图片
# result.show()
result.save('001.png')

#运行结果