mirror of
https://github.com/PaiGramTeam/SIMNet.git
synced 2024-11-27 08:06:20 +00:00
114 lines
2.5 KiB
Python
114 lines
2.5 KiB
Python
from datetime import datetime
|
|
from typing import List
|
|
|
|
from pydantic import Field
|
|
|
|
from simnet.models.base import APIModel
|
|
from simnet.models.diary import BaseDiary
|
|
|
|
__all__ = (
|
|
"DayDiaryData",
|
|
"Diary",
|
|
"DiaryAction",
|
|
"DiaryActionCategory",
|
|
"DiaryPage",
|
|
"MonthDiaryData",
|
|
)
|
|
|
|
|
|
class DiaryActionCategory(APIModel):
|
|
"""Diary category for primogems.
|
|
|
|
Attributes:
|
|
id: Category ID.
|
|
name: Category name.
|
|
amount: Amount of primogems.
|
|
percentage: Percentage of primogems.
|
|
"""
|
|
|
|
id: int = Field(alias="action_id")
|
|
name: str = Field(alias="action")
|
|
amount: int = Field(alias="num")
|
|
percentage: int = Field(alias="percent")
|
|
|
|
|
|
class MonthDiaryData(APIModel):
|
|
"""Diary data for a month.
|
|
|
|
Attributes:
|
|
current_primogems: Current amount of primogems.
|
|
current_mora: Current amount of mora.
|
|
last_primogems: Last amount of primogems.
|
|
last_mora: Last amount of mora.
|
|
primogems_rate: Primogems rate.
|
|
mora_rate: Mora rate.
|
|
categories: List of diary categories.
|
|
"""
|
|
|
|
def __init__(self, **data):
|
|
if data.get("primogem_rate"):
|
|
data["primogems_rate"] = data.pop("primogem_rate")
|
|
super().__init__(**data)
|
|
|
|
current_primogems: int
|
|
current_mora: int
|
|
last_primogems: int
|
|
last_mora: int
|
|
primogems_rate: int
|
|
mora_rate: int
|
|
categories: List[DiaryActionCategory] = Field(alias="group_by")
|
|
|
|
|
|
class DayDiaryData(APIModel):
|
|
"""Diary data for a day.
|
|
|
|
Attributes:
|
|
current_primogems: Current amount of primogems.
|
|
current_mora: Current amount of mora.
|
|
"""
|
|
|
|
current_primogems: int
|
|
current_mora: int
|
|
|
|
|
|
class Diary(BaseDiary):
|
|
"""Traveler's diary.
|
|
|
|
Attributes:
|
|
data: Diary data for a month.
|
|
day_data: Diary data for a day.
|
|
"""
|
|
|
|
data: MonthDiaryData = Field(alias="month_data")
|
|
day_data: DayDiaryData
|
|
|
|
@property
|
|
def month_data(self) -> MonthDiaryData:
|
|
return self.data
|
|
|
|
|
|
class DiaryAction(APIModel):
|
|
"""Action which earned currency.
|
|
|
|
Attributes:
|
|
action_id: Action ID.
|
|
action: Action name.
|
|
time: Time of the action.
|
|
amount: Amount of the action.
|
|
"""
|
|
|
|
action_id: int
|
|
action: str
|
|
time: datetime = Field(timezone=8)
|
|
amount: int = Field(alias="num")
|
|
|
|
|
|
class DiaryPage(BaseDiary):
|
|
"""Page of a diary.
|
|
|
|
Attributes:
|
|
actions: List of diary actions.
|
|
"""
|
|
|
|
actions: List[DiaryAction] = Field(alias="list")
|