mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 12:02:16 +00:00
233e7ab58d
Co-authored-by: luoshuijs <luoshuijs@outlook.com> Co-authored-by: Karako <karakohear@gmail.com> Co-authored-by: xtaodada <xtao@xtaolink.cn>
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from modules.gacha.banner import BannerType, GachaBanner
|
|
from modules.gacha.player.banner import PlayerGachaBannerInfo
|
|
|
|
|
|
class PlayerGachaInfo(BaseModel):
|
|
"""玩家抽卡全部信息"""
|
|
|
|
standard_banner: Optional[PlayerGachaBannerInfo] = None
|
|
event_weapon_banner: Optional[PlayerGachaBannerInfo] = None
|
|
event_character_banner: Optional[PlayerGachaBannerInfo] = None
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
if self.standard_banner is None:
|
|
self.standard_banner = PlayerGachaBannerInfo()
|
|
if self.event_weapon_banner is None:
|
|
self.event_weapon_banner = PlayerGachaBannerInfo()
|
|
if self.event_character_banner is None:
|
|
self.event_character_banner = PlayerGachaBannerInfo()
|
|
|
|
def get_banner_info(self, banner: GachaBanner) -> PlayerGachaBannerInfo:
|
|
if banner.banner_type == BannerType.EVENT:
|
|
return self.event_character_banner
|
|
if banner.banner_type == BannerType.WEAPON:
|
|
return self.event_weapon_banner
|
|
return self.standard_banner
|