PamGram/plugins/jobs/sign.py
2023-03-16 11:25:10 +08:00

28 lines
985 B
Python

import datetime
from typing import TYPE_CHECKING
from core.plugin import Plugin, job
from plugins.genshin.sign import SignSystem
from plugins.tools.sign import SignJobType
from utils.log import logger
if TYPE_CHECKING:
from telegram.ext import ContextTypes
class SignJob(Plugin):
def __init__(self, sign_system: SignSystem):
self.sign_system = sign_system
@job.run_daily(time=datetime.time(hour=0, minute=1, second=0), name="SignJob")
async def sign(self, context: "ContextTypes.DEFAULT_TYPE"):
logger.info("正在执行自动签到")
await self.sign_system.do_sign_job(context, job_type=SignJobType.START)
logger.success("执行自动签到完成")
await self.re_sign(context)
async def re_sign(self, context: "ContextTypes.DEFAULT_TYPE"):
logger.info("正在执行自动重签")
await self.sign_system.do_sign_job(context, job_type=SignJobType.REDO)
logger.success("执行自动重签完成")