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

class UserBase(BaseModel):
    """用户基础模型"""
    openid: str
    unionid: Optional[str] = None
    avatar: Optional[str] = None
    nickname: Optional[str] = None
    tryon_remain_count: Optional[int] = None
    referral_code: Optional[str] = None
    user_code: Optional[str] = None

class UserCreate(UserBase):
    """创建用户请求模型"""
    pass

class UserUpdate(BaseModel):
    """更新用户请求模型"""
    avatar: Optional[str] = None
    nickname: Optional[str] = None

class User(UserBase):
    """用户响应模型"""
    id: int
    create_time: datetime

    class Config:
        from_attributes = True