diff --git a/plugins/genshin/birthday.py b/plugins/genshin/birthday.py
index cf050af..71ee141 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 1000952..9e9ccbe 100644
--- a/resources/bot/help/help.html
+++ b/resources/bot/help/help.html
@@ -145,6 +145,10 @@
/birthday
角色生日
+
+
/birthday_card
+
领取角色生日画片
+