from pydantic import BaseModel
from datetime import datetime
from typing import Optional, Dict

class CustomerRelationshipBase(BaseModel):
    company_code: str
    customer_name: str
    contact_person: str
    phone: Optional[str] = None
    email: Optional[str] = None
    social_media: Optional[Dict[str, str]] = None
    address: Optional[str] = None
    notes: Optional[str] = None

class CustomerRelationshipCreate(CustomerRelationshipBase):
    pass

class CustomerRelationshipUpdate(BaseModel):
    customer_name: Optional[str] = None
    contact_person: Optional[str] = None
    phone: Optional[str] = None
    email: Optional[str] = None
    social_media: Optional[Dict[str, str]] = None
    address: Optional[str] = None
    notes: Optional[str] = None

class CustomerRelationshipResponse(CustomerRelationshipBase):
    id: int
    created_at: datetime
    updated_at: Optional[datetime] = None
    
    class Config:
        from_attributes = True