Add 'backend/' from commit '48a644fb354d6c6efcbd12bc1b4a2cb83137b68e'
git-subtree-dir: backend git-subtree-mainline:545699d16f
git-subtree-split:48a644fb35
This commit is contained in:
22
backend/schemas/auth.py
Normal file
22
backend/schemas/auth.py
Normal file
@@ -0,0 +1,22 @@
|
||||
from pydantic import BaseModel
|
||||
|
||||
class TokenResponse(BaseModel):
|
||||
access_token: str
|
||||
refresh_token: str
|
||||
token_type: str
|
||||
access_token_exp: int
|
||||
refresh_token_exp: int
|
||||
|
||||
class TokenPayload(BaseModel):
|
||||
id: int
|
||||
username: str
|
||||
role: str
|
||||
exp: int
|
||||
token_type: str
|
||||
|
||||
class LoginRequest(BaseModel):
|
||||
username: str
|
||||
password: str
|
||||
|
||||
class RefreshTokenRequest(BaseModel):
|
||||
refresh_token: str
|
43
backend/schemas/user.py
Normal file
43
backend/schemas/user.py
Normal file
@@ -0,0 +1,43 @@
|
||||
from datetime import datetime
|
||||
from enum import Enum
|
||||
from pydantic import BaseModel, Field, root_validator
|
||||
from typing import Optional
|
||||
|
||||
# 用户角色枚举
|
||||
class UserRole(str, Enum):
|
||||
SYSTEM_ADMIN = "system_admin"
|
||||
ADMIN = "admin"
|
||||
USER = "user"
|
||||
|
||||
# 基础用户模型
|
||||
class UserBase(BaseModel):
|
||||
username: str = Field(..., max_length=50, description="用户名")
|
||||
role: UserRole = Field(default=UserRole.USER, description="用户角色")
|
||||
description: Optional[str] = Field(None, max_length=255, description="用户描述")
|
||||
|
||||
# 用户创建模型
|
||||
class UserCreate(UserBase):
|
||||
password: str = Field(..., min_length=6, max_length=255, description="用户密码")
|
||||
|
||||
|
||||
# 用户更新模型
|
||||
class UserUpdate(BaseModel):
|
||||
username: Optional[str] = Field(None, max_length=50, description="用户名")
|
||||
role: Optional[UserRole] = Field(None, description="用户角色")
|
||||
description: Optional[str] = Field(None, max_length=255, description="用户描述")
|
||||
|
||||
# 可选:确保至少更新一个字段
|
||||
@root_validator
|
||||
def validate_at_least_one_field(cls, values):
|
||||
if not any(values.values()):
|
||||
raise ValueError("至少需要更新一个字段")
|
||||
return values
|
||||
|
||||
# 用户响应模型
|
||||
class UserResponse(UserBase):
|
||||
id: int = Field(..., description="用户ID")
|
||||
created_at: datetime = Field(..., description="创建时间")
|
||||
updated_at: datetime = Field(..., description="更新时间")
|
||||
|
||||
class Config:
|
||||
orm_mode = True # 允许从 ORM 对象加载数据
|
Reference in New Issue
Block a user