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

class SlotBase(BaseModel):
    submit_country: str
    submit_city: Optional[str] = None
    travel_country: str
    visa_type: str
    lasted_slot_date: Optional[date] = None
    available_dates: Optional[str] = None
    available_times: Optional[str] = None
    slot_number: Optional[int] = None
    website: Optional[str] = None
    open_mode: Optional[int] = 0

class SlotCreate(SlotBase):
    pass

class SlotOut(SlotBase):
    id: int

    class Config:
        orm_mode = True