xiaobei_selenium_automation/common/image_identify.py

43 lines
1.4 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/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
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)
left = ele.location['x'] + 220
top = ele.location['y'] + 90
right = ele.size['width'] + left + 85
height = ele.size['height'] + top + 15
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)
return res