from pydantic import BaseModel
from typing import Optional
from datetime import date


class AdminUnitType(BaseModel):
    id:          int
    name:        str
    slug:        str
    description: Optional[str]


class AdminUnit(BaseModel):
    id:             int
    type_id:        int
    type_name:      str
    name:           str
    established_at: Optional[date]
    abolished_at:   Optional[date]
    is_active:      bool


class AdminUnitCreate(BaseModel):
    type_id:        int
    name:           str
    established_at: Optional[date] = None


class AdminUnitAssign(BaseModel):
    unit_id:     int
    valid_from:  date
    valid_until: Optional[date] = None


class AdminUnitClose(BaseModel):
    abolished_at: date
    valid_until:  date