MibooGram/plugins/genshin/hilichurls.py

from typing import Dict

from aiofiles import open as async_open
from telegram import Update
from telegram.ext import CallbackContext, filters

from core.plugin import Plugin, handler
from utils.const import RESOURCE_DIR
from utils.log import logger

try:
    import ujson as jsonlib

except ImportError:
    import json as jsonlib

__all__ = ("HilichurlsPlugin",)


class HilichurlsPlugin(Plugin):
    """丘丘语字典."""

    hilichurls_dictionary: Dict[str, str]

    async def initialize(self) -> None:
        """加载数据文件.数据整理自 https://wiki.biligame.com/ys By @zhxycn."""
        async with async_open(RESOURCE_DIR / "json/hilichurls_dictionary.json", encoding="utf-8") as file:
            self.hilichurls_dictionary = jsonlib.loads(await file.read())

    @handler.command(command="hilichurls", 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:
            msg = args[0]
        else:
            reply_message = await message.reply_text("请输入要查询的丘丘语。")
            if filters.ChatType.GROUPS.filter(reply_message):
                self.add_delete_message_job(message)
                self.add_delete_message_job(reply_message)
            return
        search = str.casefold(msg)  # 忽略大小写以方便查询
        if search not in self.hilichurls_dictionary:
            reply_message = await message.reply_text(f"在丘丘语字典中未找到 {msg}。")
            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", msg)
        result = self.hilichurls_dictionary[f"{search}"]
        await message.reply_markdown_v2(f"丘丘语: `{search}`\n\n`{result}`")