from datetime import datetime

from pydantic import BaseModel
from typing import Optional


class UserCreate(BaseModel):
    name: str
    avatar: Optional[str] = None


class UserUpdate(BaseModel):
    name: Optional[str] = None
    avatar: Optional[str] = None


class UserResponse(BaseModel):
    id: int
    name: str
    avatar: Optional[str] = None
    create_time: datetime
    update_time: datetime

    model_config = {
        "from_attributes": True,
        "json_encoders": {
            datetime: lambda dt: dt.strftime("%Y-%m-%d %H:%M:%S")
        }
    }