from pydantic import BaseModel
from typing import Optional, Dict, Any
from datetime import datetime


class VasProductBase(BaseModel):
    title: Optional[str] = None
    description: Optional[str] = None
    extra_fields: Optional[Dict[str, Any]] = None

    enabled: Optional[int] = None

class VasProductCreate(BaseModel):
    country: str
    city: str
    visa_type: Optional[str]

    provider: str

    title: str
    description: Optional[str]
    extra_fields: Optional[Dict[str, Any]]

    price_amount: int
    price_currency: str

    schema_id: Optional[int]
    
class VasProductUpdate(VasProductBase):
    pass

class VasProductOut(VasProductBase):
    id: int

    country: str
    city: str
    visa_type: Optional[str]

    provider: str

    price_amount: int
    price_currency: str

    schema_id: Optional[int]

    created_at: datetime
    updated_at: datetime

    class Config:
        orm_mode = True