"""Pydantic v2 models for Scenar Creator."""

from datetime import date, time
from typing import List, Optional

from pydantic import BaseModel, Field


class Block(BaseModel):
    datum: date
    zacatek: time
    konec: time
    program: str
    typ: str
    garant: Optional[str] = None
    poznamka: Optional[str] = None


class ProgramType(BaseModel):
    code: str
    description: str
    color: str  # hex #RRGGBB


class EventInfo(BaseModel):
    title: str = Field(..., max_length=200)
    detail: str = Field(..., max_length=500)


class ScenarioDocument(BaseModel):
    version: str = "1.0"
    event: EventInfo
    program_types: List[ProgramType]
    blocks: List[Block]