from pydantic import BaseModel
from typing import Optional, List

class CustomsBrokerDTO(BaseModel):
    type: Optional[str]
    broker_key: str
    name: Optional[str]
    address: Optional[str]
    postal_code: Optional[str]
    city: Optional[str]
    state: Optional[str]
    phone: Optional[str]
    fax: Optional[str]
    email: Optional[str]
    country: Optional[str]
    tax_id: Optional[str]
    personal_id: Optional[str]
    position: Optional[str]
    license: Optional[str]
    company: Optional[str]
    contact: Optional[str]
    tenant_id: str
    company_id: str

    class Config:
        orm_mode = True


class CustomsBrokerVUCreateDTO(BaseModel):
    certificate_path: Optional[str]
    key_path: Optional[str]
    access_key: Optional[str]
    fiel_format: Optional[str]
    signature_read_path: Optional[str]
    archive_path: Optional[str]
    fiel_access_key: Optional[str]
    web_service_user: Optional[str]
    web_service_access_key: Optional[str]
    vu_email: Optional[str]
    vu_figure_type: Optional[str]
    xml_files_path: Optional[str]
    query_tax_id: Optional[str]
    doda_certificate_path: Optional[str]
    doda_key_path: Optional[str]
    doda_web_service_user: Optional[str]
    doda_web_service_access_key: Optional[str]
    doda_fiel_access_key: Optional[str]
    doda_xml_files_path: Optional[str]

    class Config:
        orm_mode = True


class CustomsBrokerPersonnelDTO(BaseModel):
    broker_key: str
    line: int
    name: Optional[str]
    tax_id: Optional[str]
    personal_id: Optional[str]
    position: Optional[str]
    license: Optional[str]
    first_name: Optional[str]
    last_name: Optional[str]
    middle_name: Optional[str]
    email: Optional[str]

    class Config:
        orm_mode = True