加入验证码识别模块，完成基本识别

2023-05-12 08:58:17 +08:00
parent 925faa7345
commit ad3e42968f
2 changed files with 103 additions and 0 deletions
@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# @Time    :2023/5/4 8:01
+# @Author  :lzh
+# @File    : __init__.py.py
+# @Software: PyCharm
@@ -0,0 +1,98 @@
+# -*- coding: utf-8 -*-
+# @Time    :2023/5/4 8:02
+# @File    : usual_code.py
+# @Software: PyCharm
+# 普通四字母验证码
+
+import time
+import cv2 as cv
+import pytesseract
+from PIL import Image
+import requests
+import numpy as np
+
+
+def recognize_text(image_url):
+    # 使用OpenCV解码图像
+    image = cv.imread(image_url)
+    blur = cv.pyrMeanShiftFiltering(image, sp=8, sr=60)
+    # 灰度图像
+    gray = cv.cvtColor(blur, cv.COLOR_BGR2GRAY)
+    # 二值化
+    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU)
+    print(f'二值化自适应阈值：{ret}')
+    # 形态学操作  获取结构元素  开操作
+    kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 2))
+    bin1 = cv.morphologyEx(binary, cv.MORPH_OPEN, kernel)
+    kernel = cv.getStructuringElement(cv.MORPH_OPEN, (2, 3))
+    bin2 = cv.morphologyEx(bin1, cv.MORPH_OPEN, kernel)
+    # 逻辑运算  让背景为白色  字体为黑  便于识别
+    cv.bitwise_not(bin2, bin2)
+    # 识别
+    test_message = Image.fromarray(bin2)
+    text = pytesseract.image_to_string(test_message)
+    print(f'识别结果：{text}')
+
+
+def save_image(image_url,name="code.jpg"):
+    # 图片保存到本地
+    response = requests.get(image_url)
+    with open(name, 'wb') as f:
+        f.write(response.content)
+
+available = {
+    # 中昱维信
+    "https://vip.veesing.com/vip-web/register/vcode": {
+        # 验证码地址
+        "code": "https://vip.veesing.com/vip-web/register/photocode?timestamp=%s",
+        # 发送验证码参数
+        "parameter": {
+            "phone": "18600000000",
+            "code": "1234"
+        },
+        "headers": {
+            "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
+            "referer": "https://vip.veesing.com/",
+            "User-Agent": "Mozilla/5.0 (Linux; Android 10; Redmi K30 Pro) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.72 Mobile Safari/537.36"
+        }
+    },
+    # 逐浪小说
+    "https://www.zhulang.com/home/login/sendCode.html": {
+        "code": "https://www.zhulang.com/login/verify.html?_t=%s",
+        "parameter": {
+            "phone": "18600000000",
+            "code": "1234"
+        },
+        # 验证码参数映射,如果验证码参数名不是code,则需要映射
+        "map_param": {
+            "code": "charcode"
+        },
+        "headers": {
+            "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
+            "referer": "https://www.zhulang.com/login/phone.html",
+            "User-Agent": "Mozilla/5.0 (Linux; Android 10; Redmi K30 Pro) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.72 Mobile Safari/537.36"
+        }
+    }
+}
+
+for index,target in enumerate(available):
+    img_name = "code_%s.jpg" % str(index)
+    save_image(available[target]["code"] % str(time.time() * 1000),img_name)
+    validate_code = recognize_text(img_name)
+    if "map_param" in available[target]:
+        for key in available[target]["map_param"]:
+            available[target]["parameter"][available[target]["map_param"][key]] = validate_code
+            # delte the old key
+            del available[target]["parameter"][key]
+    else:
+        available[target]["parameter"]["code"] = validate_code
+    available[target]["parameter"]["phone"] = "13886686058"
+    # post提供表单数据
+    response = requests.post(target, data=available[target]["parameter"], headers=available[target]["headers"])
+    try:
+        print(response.json())
+    except:
+        print(response.text)
+
+# for url in ["code.jpg","code_0.jpg","code_1.jpg"]:
+#     recognize_text(url)