mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-12-04 20:47:27 +00:00
87 lines
3.7 KiB
Python
87 lines
3.7 KiB
Python
|
from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
|
||
|
|
from telegram.constants import ChatAction, ParseMode
|
||
|
|
from telegram.ext import CallbackContext, filters
|
||
|
|
|
||
|
|
from core.plugin import Plugin, handler
|
||
|
|
from core.services.game.services import GameCacheService
|
||
|
|
from core.services.search.models import StrategyEntry
|
||
|
|
from core.services.search.services import SearchServices
|
||
|
|
from core.services.wiki.services import WikiService
|
||
|
|
from metadata.shortname import roleToName, roleToTag
|
||
|
|
from utils.log import logger
|
||
|
|
|
||
|
|
|
||
|
|
class AvatarsPlugin(Plugin):
|
||
|
|
"""角色图鉴查询"""
|
||
|
|
|
||
|
|
KEYBOARD = [
|
||
|
|
[InlineKeyboardButton(text="查看角色图鉴列表并查询", switch_inline_query_current_chat="查看角色图鉴列表并查询")]
|
||
|
|
]
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
cache_service: GameCacheService = None,
|
||
|
|
wiki_service: WikiService = None,
|
||
|
|
search_service: SearchServices = None,
|
||
|
|
):
|
||
|
|
self.cache_service = cache_service
|
||
|
|
self.wiki_service = wiki_service
|
||
|
|
self.search_service = search_service
|
||
|
|
|
||
|
|
@handler.command(command="avatar", block=False)
|
||
|
|
@handler.message(filters=filters.Regex("^角色图鉴查询(.*)"), block=False)
|
||
|
|
async def command_start(self, update: Update, context: CallbackContext) -> None:
|
||
|
|
message = update.effective_message
|
||
|
|
args = self.get_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(message)
|
||
|
|
self.add_delete_message_job(reply_message)
|
||
|
|
return
|
||
|
|
character_name = roleToName(character_name)
|
||
|
|
file_path = self.wiki_service.raider.raider_guide_for_role_path / f"{character_name}.png"
|
||
|
|
if not file_path.exists():
|
||
|
|
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(message)
|
||
|
|
self.add_delete_message_job(reply_message)
|
||
|
|
return
|
||
|
|
self.log_user(update, logger.info, "查询角色图鉴命令请求 || 参数 %s", character_name)
|
||
|
|
await message.reply_chat_action(ChatAction.UPLOAD_PHOTO)
|
||
|
|
caption = "From 米游社@听语惊花"
|
||
|
|
if file_id := await self.cache_service.get_avatar_cache(character_name):
|
||
|
|
await message.reply_document(
|
||
|
|
document=file_id,
|
||
|
|
caption=caption,
|
||
|
|
filename=f"{character_name}.png",
|
||
|
|
parse_mode=ParseMode.HTML,
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
reply_photo = await message.reply_document(
|
||
|
|
document=open(file_path, "rb"),
|
||
|
|
caption=caption,
|
||
|
|
filename=f"{character_name}.png",
|
||
|
|
parse_mode=ParseMode.HTML,
|
||
|
|
)
|
||
|
|
if reply_photo.document:
|
||
|
|
tags = roleToTag(character_name)
|
||
|
|
photo_file_id = reply_photo.document.file_id
|
||
|
|
await self.cache_service.set_avatar_cache(character_name, photo_file_id)
|
||
|
|
entry = StrategyEntry(
|
||
|
|
key=f"plugin:avatar:{character_name}",
|
||
|
|
title=character_name,
|
||
|
|
description=f"{character_name} 角色图鉴",
|
||
|
|
tags=tags,
|
||
|
|
caption=caption,
|
||
|
|
parse_mode="HTML",
|
||
|
|
document_file_id=photo_file_id,
|
||
|
|
)
|
||
|
|
await self.search_service.add_entry(entry)
|