PamGram/plugins/jobs/sign.py

import datetime
from typing import TYPE_CHECKING

from core.plugin import Plugin, job
from plugins.starrail.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("执行自动重签完成")