getcode.py

import requests
import base64
from verifycode import recognize_verifycode
from config import VERIFYCODE_URL, HEADERS

def getcode():
    """
    获取图片验证码

    Returns:
        tuple: (验证码文本, 验证码ID) 或 None（失败时）
    """
    json_data = {}

    response = requests.post(VERIFYCODE_URL, headers=HEADERS, json=json_data, verify=False)
    # 提取响应文本中的msg和verificationCodeBase64字段

    response_data = response.json()
    msg = response_data.get('msg')
    data = response_data.get('data', {})
    verificationCodeBase64 = data.get('verificationCodeBase64', '')
    verificationCodeId = data.get('verificationCodeId', '')


    # 检查是否有验证码数据
    if not verificationCodeBase64:
        print("警告：未获取到验证码图片数据（verificationCodeBase64为空）")
        return None

    # 将verificationCodeBase64进行base64解码
    verificationCode = base64.b64decode(verificationCodeBase64)


    # 将解码后的image信息传入验证码识别函数
    code = recognize_verifycode(verificationCode)
    print(f"识别出的验证码: {code}")
    return code,verificationCodeId