SIMNet/simnet/models/zzz/diary.py
2024-09-25 23:14:21 +08:00

99 lines
2.2 KiB
Python

from typing import List
from pydantic import Field
from simnet.models.base import APIModel
from simnet.models.diary import BaseDiary
__all__ = (
"ZZZDiaryDataList",
"ZZZDiaryActionCategory",
"ZZZMonthDiaryData",
"ZZZDiary",
)
class ZZZDiaryDataList(APIModel):
"""List of diary data.
Attributes:
id: Data ID.
name: Data name.
amount: Amount of data.
"""
id: str = Field(alias="data_type")
name: str = Field(alias="data_name")
amount: int = Field(alias="count")
class ZZZDiaryActionCategory(APIModel):
"""Diary category for PolychromesData .
Attributes:
action: Category name.
amount: Amount of rails_pass.
percentage: Percentage of rails_pass.
"""
action: str
amount: int = Field(alias="num")
percentage: int = Field(alias="percent")
@property
def name(self) -> str:
return {
"growth_rewards": "成长奖励",
"daily_activity_rewards": "日常活跃奖励",
"mail_rewards": "邮件奖励",
"event_rewards": "活动奖励",
"hollow_rewards": "零号空洞奖励",
"shiyu_rewards": "式舆防卫战奖励",
"other_rewards": "其他奖励",
}.get(self.action, "其他奖励")
class ZZZMonthDiaryData(APIModel):
"""Diary data for a month.
Attributes:
categories: List of diary categories.
"""
list: List[ZZZDiaryDataList]
categories: List[ZZZDiaryActionCategory] = Field(alias="income_components")
class ZZZDiaryRoleInfo(APIModel):
"""Role info for a diary."""
nickname: str
avatar: str
class ZZZDiary(BaseDiary):
"""Traveler's diary.
Attributes:
data: Diary data for a month.
optional_month: Optional month.
current_month: Current month.
data_month: Data month.
"""
data: ZZZMonthDiaryData = Field(alias="month_data")
optional_month: List[str]
current_month: str
data_month: str
role_info: ZZZDiaryRoleInfo
@property
def data_id(self) -> int:
"""Get the data ID."""
return self.month
@property
def month_data(self) -> ZZZMonthDiaryData:
"""Diary data for a month."""
return self.data