SIMNet/simnet/models/starrail/chronicle/stats.py

"""Starrail chronicle stats."""
import typing

from pydantic import Field

from simnet.models.base import APIModel

from .. import character

__all__ = [
    "PartialStarRailUserStats",
    "StarRailUserInfo",
    "StarRailUserStats",
    "StarRailStats",
]


class StarRailStats(APIModel):
    """Overall user stats."""

    active_days: int
    avatar_num: int
    achievement_num: int
    chest_num: int
    abyss_process: str


class PartialStarRailUserStats(APIModel):
    """User stats with characters without equipment."""

    stats: StarRailStats
    characters: typing.Sequence[character.StarRailPartialCharacter] = Field(alias="avatar_list")


class StarRailUserInfo(APIModel):
    """User info."""

    nickname: str
    server: str = Field(alias="region")
    level: int
    avatar: str


class StarRailUserStats(PartialStarRailUserStats):
    """User stats."""

    info: StarRailUserInfo