mirror of
https://github.com/0-8-4/miui-auto-tasks.git
synced 2024-11-16 04:35:51 +00:00
762273a66b
* fix: 修复无法构建docker镜像的问题 * Update Dockerfile * chore: 更新日志收集方法 * chore: 遵守代码规范 * chore: update issue templates * chore: 删除测试代码 * chore: 更改日志输出样式 * chore: docker生成配置时随机生成自动运行时间 * chore: trailing-whitespace * chore: 日志添加换行
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
'''
|
|
Date: 2023-11-13 20:29:19
|
|
LastEditors: Night-stars-1 nujj1042633805@gmail.com
|
|
LastEditTime: 2023-11-24 16:23:43
|
|
'''
|
|
# new Env("MIUI-Auto-Task") # pylint: disable=missing-module-docstring
|
|
# cron 30 8 * * * miuitask.py
|
|
|
|
import asyncio
|
|
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
|
|
from utils.api.login import Login
|
|
from utils.api.sign import BaseSign
|
|
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():
|
|
token = await get_token(cookies["cUserId"])
|
|
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__):
|
|
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__":
|
|
HOUR = _conf.preference.hour
|
|
MINUTE = _conf.preference.minute
|
|
if HOUR and MINUTE:
|
|
log.info(f"脚本将在每日{HOUR}:{MINUTE}自动运行")
|
|
# 创建一个新的事件循环
|
|
loop = asyncio.get_event_loop()
|
|
scheduler = AsyncIOScheduler()
|
|
scheduler.add_job(main, 'cron', hour=HOUR, minute=MINUTE, id='miuitask')
|
|
scheduler.start()
|
|
try:
|
|
loop.run_forever()
|
|
except (KeyboardInterrupt, SystemExit):
|
|
pass
|
|
finally:
|
|
scheduler.shutdown()
|
|
else:
|
|
asyncio.run(main())
|