models.py

"""Pydantic response models."""

from __future__ import annotations

from datetime import UTC, datetime

from pydantic import BaseModel

from checker import ServerStatus, ServerTag


def utc_now_iso() -> str:
    return datetime.now(UTC).strftime("%Y-%m-%dT%H:%M:%SZ")


class ServerOut(BaseModel):
    id: int
    name: str
    status: ServerStatus
    tag: ServerTag | None = None


class RegionOut(BaseModel):
    key: str
    name: str
    error: str | None = None
    cached: bool
    queried_at: str
    total: int
    servers: list[ServerOut]


class StatusResponse(BaseModel):
    generated_at: str
    regions: list[RegionOut]


class SummaryRegionOut(BaseModel):
    key: str
    name: str
    error: str | None = None
    cached: bool
    queried_at: str
    total: int
    normal: int = 0
    maintenance: int = 0
    full: int = 0
    reg_full: int = 0
    unopened: int = 0
    unknown: int = 0


class SummaryResponse(BaseModel):
    generated_at: str
    regions: list[SummaryRegionOut]


class RegionInfoOut(BaseModel):
    key: str
    name: str
    protocol: str


class RegionsResponse(BaseModel):
    regions: list[RegionInfoOut]


class SingleRegionResponse(RegionOut):
    pass


class SingleServerResponse(BaseModel):
    region: str
    region_name: str
    server: ServerOut | None = None


class HealthResponse(BaseModel):
    status: str