from pydantic import BaseModel
from typing import Optional


class LeadCreate(BaseModel):
    name: str
    partner_id: Optional[int] = None
    contact_name: Optional[str] = None
    phone: Optional[str] = None
    mobile: Optional[str] = None
    email_from: Optional[str] = None
    description: Optional[str] = None
    expected_revenue: Optional[float] = None
    source: Optional[str] = "WhatsApp"


class LeadResponse(BaseModel):
    id: int
    name: str
    stage_id: int
    stage_name: str
    partner_id: Optional[int] = None
    partner_name: Optional[str] = None
    contact_name: Optional[str] = None
    phone: Optional[str] = None
    email_from: Optional[str] = None
    expected_revenue: float
    probability: float
    user_id: Optional[int] = None
    user_name: Optional[str] = None


class LeadSearchResult(BaseModel):
    id: int
    name: str
    stage_name: str
    expected_revenue: float
    probability: float