from pydantic import BaseModel
from typing import Optional

class VisaflyConfigBase(BaseModel):
    submission_country: str
    submission_country_code: str
    submission_city: str
    submission_city_code: str
    travel_country: str
    travel_country_code: str
    visa_type: str
    provider: Optional[str] = None
    field_requirement_type: Optional[str] = None

class VisaflyConfigCreate(VisaflyConfigBase):
    pass

class VisaflyConfigOut(VisaflyConfigBase):
    id: int

    class Config:
        orm_mode = True