mirror of
https://github.com/PaiGramTeam/SIMNet.git
synced 2024-11-22 22:37:24 +00:00
65 lines
1.4 KiB
Python
65 lines
1.4 KiB
Python
|
from datetime import datetime
|
||
|
|
from enum import IntEnum
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from pydantic import Field, validator
|
||
|
|
|
||
|
|
from simnet.models.base import APIModel
|
||
|
|
|
||
|
|
|
||
|
|
class ZZZBannerType(IntEnum):
|
||
|
|
"""Banner types in wish histories."""
|
||
|
|
|
||
|
|
STANDARD = PERMANENT = NOVICE = 1
|
||
|
|
"""Permanent standard banner."""
|
||
|
|
|
||
|
|
CHARACTER = 2
|
||
|
|
"""Rotating character banner."""
|
||
|
|
|
||
|
|
WEAPON = 3
|
||
|
|
"""Rotating weapon banner."""
|
||
|
|
|
||
|
|
BANGBOO = 5
|
||
|
|
"""BangBoo banner."""
|
||
|
|
|
||
|
|
|
||
|
|
class ZZZWish(APIModel):
|
||
|
|
"""Wish made on any banner."""
|
||
|
|
|
||
|
|
uid: int
|
||
|
|
"""User ID of the wish maker."""
|
||
|
|
|
||
|
|
id: int
|
||
|
|
"""ID of the wished item."""
|
||
|
|
|
||
|
|
type: str = Field(alias="item_type")
|
||
|
|
"""Type of the wished item."""
|
||
|
|
|
||
|
|
item_id: int = Field(alias="item_id")
|
||
|
|
"""ID of the wished item."""
|
||
|
|
|
||
|
|
name: str
|
||
|
|
"""Name of the wished item."""
|
||
|
|
|
||
|
|
rarity: int = Field(alias="rank_type")
|
||
|
|
"""Rarity of the wished item."""
|
||
|
|
|
||
|
|
time: datetime
|
||
|
|
"""Time when the wish was made."""
|
||
|
|
|
||
|
|
banner_id: int = Field(alias="gacha_id")
|
||
|
|
"""ID of the banner the wish was made on."""
|
||
|
|
|
||
|
|
banner_type: ZZZBannerType = Field(alias="gacha_type")
|
||
|
|
"""Type of the banner the wish was made on."""
|
||
|
|
|
||
|
|
@validator("banner_type", pre=True)
|
||
|
|
def cast_banner_type(cls, v: Any) -> int:
|
||
|
|
"""Converts the banner type from any type to int."""
|
||
|
|
return int(v)
|
||
|
|
|
||
|
|
@validator("rarity")
|
||
|
|
def add_rarity(cls, v: int) -> int:
|
||
|
|
"""Add rarity 1."""
|
||
|
|
return v + 1
|