diff --git a/metadata/genshin.py b/metadata/genshin.py index 253d651..2745e77 100644 --- a/metadata/genshin.py +++ b/metadata/genshin.py @@ -3,7 +3,7 @@ from __future__ import annotations import functools -from typing import Any, Generic, ItemsView, Iterator, KeysView, TypeVar, Optional +from typing import Any, Generic, ItemsView, Iterator, KeysView, TypeVar, Optional, ValuesView import ujson as json @@ -22,7 +22,7 @@ __all__ = [ "game_id_to_role_id", "Data", "weapon_to_game_id", - "avatar_to_game_id" + "avatar_to_game_id", ] K = TypeVar("K") @@ -78,6 +78,9 @@ class Data(dict, Generic[K, V]): def keys(self) -> KeysView[K, V]: return self.data.keys() + def values(self) -> ValuesView[K, V]: + return self.data.values() + def items(self) -> ItemsView[K, V]: return self.data.items() @@ -105,13 +108,9 @@ def game_id_to_role_id(gid: str) -> int | None: @functools.lru_cache() def weapon_to_game_id(name: str) -> Optional[int]: - return next( - (int(key) for key, value in WEAPON_DATA.items() if value['name'] == name), None - ) + return next((int(key) for key, value in WEAPON_DATA.items() if value["name"] == name), None) @functools.lru_cache() def avatar_to_game_id(name: str) -> Optional[int]: - return next( - (int(key) for key, value in AVATAR_DATA.items() if value['name'] == name), None - ) + return next((int(key) for key, value in AVATAR_DATA.items() if value["name"] == name), None) diff --git a/plugins/genshin/birthday.py b/plugins/genshin/birthday.py new file mode 100644 index 0000000..908a202 --- /dev/null +++ b/plugins/genshin/birthday.py @@ -0,0 +1,77 @@ +import re +from datetime import datetime + +from telegram import Update +from telegram.ext import CommandHandler, CallbackContext +from telegram.ext import filters + +from core.baseplugin import BasePlugin +from core.plugin import Plugin, handler +from metadata.genshin import AVATAR_DATA +from metadata.shortname import roleToId, roleToName +from utils.bot import get_all_args +from utils.decorators.error import error_callable +from utils.decorators.restricts import restricts +from utils.log import logger + + +class BirthdayPlugin(Plugin, BasePlugin): + """Birthday.""" + + def __init__(self): + """Load Data.""" + self.birthday_list = {} + for value in AVATAR_DATA.values(): + key = "_".join([str(i) for i in value["birthday"]]) + data = self.birthday_list.get(key, []) + data.append(value["name"]) + self.birthday_list.update({key: data}) + + @handler(CommandHandler, command="birthday", block=False) + @restricts() + @error_callable + async def command_start(self, update: Update, context: CallbackContext) -> None: + message = update.effective_message + user = update.effective_user + key = datetime.now().strftime("%m_%d") + args = get_all_args(context) + if len(args) >= 1: + msg = args[0] + logger.info(f"用户 {user.full_name}[{user.id}] 查询角色生日命令请求 || 参数 {msg}") + if re.match(r"\d{1,2}.\d{1,2}", msg): + try: + key = re.findall(r"\d+", msg)[0] + "_" + re.findall(r"\d+", msg)[1] + day_list = self.birthday_list.get(key, []) + date = re.findall(r"\d+", msg)[0] + "月" + re.findall(r"\d+", msg)[1] + "日" + text = f"{date} 是 {'、'.join(day_list)} 的生日哦~" if day_list else f"{date} 没有角色过生日哦~" + except IndexError: + text = "请输入正确的日期格式,如1-1,或输入正确的角色名称。" + reply_message = await message.reply_text(text) + 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) + else: + try: + name = roleToName(msg) + aid = str(roleToId(msg)) + birthday = AVATAR_DATA[aid]["birthday"] + text = f"{name} 的生日是 {birthday[0]}月{birthday[1]}日 哦~" + reply_message = await message.reply_text(text) + 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) + except KeyError: + reply_message = await message.reply_text( + "请输入正确的日期格式,如1-1,或输入正确的角色名称。" + ) + 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) + else: + logger.info(f"用户 {user.full_name}[{user.id}] 查询今日角色生日列表") + today_list = self.birthday_list.get(key, []) + text = f"今天是 {'、'.join(today_list)} 的生日哦~" if today_list else "今天没有角色过生日哦~" + reply_message = await message.reply_text(text) + 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) diff --git a/resources/bot/help/help.html b/resources/bot/help/help.html index e347cfd..516a28b 100644 --- a/resources/bot/help/help.html +++ b/resources/bot/help/help.html @@ -133,6 +133,10 @@
/hilichurls
丘丘语字典
+
+
/birthday
+
角色生日
+