diff --git a/plugins/genshin/birthday.py b/plugins/genshin/birthday.py index cf050aff..71ee1410 100644 --- a/plugins/genshin/birthday.py +++ b/plugins/genshin/birthday.py @@ -1,18 +1,35 @@ import re from datetime import datetime +from typing import List -from telegram import Update -from telegram.ext import CommandHandler, CallbackContext +from genshin import Client, GenshinException +from genshin.client.routes import Route +from genshin.utility import recognize_genshin_server +from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup +from telegram.constants import ParseMode +from telegram.ext import CommandHandler, CallbackContext, MessageHandler from telegram.ext import filters +from telegram.helpers import create_deep_linked_url from core.baseplugin import BasePlugin +from core.cookies import CookiesService +from core.cookies.error import CookiesNotFoundError from core.plugin import Plugin, handler +from core.user import UserService +from core.user.error import UserNotFoundError from metadata.genshin import AVATAR_DATA from metadata.shortname import roleToId, roleToName from utils.bot import get_args from utils.decorators.error import error_callable from utils.decorators.restricts import restricts +from utils.genshin import fetch_hk4e_token_by_cookie, recognize_genshin_game_biz +from utils.helpers import get_genshin_client from utils.log import logger +from utils.models.base import RegionEnum + +BIRTHDAY_URL = Route( + "https://hk4e-api.mihoyo.com/event/birthdaystar/account/post_my_draw", +) def rm_starting_str(string, starting): @@ -25,7 +42,11 @@ def rm_starting_str(string, starting): class BirthdayPlugin(Plugin, BasePlugin): """Birthday.""" - def __init__(self): + def __init__( + self, + user_service: UserService = None, + cookie_service: CookiesService = None, + ): """Load Data.""" self.birthday_list = {} for value in AVATAR_DATA.values(): @@ -33,6 +54,16 @@ class BirthdayPlugin(Plugin, BasePlugin): data = self.birthday_list.get(key, []) data.append(value["name"]) self.birthday_list.update({key: data}) + self.user_service = user_service + self.cookie_service = cookie_service + + def get_today_birthday(self) -> List[str]: + key = ( + rm_starting_str(datetime.now().strftime("%m"), "0") + + "_" + + rm_starting_str(datetime.now().strftime("%d"), "0") + ) + return self.birthday_list.get(key, []) @handler(CommandHandler, command="birthday", block=False) @restricts() @@ -87,7 +118,7 @@ class BirthdayPlugin(Plugin, BasePlugin): 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, []) + today_list = self.get_today_birthday() if key == "6_1": text = f"今天是 派蒙、{'、'.join(today_list)} 的生日哦~" else: @@ -96,3 +127,72 @@ class BirthdayPlugin(Plugin, BasePlugin): 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) + + @staticmethod + async def get_card(client: Client, role_id: int) -> None: + """领取画片""" + url = BIRTHDAY_URL.get_url() + params = { + "game_biz": recognize_genshin_game_biz(client.uid), + "lang": "zh-cn", + "badge_uid": client.uid, + "badge_region": recognize_genshin_server(client.uid), + "activity_id": "20220301153521", + } + json = { + "role_id": role_id, + } + await client.cookie_manager.request(url, method="POST", params=params, json=json) + + @handler(CommandHandler, command="birthday_card", block=False) + @handler(MessageHandler, filters=filters.Regex("^领取角色生日画片$"), block=False) + @restricts() + @error_callable + async def command_birthday_card_start(self, update: Update, context: CallbackContext) -> None: + message = update.effective_message + user = update.effective_user + logger.info("用户 %s[%s] 领取生日画片命令请求", user.full_name, user.id) + today_list = self.get_today_birthday() + if not today_list: + reply_message = await message.reply_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) + return + try: + client = await get_genshin_client(user.id) + if client.region == RegionEnum.HOYOLAB: + text = "此功能当前只支持国服账号哦~" + else: + await fetch_hk4e_token_by_cookie(client) + for name in today_list.copy(): + if role_id := roleToId(name): + try: + await self.get_card(client, role_id) + except GenshinException as e: + if e.retcode in {-512008, -512009}: # 未过生日、已领取过 + today_list.remove(name) + if today_list: + text = f"成功领取了 {'、'.join(today_list)} 的生日画片~" + else: + text = "没有领取到生日画片哦 ~ 可能是已经领取过了" + 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 (UserNotFoundError, CookiesNotFoundError): + buttons = [[InlineKeyboardButton("点我绑定账号", url=create_deep_linked_url(context.bot.username, "set_cookie"))]] + if filters.ChatType.GROUPS.filter(message): + reply_msg = await message.reply_text( + "此功能需要绑定cookie后使用,请先私聊派蒙绑定账号", + reply_markup=InlineKeyboardMarkup(buttons), + parse_mode=ParseMode.HTML, + ) + self._add_delete_message_job(context, reply_msg.chat_id, reply_msg.message_id, 30) + self._add_delete_message_job(context, message.chat_id, message.message_id, 30) + else: + await message.reply_text( + "此功能需要绑定cookie后使用,请先私聊派蒙进行绑定", + parse_mode=ParseMode.HTML, + reply_markup=InlineKeyboardMarkup(buttons), + ) diff --git a/resources/bot/help/help.html b/resources/bot/help/help.html index 1000952c..9e9ccbeb 100644 --- a/resources/bot/help/help.html +++ b/resources/bot/help/help.html @@ -145,6 +145,10 @@
/birthday
角色生日
+
+
/birthday_card
+
领取角色生日画片
+