from pydantic import BaseModel
from typing import Optional

class TrailerBaseDTO(BaseModel):
    trailer_number: str
    ace_trailer_number: Optional[str]
    trailer_type_key: Optional[str]
    seal: Optional[str]
    entity_code: Optional[str]
    plate_number: Optional[str]
    state: Optional[str]
    country: Optional[str]
    container_key: Optional[str]
    company_id: str
    tenant_id: str

class TrailerCreateDTO(TrailerBaseDTO):
    pass

class TrailerResponseDTO(TrailerBaseDTO):
    class Config:
        from_attributes = True