mirror of
https://github.com/0-8-4/miui-auto-tasks.git
synced 2024-12-04 20:27:24 +00:00
f1d4605af7
* fix: 修复无法构建docker镜像的问题 * Update Dockerfile * chore: 更新日志收集方法 * chore: 遵守代码规范 * chore: update issue templates * chore: 删除测试代码 * chore: 更改日志输出样式 * chore: docker生成配置时随机生成自动运行时间 * chore: trailing-whitespace * chore: 日志添加换行 * chore: 未配置验证码解决方案时直接跳过 * chore: 使用遍历方法找到键值，提升泛用性 * chore: 使用cron执行自动任务 * chore: snake_case * chore: 修改运行时间 * Update config.py
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
'''
|
|
Date: 2023-11-13 19:55:22
|
|
LastEditors: Night-stars-1 nujj1042633805@gmail.com
|
|
LastEditTime: 2023-11-25 15:59:40
|
|
'''
|
|
|
|
from .request import post
|
|
from .logger import log
|
|
from .config import ConfigManager
|
|
from .data_model import ApiResultHandler, GeetestResult
|
|
|
|
_conf = ConfigManager.data_obj
|
|
|
|
def find_key(data: dict, key: str):
|
|
"""递归查找字典中的key"""
|
|
for dkey, dvalue in data.items():
|
|
if dkey == key:
|
|
return dvalue
|
|
if isinstance(dvalue, dict):
|
|
find_key(dvalue, key)
|
|
return None
|
|
|
|
async def get_validate(gt: str, challenge: str) -> GeetestResult: # pylint: disable=invalid-name
|
|
"""获取人机验证结果"""
|
|
try:
|
|
validate = None
|
|
if _conf.preference.geetest_url:
|
|
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 = find_key(geetest.data, "challenge")
|
|
validate = find_key(geetest.data, "validate")
|
|
return GeetestResult(challenge=challenge, validate=validate)
|
|
else:
|
|
return GeetestResult(challenge="", validate="")
|
|
except Exception: # pylint: disable=broad-exception-caught
|
|
log.exception("获取人机验证结果异常")
|
|
return GeetestResult(challenge="", validate="")
|