miui-auto-tasks/utils/captcha.py

"""
Date: 2023-11-13 19:55:22
LastEditors: Night-stars-1 nujj1042633805@gmail.com
LastEditTime: 2023-11-18 14:30:41
"""

from .request import post
from .logger import log
from .config import ConfigManager
from .data_model import ApiResultHandler, GeetestResult

_conf = ConfigManager.data_obj


async def get_validate(gt: str, challenge: str) -> GeetestResult:  # pylint: disable=invalid-name
    """获取人机验证结果"""
    try:
        validate = None
        params = _conf.preference.geetest_params.copy()
        for key, value in params.items():
            if isinstance(value, str):
                params[key] = value.format(gt=gt, challenge=challenge)
        data = _conf.preference.geetest_data.copy()
        for key, value in data.items():
            if isinstance(value, str):
                data[key] = value.format(gt=gt, challenge=challenge)
        response = await post(
            _conf.preference.geetest_url,
            params=params,
            json=data,
        )
        log.debug(response.text)
        geetest_data = response.json()
        geetest = ApiResultHandler(geetest_data)
        challenge = geetest.data["challenge"]
        validate = geetest.data["validate"]
        return GeetestResult(challenge=challenge, validate=validate)
    except Exception:  # pylint: disable=broad-exception-caught
        log.exception("获取人机验证结果异常")
        return GeetestResult(challenge="", validate="")