from pydantic import BaseModel
from datetime import datetime
from typing import Optional


# 新增顾问 请求体
class ConsultantCreate(BaseModel):
    consultant_no: str
    consultant_name: str
    phone: Optional[str] = None
    email: Optional[str] = None
    remark: Optional[str] = None


# 修改顾问 请求体
class ConsultantUpdate(BaseModel):
    consultant_name: Optional[str] = None
    phone: Optional[str] = None
    email: Optional[str] = None
    remark: Optional[str] = None


# 返回模型
class ConsultantResponse(BaseModel):
    id: int
    consultant_no: str
    consultant_name: str
    phone: Optional[str]
    email: Optional[str]
    remark: Optional[str]
    created_at: datetime
    updated_at: datetime