mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-25 09:37:30 +00:00
102 lines
4.7 KiB
Python
102 lines
4.7 KiB
Python
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.constants import ChatAction
|
|
from telegram.ext import filters, CommandHandler, MessageHandler
|
|
|
|
from logger import Log
|
|
from manager import listener_plugins_class
|
|
from metadata.shortname import weaponToName
|
|
from model.helpers import url_to_file
|
|
from plugins.base import BasePlugins, restricts
|
|
from plugins.errorhandler import conversation_error_handler
|
|
from service import BaseService
|
|
from utils.base import PaimonContext
|
|
|
|
|
|
@listener_plugins_class(need_service=True)
|
|
class Weapon(BasePlugins):
|
|
"""
|
|
武器查询
|
|
"""
|
|
|
|
KEYBOARD = [[InlineKeyboardButton(text="查看武器列表并查询", switch_inline_query_current_chat="查看武器列表并查询")]]
|
|
|
|
def __init__(self, service: BaseService):
|
|
self.service = service
|
|
|
|
@classmethod
|
|
def create_handlers(cls, service: BaseService) -> list:
|
|
weapon = cls(service)
|
|
return [
|
|
CommandHandler("weapon", weapon.command_start, block=False),
|
|
MessageHandler(filters.Regex("^武器查询(.*)"), weapon.command_start, block=False)
|
|
]
|
|
|
|
@conversation_error_handler
|
|
@restricts()
|
|
async def command_start(self, update: Update, context: PaimonContext) -> None:
|
|
message = update.message
|
|
user = update.effective_user
|
|
args = context.args
|
|
match = context.match
|
|
weapon_name: str = ""
|
|
if args is None:
|
|
if match is not None:
|
|
match_data = match.group(1)
|
|
if match_data != "":
|
|
weapon_name = match_data
|
|
else:
|
|
if len(args) >= 1:
|
|
weapon_name = args[0]
|
|
if weapon_name == "":
|
|
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
|
|
weapon_name = weaponToName(weapon_name)
|
|
weapons_list = await self.service.wiki.get_weapons_list()
|
|
for weapon in weapons_list:
|
|
if weapon["name"] == weapon_name:
|
|
weapon_data = weapon
|
|
break
|
|
else:
|
|
reply_message = await message.reply_text(f"没有找到 {weapon_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}] 查询武器命令请求 || 参数 {weapon_name}")
|
|
await message.reply_chat_action(ChatAction.TYPING)
|
|
|
|
async def input_template_data(_weapon_data):
|
|
_template_data = {
|
|
"weapon_name": _weapon_data["name"],
|
|
"weapon_info_type_img": await url_to_file(_weapon_data["type"]["icon"]),
|
|
"progression_secondary_stat_value": _weapon_data["secondary"]["max"],
|
|
"progression_secondary_stat_name": _weapon_data["secondary"]["name"],
|
|
"weapon_info_source_img": await url_to_file(_weapon_data["source_img"]),
|
|
"progression_base_atk": _weapon_data["atk"]["max"],
|
|
"weapon_info_source_list": [],
|
|
"special_ability_name": _weapon_data["passive_ability"]["name"],
|
|
"special_ability_info": _weapon_data["passive_ability"]["description"],
|
|
}
|
|
_template_data["weapon_info_source_list"].append(
|
|
await url_to_file(_weapon_data["materials"]["ascension"]["icon"])
|
|
)
|
|
_template_data["weapon_info_source_list"].append(
|
|
await url_to_file(_weapon_data["materials"]["elite"]["icon"])
|
|
)
|
|
_template_data["weapon_info_source_list"].append(
|
|
await url_to_file(_weapon_data["materials"]["monster"]["icon"])
|
|
)
|
|
return _template_data
|
|
|
|
template_data = await input_template_data(weapon_data)
|
|
png_data = await self.service.template.render('genshin/weapon', "weapon.html", template_data,
|
|
{"width": 540, "height": 540})
|
|
await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
|
|
await message.reply_photo(png_data, filename=f"{template_data['weapon_name']}.png",
|
|
allow_sending_without_reply=True)
|