miui-auto-tasks/miuitask.py

'''
Date: 2023-11-13 20:29:19
LastEditors: Night-stars-1 nujj1042633805@gmail.com
LastEditTime: 2023-12-03 01:53:48
'''
# new Env("MIUI-Auto-Task") # pylint: disable=missing-module-docstring
# cron 30 8 * * * miuitask.py

import asyncio

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:
        login_obj = Login(account)
        if cookies := await login_obj.login():
            sign_obj = BaseSign(cookies)
            daily_tasks = await sign_obj.check_daily_tasks()
            sign_task_obj = sign_obj.AVAILABLE_SIGNS  # 签到任务对象合集
            for task in daily_tasks:
                if not task.showType:
                    log.info(f"开始执行{task.name}任务")
                    if task_obj := sign_task_obj.get(task.name):  # 签到任务对象
                        if getattr(account, task_obj.__name__):
                            token = await get_token(cookies["cUserId"]) if task_obj == CheckIn else None
                            await task_obj(cookies, token).sign()
                        else:
                            log.info(f"任务{task.name}被禁用")
                    else:
                        log.error(f"未找到{task.name}任务")
                else:
                    log.info(f"{task.name}任务已完成")
    notify_me(InterceptHandler.message)


if __name__ == "__main__":
    asyncio.run(main())