PamGram/plugins/help.py

from telegram import Update
from telegram.constants import ChatAction
from telegram.error import BadRequest
from telegram.ext import CommandHandler, CallbackContext

from app.template.service import TemplateService
from config import config
from logger import Log
from utils.app.inject import inject
from utils.decorators.error import error_callable
from utils.decorators.restricts import restricts
from utils.plugins.manager import listener_plugins_class


@listener_plugins_class()
class Help:
    """
    帮助
    """

    def __init__(self):
        self.help_png = None
        self.file_id = None

    @classmethod
    def create_handlers(cls) -> list:
        _help = cls()
        return [
            CommandHandler("help", _help.command_start, block=False),
        ]

    @error_callable
    @restricts()
    @inject
    async def command_start(self, update: Update, _: CallbackContext, template_service: TemplateService) -> None:
        message = update.message
        user = update.effective_user
        Log.info(f"用户 {user.full_name}[{user.id}] 发出help命令")
        if self.file_id is None or config.DEBUG:
            await message.reply_chat_action(ChatAction.TYPING)
            help_png = await template_service.render('bot/help', "help.html", {}, {"width": 768, "height": 768})
            await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
            reply_photo = await message.reply_photo(help_png, filename="help.png", allow_sending_without_reply=True)
            photo = reply_photo.photo[0]
            self.file_id = photo.file_id
        else:
            try:
                await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
                await message.reply_photo(self.file_id, allow_sending_without_reply=True)
            except BadRequest as error:
                self.file_id = None
                Log.error("发送图片失败，尝试清空已经保存的file_id，错误信息为", error)
                await message.reply_text("发送图片失败", allow_sending_without_reply=True)