3. selenium_pytesseract 识别验证码

前言：

很多网站的注册，登录都会有验证码，有安全作用的一方面，也避免了一些持续访问攻击对服务产生性能问题。

自动化测试中对验证码处理的办法一般有三种：

1. 让开发写一个固定的验证码
2. 自动化测试的时候让开发去掉验证码
3. 自己想办法识别验证码

接下来使用python 中的pytesseract 模块和 PIL 模块解决一些不太复杂的验证码。
PIL（Python Imaging Library）是Python常用的图像处理库。
pytesseract 识别验证码。

思路步骤：

1. 获取验证码页面
2. 确定验证码坐标和大小
3. 通过坐标和尺寸截取验证码图片，PIL 模块处理图片
4. 通过pytesseract 识别验证码

安装PIL 和 pytesseract

pip install pytesseract -i https://pypi.douban.com/simple
pip install pillow -i https://pypi.douban.com/simple

Pillow 是 pil 的一个分支。

以下开始截取验证码图片。

找到验证码图片 id

截取验证码

import time

from selenium import webdriver
from time import sleep
from PIL import Image

driver = webdriver.Chrome()
driver.get("http://120.25.209.187:12306/jpress/user/register")
sleep(2)

driver.maximize_window()               # 页面最大化

t = time.time()                        # 时间戳
picture_name1 = str(t)+'.png'          # 图片名称
driver.save_screenshot(picture_name1)  # 截取登录页面图片

# 获取图片验证码坐标和尺寸
ce = driver.find_element_by_id("captchaimg")
print(ce.location)
left = ce.location['x']
top = ce.location['y']
right = ce.size['width']+left
height = ce.size['height']+top

sleep(2)
# 截取验证码图片
im = Image.open(picture_name1)
img = im.crop((left, top, right, height))

t = time.time()
picture_name2 = str(t)+'.png'              # 验证码图片命名为 picture_name2
img.save(picture_name2)

driver.close()