from pydantic import BaseModel


class SchemeVersionListItem(BaseModel):
    scheme_version_id: str
    scheme_id: str
    version_number: int
    status: str
    normalized_storage_path: str
    normalized_elements_count: int
    normalized_seats_count: int
    normalized_groups_count: int
    normalized_sectors_count: int
    created_at: str


class SchemeVersionListResponse(BaseModel):
    items: list[SchemeVersionListItem]
    total: int


class SchemeVersionCreateResponse(BaseModel):
    scheme_id: str
    scheme_version_id: str
    version_number: int
    status: str
    normalized_storage_path: str


class EnsureDraftResponse(BaseModel):
    scheme_id: str
    scheme_version_id: str
    version_number: int
    status: str
    normalized_storage_path: str
    created: bool
    source_scheme_version_id: str | None = None