74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
|
# app/models/schemas/users.py
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from pydantic import BaseModel, EmailStr, Field
|
|||
|
|
|
|||
|
|
from app.models.domain.users import User
|
|||
|
|
from app.models.schemas.rwschema import RWSchema
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ===============================
|
|||
|
|
# 旧有登录/创建（兼容保留）
|
|||
|
|
# ===============================
|
|||
|
|
class UserInLogin(RWSchema):
|
|||
|
|
email: EmailStr
|
|||
|
|
password: str
|
|||
|
|
|
|||
|
|
|
|||
|
|
class UserInCreate(UserInLogin):
|
|||
|
|
username: str
|
|||
|
|
|
|||
|
|
|
|||
|
|
class UserInUpdate(BaseModel):
|
|||
|
|
username: Optional[str] = None
|
|||
|
|
email: Optional[EmailStr] = None
|
|||
|
|
password: Optional[str] = None
|
|||
|
|
bio: Optional[str] = None
|
|||
|
|
image: Optional[str] = None
|
|||
|
|
phone: Optional[str] = None
|
|||
|
|
user_type: Optional[str] = None
|
|||
|
|
company_name: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ===============================
|
|||
|
|
# 新增：邮箱验证码注册/登录/Token
|
|||
|
|
# ===============================
|
|||
|
|
class SendEmailCodeIn(RWSchema):
|
|||
|
|
"""发送邮箱验证码"""
|
|||
|
|
email: EmailStr
|
|||
|
|
# 可选场景：register/reset/login，不传默认 register
|
|||
|
|
scene: Optional[str] = "register"
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RegisterWithEmailIn(RWSchema):
|
|||
|
|
"""邮箱注册：邮箱 + 密码 + 确认密码 + 验证码"""
|
|||
|
|
email: EmailStr
|
|||
|
|
password: str = Field(min_length=6, max_length=64)
|
|||
|
|
confirm_password: str = Field(min_length=6, max_length=64)
|
|||
|
|
code: str = Field(min_length=4, max_length=8)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LoginWithPasswordIn(RWSchema):
|
|||
|
|
"""邮箱 + 密码登录"""
|
|||
|
|
email: EmailStr
|
|||
|
|
password: str
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TokenOut(RWSchema):
|
|||
|
|
token: str
|
|||
|
|
token_type: str = "Token"
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ===============================
|
|||
|
|
# 响应模型（带 email_verified）
|
|||
|
|
# ===============================
|
|||
|
|
class UserWithToken(User):
|
|||
|
|
token: str
|
|||
|
|
phone: Optional[str] = None
|
|||
|
|
user_type: Optional[str] = None
|
|||
|
|
company_name: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class UserInResponse(RWSchema):
|
|||
|
|
user: UserWithToken
|