51 lines
1001 B
Python
51 lines
1001 B
Python
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
# Shared properties
|
|
class UserBase(BaseModel):
|
|
# 用户名
|
|
name: str
|
|
# 邮箱，未启用
|
|
email: Optional[str] = None
|
|
# 状态
|
|
is_active: Optional[bool] = True
|
|
# 超级管理员
|
|
is_superuser: bool = False
|
|
# 头像
|
|
avatar: Optional[str] = None
|
|
# 是否开启二次验证
|
|
is_otp: Optional[bool] = False
|
|
|
|
|
|
# Properties to receive via API on creation
|
|
class UserCreate(UserBase):
|
|
name: str
|
|
email: Optional[str] = None
|
|
password: Optional[str] = None
|
|
|
|
|
|
# Properties to receive via API on update
|
|
class UserUpdate(UserBase):
|
|
name: str
|
|
password: Optional[str] = None
|
|
|
|
|
|
class UserInDBBase(UserBase):
|
|
id: Optional[int] = None
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
|
|
# Additional properties to return via API
|
|
class User(UserInDBBase):
|
|
name: str
|
|
email: Optional[str] = None
|
|
|
|
|
|
# Additional properties stored in DB
|
|
class UserInDB(UserInDBBase):
|
|
hashed_password: str
|