selenium之截图功能

使用selenium进行验证码图片的截图：

1. 将整个页面截取一个大图；

2. 在页面上通过xpath/cssselect定位到验证码图片；

3. 获取验证码图片在整个大图中的坐标；

4. 根据坐标从大图中截取小图；

下面以截取云打码的验证码为例,完整代码如下:

# PIL属于Pillow模块下的一个模块，主要是用来处理图片。
from PIL import Image
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
import requests


driver = webdriver.Firefox()

driver.get('http://www.yundama.com/')
# 找到验证码
image = WebDriverWait(driver, 10).until(lambda driver: driver.find_element_by_id('verifyImg'))

# 截取大图
# driver.save_screenshot()保存屏幕截图
driver.save_screenshot('page.png')

# 确定图片距离页面上部和左侧的大小
left = image.location['x']
top = image.location['y']
# 确定图片距离页面底部和右侧的大小
right = left + image.size['width']
bottom = top + image.size['height']

# 在大图上截取小图
# 打开大图
img = Image.open('page.png')
# 截取小图
img = img.crop((left, top, right, bottom))
# 保存小图
img.save('captcha.png')


# 单独发送请求获取验证码
res = requests.get('http://www.yundama.com/index/captcha').content
with open('my.png', 'wb') as f:
    f.write(res)

selenium之截图功能

猜你喜欢