# app/schemas/slot_snapshot.py
from pydantic import BaseModel
from typing import Any, Optional
from datetime import datetime, date

class SlotSnapshotBase(BaseModel):
    country: str
    city: str
    visa_type: str
    routing_key: str
    availability_status: str
    earliest_date: Optional[date] = None
    website: Optional[str] = None
    availability: Any
    snapshot_source: str
    snapshot_at: datetime
    
class SlotSnapshotCreate(SlotSnapshotBase):
    pass 

class SlotSnapshotOut(SlotSnapshotBase):
    id: int
    model_config = {
        "from_attributes": True
    }
    
class SlotOverviewOut(SlotSnapshotBase):
    id: int
    routing_key:  Optional[str]=None
    snapshot_source: Optional[str]=None
    last_check_at: Optional[datetime]=None
    availability: Optional[Any]=None
    model_config = {
        "from_attributes": True
    }