PaiGramTeam/PamGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PamGram.git synced 2024-11-16 12:02:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎨 Make GachaHandle.de_item_list Raise GachaDataFound

Browse Source
This commit is contained in:
洛水居室 2022-12-07 11:03:42 +08:00
parent efc3dedc74
commit f2b89134cb
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
plugins/genshin/gacha/gacha.py
View File

@ -34,6 +34,14 @@ class GachaNotFound(Exception):
super().__init__(f"{gacha_name} gacha not found") super().__init__(f"{gacha_name} gacha not found")
class GachaDataFound(Exception):
"""卡池数据未找到"""
def __init__(self, item_id: int):
self.item_id = item_id
super().__init__(f"item_id[{item_id}] data not found")
class GachaRedis: class GachaRedis:
def __init__(self, redis: RedisDB): def __init__(self, redis: RedisDB):
self.client = redis.client self.client = redis.client
@ -152,12 +160,16 @@ class Gacha(Plugin, BasePlugin):
data = WEAPON_DATA.get(str(item_id)) data = WEAPON_DATA.get(str(item_id))
avatar = self.assets_service.weapon(item_id) avatar = self.assets_service.weapon(item_id)
gacha = await avatar.gacha() gacha = await avatar.gacha()
if gacha is None:
raise GachaDataFound(item_id)
data.setdefault("url", gacha.as_uri()) data.setdefault("url", gacha.as_uri())
gacha_item.append(data) gacha_item.append(data)
elif 10000000 <= item_id <= 19999999: elif 10000000 <= item_id <= 19999999:
data = AVATAR_DATA.get(str(item_id)) data = AVATAR_DATA.get(str(item_id))
avatar = self.assets_service.avatar(item_id) avatar = self.assets_service.avatar(item_id)
gacha = await avatar.gacha_card() gacha = await avatar.gacha_card()
if gacha is None:
raise GachaDataFound(item_id)
data.setdefault("url", gacha.as_uri()) data.setdefault("url", gacha.as_uri())
gacha_item.append(data) gacha_item.append(data)
return gacha_item return gacha_item
@ -203,7 +215,15 @@ class Gacha(Plugin, BasePlugin):
player_gacha_info.event_weapon_banner.wish_item_id = 0 player_gacha_info.event_weapon_banner.wish_item_id = 0
# 执行抽卡 # 执行抽卡
item_list = self.banner_system.do_pulls(player_gacha_info, banner, 10) item_list = self.banner_system.do_pulls(player_gacha_info, banner, 10)
try:
data = await self.de_item_list(item_list) data = await self.de_item_list(item_list)
except GachaDataFound as exc:
logger.warning("角色 item_id[%s] 抽卡立绘未找到", exc.item_id)
reply_message = await message.reply_text("出错了呜呜呜 ~ 卡池部分数据未找到！")
if filters.ChatType.GROUPS.filter(message):
self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id, 60)
self._add_delete_message_job(context, message.chat_id, message.message_id, 60)
return
player_gacha_banner_info = player_gacha_info.get_banner_info(banner) player_gacha_banner_info = player_gacha_info.get_banner_info(banner)
template_data = { template_data = {
"name": f"{user.full_name}", "name": f"{user.full_name}",