from pydantic import BaseModel
from typing import Optional


class PartnerBase(BaseModel):
    name: str
    phone: Optional[str] = None
    mobile: Optional[str] = None
    email: Optional[str] = None
    street: Optional[str] = None
    city: Optional[str] = None
    country_id: Optional[int] = None
    comment: Optional[str] = None


class PartnerCreate(PartnerBase):
    pass


class PartnerUpdate(BaseModel):
    name: Optional[str] = None
    phone: Optional[str] = None
    mobile: Optional[str] = None
    email: Optional[str] = None
    street: Optional[str] = None
    city: Optional[str] = None
    comment: Optional[str] = None


class PartnerResponse(PartnerBase):
    id: int
    display_name: Optional[str] = None
    credit: Optional[float] = None
    debit: Optional[float] = None
    credit_limit: Optional[float] = None


class PartnerSearchResult(BaseModel):
    id: int
    name: str
    phone: Optional[str] = None
    mobile: Optional[str] = None
    email: Optional[str] = None