PaiGram/plugins/genshin/material.py

from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.constants import ChatAction, ParseMode
from telegram.ext import filters, ConversationHandler, CommandHandler, MessageHandler, CallbackContext

from core.game.services import GameMaterialService
from logger import Log
from plugins.base import BasePlugins
from utils.bot import get_all_args
from utils.decorators.error import error_callable
from utils.decorators.restricts import restricts
from utils.helpers import url_to_file
from utils.plugins.manager import listener_plugins_class
from utils.service.inject import inject


@listener_plugins_class()
class Material(BasePlugins):
    """角色培养素材查询"""

    KEYBOARD = [[InlineKeyboardButton(
        text="查看角色培养素材列表并查询",
        switch_inline_query_current_chat="查看角色培养素材列表并查询")]]

    @inject
    def __init__(self, game_material_service: GameMaterialService = None):
        self.game_material_service = game_material_service

    @classmethod
    def create_handlers(cls) -> list:
        material = cls()
        return [
            CommandHandler("material", material.command_start, block=False),
            MessageHandler(filters.Regex("^角色培养素材查询(.*)"), material.command_start, block=False),
        ]

    @error_callable
    @restricts(return_data=ConversationHandler.END)
    async def command_start(self, update: Update, context: CallbackContext) -> None:
        message = update.message
        user = update.effective_user
        args = get_all_args(context)
        if len(args) >= 1:
            character_name = args[0]
        else:
            reply_message = await message.reply_text("请回复你要查询的培养素材的角色名",
                                                     reply_markup=InlineKeyboardMarkup(self.KEYBOARD))
            if filters.ChatType.GROUPS.filter(reply_message):
                self._add_delete_message_job(context, message.chat_id, message.message_id)
                self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
            return
        url = await self.game_material_service.get_material(character_name)
        if not url:
            reply_message = await message.reply_text(f"没有找到 {character_name} 的培养素材",
                                                     reply_markup=InlineKeyboardMarkup(self.KEYBOARD))
            if filters.ChatType.GROUPS.filter(reply_message):
                self._add_delete_message_job(context, message.chat_id, message.message_id)
                self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
            return
        Log.info(f"用户 {user.full_name}[{user.id}] 查询角色培养素材命令请求 || 参数 {character_name}")
        await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
        file_path = await url_to_file(url, "")
        caption = "From 米游社 友人A " \
                  f"查看 [原图]({url})"
        await message.reply_photo(photo=open(file_path, "rb"), caption=caption, filename=f"{character_name}.png",
                                  allow_sending_without_reply=True, parse_mode=ParseMode.MARKDOWN_V2)