使用python库解决登录的验证码识别-图片验证码

前言：

在UI自动化测试和爬虫测试中，验证码是个比较头疼的问题，包括：图片验证码，滑块验证码，等一些常见的验证码场景。

识别验证码的python 库有很多，用起来也并不简单，这里推荐一个简单实用的识别验证码的库 ddddocr (带带弟弟ocr)库。

今天先用一个图片验证码示例来演示下：

准备：

1.安装库ddddocr

2.安装库pillow

3.以一个带验证码登录的网站为示例：https://xqctk.jtys.sz.gov.cn/

开始编码：

1代码如下：

import time
from ddddocr import DdddOcr
from selenium import webdriver
from selenium.webdriver.common.by import By
from PIL import Image  # 需安装 pillow

path = 'https://xqctk.jtys.sz.gov.cn/'


def get_picture_code(locator):
    pic_save_path = 'D://html.png'
    cron_path = 'D://code.png'
    browser.save_screenshot(pic_save_path)  # 保存为图片
    location = browser.find_element(By.XPATH, locator)  # 获取需要识别的元素对象
    size = location.size  # 获取需要识别的元素尺寸
    position = (int(location.location['x']), int(location.location['y']), int(location.location['x'] + size['width']), int(location.location['y'] + size['height']))
    Image.open(pic_save_path).crop(box=position).save(cron_path)  # crop进行裁剪
    with open(cron_path, 'rb') as f:
        img_bytes = f.read()
        res = DdddOcr().classification(img_bytes)
    return res


browser = webdriver.Chrome()  # 打开浏览器
browser.maximize_window()  # 浏览器最大化
browser.get(path)  # 打开网页
time.sleep(5)
pic_code = get_picture_code("//td[@id='incrementGetValidCodeImg']//img")
browser.find_element(By.XPATH, "//input[@id='incrementValidCode']").send_keys(pic_code)
time.sleep(5)
browser.quit()

运行结果：