from pydantic import BaseModel


class JdeOfficeDB(BaseModel):
    code: str
    title: str
    kladr_code: str
    aex_only: str
    mst_pr_aex: str
    mst_pr_virt: str
    addr: str
    features: str
    coords: dict[str, str]
    city: str
    country_code: str
    contry_name: str
    max_ves: str
    max_obyom: str
    max_ves_gm: str
    max_obyom_gm: str
    max_l_gm: str
    max_w_gm: str
    max_h_gm: str


class JdeOfficeLocalDB(BaseModel):
    code: str
    features: str | None = None
    contact_person_id: int | None = None
    person_count: int | None = None
    rating: int | None = None


class ProfileDB(BaseModel):
    id: int
    full_name: str | None
    phone: str | None
    email: str | None