xiaobei_selenium_automation/common/image_identify.py

46 lines
1.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
# ==================================================
# @Time : 2023-06-23 13:55:01
# @Author : 小北
# @微信xiaobei_upup
# @Email : 2211484376@qq.com
# @QQ群585971269
# @微信群:可加我微信拉你
# @Site : 苏州
# @Desc : 免费验证码识别
# @跳槽辅导:初中高级测试跳槽涨薪就业辅导,详情咨询微信
# @求职辅导:初中高级测试求职就业速成辅导,详情咨询微信
# @特色: 小北独创VIP面试速成课程拿下心仪的offer
# @如何付款:小程序付款 or 支付宝 or 微信
# ==================================================
"""
from PIL import Image
import ddddocr
2023-11-02 16:07:54 +08:00
def image_identify(driver, ele, whole_name, crop_name):
"""
:param driver: 浏览器驱动
:param ele: 验证码的元素
:param whole_name: 整个页面的截图名字
:param crop_name: 页面中对于验证码的抠图名字
:return: 返回验证码识别出来的字符串
"""
driver.save_screenshot(whole_name)
2023-11-02 16:07:54 +08:00
left = ele.location['x']
top = ele.location['y']
right = ele.size['width'] + left
height = ele.size['height'] + top
img = Image.open(whole_name).crop((left, top, right, height))
img.save(crop_name)
ocr = ddddocr.DdddOcr()
with open(crop_name, 'rb') as f:
image = f.read()
res = ocr.classification(image)
print(res)
2023-11-02 16:07:54 +08:00
return res