models.py

from pydantic import BaseModel, validator
from decimal import Decimal
from typing import List, Optional
import datetime


class PostResponseData(BaseModel):
    """Response body for POST (INSERT) actions"""

    inserted: int
    xtime: float
    last_id: Optional[int] = None
    trx: Optional[List[int]] = None
    rank: Optional[int] = None


class Checkpoint(BaseModel):
    """Body for adding or updating **Checkpoints** table entry"""

    cp: int
    run_time: int
    start_vel_x: Decimal
    start_vel_y: Decimal
    start_vel_z: Decimal
    end_vel_x: Decimal
    end_vel_y: Decimal
    end_vel_z: Decimal
    end_touch: int
    attempts: int


class CurrentRun(BaseModel):
    """Body for adding or updating **MapTimes** table entry"""

    player_id: int
    map_id: int
    run_time: int
    start_vel_x: Decimal
    start_vel_y: Decimal
    start_vel_z: Decimal
    end_vel_x: Decimal
    end_vel_y: Decimal
    end_vel_z: Decimal
    style: int = 0
    type: int = 0
    stage: int = 0
    checkpoints: Optional[List[Checkpoint]] = (
        None  # Required when adding a Map Run Time (type = 0)
    )
    replay_frames: str
    run_date: Optional[int] = None

    @validator("run_date", pre=True, always=True)
    def default_timestamp(cls, v):
        """Automatically add the `UNIX` timestamps so we don't need to include them in the Body of the API call"""
        if v is None or v == 0:
            return int(datetime.datetime.now(datetime.timezone.utc).timestamp())
        return v


class MapInfoModel(BaseModel):
    """Body for adding or updating **Map** entry"""

    id: int = None
    name: str
    author: str = "Unknown"
    tier: int
    stages: int
    bonuses: int = 0
    ranked: int = 0
    date_added: int = None
    last_played: int = None

    @validator("date_added", "last_played", pre=True, always=True)
    def default_timestamp(cls, v):
        """Automatically add the `UNIX` timestamps so we don't need to include them in the Body of the API call"""
        if v is None or v == 0:
            return int(datetime.datetime.now(datetime.timezone.utc).timestamp())
        return v

    @validator("ranked", pre=True)
    def ranked_bool_to_int(cls, v):
        return int(v)  # True -> 1, False -> 0, or passthrough if already int


class PlayerSurfProfile(BaseModel):
    """Model for player profiles"""

    id: int
    name: str
    steam_id: int
    country: str
    join_date: int
    last_seen: int
    connections: int

    @validator("join_date", "last_seen", pre=True, always=True)
    def default_timestamp(cls, v):
        """Automatically add the `UNIX` timestamps so we don't need to include them in the Body of the API call"""
        if v is None or v == 0:
            return int(datetime.datetime.now(datetime.timezone.utc).timestamp())
        return v