mirror of
https://github.com/PaiGramTeam/SIMNet.git
synced 2024-11-25 23:51:12 +00:00
120 lines
2.7 KiB
Python
120 lines
2.7 KiB
Python
from datetime import datetime
|
|
from typing import List, Optional
|
|
|
|
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.
|
|
"""
|
|
|
|
current_primogems: int
|
|
current_mora: int
|
|
last_primogems: int
|
|
last_mora: int
|
|
primogems_rate: int = Field(aliases="primogem_rate")
|
|
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.
|
|
date: Request date.
|
|
"""
|
|
|
|
data: MonthDiaryData = Field(alias="month_data")
|
|
day_data: DayDiaryData
|
|
date: Optional[str] = ""
|
|
|
|
@property
|
|
def data_id(self) -> int:
|
|
"""Get the data ID."""
|
|
date = (self.date or datetime.now().strftime("%Y-%m-%d")).split("-")
|
|
year, month = int(date[0]), int(date[1])
|
|
if month < self.month:
|
|
year -= 1
|
|
return year * 100 + self.month
|
|
|
|
@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")
|