from pydantic import BaseModel
from typing import Optional

class VehicleBaseDTO(BaseModel):
    vehicle_key: str
    ace_vehicle_key: Optional[str]
    transporter_key: Optional[str]
    transport_identifier: Optional[str]
    transport_type: Optional[str]
    entity_code: Optional[str]
    transponder_number: Optional[str]
    dot_number: Optional[str]
    plate_number: Optional[str]
    city: Optional[str]
    state: Optional[str]
    country: Optional[str]
    seal: Optional[str]
    insurance_company_name: Optional[str]
    insurance_number: Optional[str]
    insurance_amount: Optional[float]
    insurance_date: Optional[int]
    box_number: Optional[str]
    brand: Optional[str]
    year: Optional[str]
    series: Optional[str]
    description: Optional[str]
    engine_number: Optional[str]
    sct_permission: Optional[str]
    color: Optional[str]
    container_key: Optional[str]
    company_id: str
    tenant_id: str

class VehicleCreateDTO(VehicleBaseDTO):
    pass

class VehicleResponseDTO(VehicleBaseDTO):
    class Config:
        orm_mode = True