mirror of
https://github.com/0-8-4/miui-auto-tasks.git
synced 2024-11-16 04:35:51 +00:00
60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
'''
|
|
Date: 2023-11-13 20:29:19
|
|
LastEditors: Night-stars-1 nujj1042633805@gmail.com
|
|
LastEditTime: 2023-12-18 19:21:36
|
|
'''
|
|
# new Env("MIUI-Auto-Task") # pylint: disable=missing-module-docstring
|
|
# cron 30 8 * * * miuitask.py
|
|
|
|
import asyncio
|
|
|
|
from tenacity import RetryError, Retrying, stop_after_attempt
|
|
|
|
from utils.api.login import Login
|
|
from utils.api.sign import BaseSign, CheckIn
|
|
from utils.config import ConfigManager
|
|
from utils.logger import InterceptHandler, log
|
|
from utils.request import notify_me
|
|
from utils.system_info import print_info
|
|
from utils.utils import get_token
|
|
|
|
_conf = ConfigManager.data_obj
|
|
|
|
|
|
async def main():
|
|
"""启动签到"""
|
|
print_info()
|
|
for account in _conf.accounts:
|
|
try:
|
|
for attempt in Retrying(stop=stop_after_attempt(2)):
|
|
with attempt:
|
|
login_obj = Login(account)
|
|
if cookies := await login_obj.login():
|
|
sign_obj = BaseSign(cookies, account.user_agent)
|
|
daily_tasks = await sign_obj.check_daily_tasks()
|
|
sign_task_obj = sign_obj.AVAILABLE_SIGNS # 签到任务对象合集
|
|
for task in daily_tasks:
|
|
log.info(f"开始执行{task.name}任务")
|
|
if task.showType:
|
|
log.info(f"{task.name}任务已完成")
|
|
continue
|
|
if not (task_obj := sign_task_obj.get(task.name)): # 签到任务对象
|
|
log.error(f"未找到{task.name}任务")
|
|
continue
|
|
if not getattr(account, task_obj.__name__):
|
|
log.info(f"任务{task.name}被禁用")
|
|
continue
|
|
token = await get_token(cookies["cUserId"]) if task_obj == CheckIn else None
|
|
status, reason = await task_obj(cookies, account.user_agent, token).sign()
|
|
if not status and reason == "cookie":
|
|
raise ValueError("Cookie失效")
|
|
user_info = await sign_obj.user_info()
|
|
log.info(f"{user_info.title} 成长值: {user_info.point}")
|
|
except RetryError:
|
|
...
|
|
notify_me(InterceptHandler.message)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|