Build lightweight AI agent admin
This commit is contained in:
Codex
@@ -0,0 +1,295 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
User Schema - 用户数据验证模式
|
||||
"""
|
||||
from datetime import datetime, date
|
||||
from typing import Optional, List
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, Field, field_validator, EmailStr
|
||||
|
||||
from app.base_schema import CSTDatetime
|
||||
|
||||
|
||||
class UserBase(BaseModel):
|
||||
"""用户基础Schema"""
|
||||
username: str = Field(..., min_length=3, max_length=150, description="用户名")
|
||||
email: Optional[str] = Field(None, max_length=255, description="邮箱")
|
||||
mobile: Optional[str] = Field(None, max_length=11, description="手机号")
|
||||
avatar: Optional[str] = Field(None, max_length=36, description="头像UUID")
|
||||
name: Optional[str] = Field(None, max_length=64, description="真实姓名")
|
||||
gender: int = Field(default=0, ge=0, le=2, description="性别：0-未知, 1-男, 2-女")
|
||||
user_type: int = Field(default=1, ge=0, le=2, description="用户类型：0-系统用户, 1-普通用户, 2-外部用户")
|
||||
user_status: int = Field(default=1, ge=0, le=2, description="用户状态：0-禁用, 1-正常, 2-锁定")
|
||||
birthday: Optional[date] = Field(None, description="生日")
|
||||
city: Optional[str] = Field(None, max_length=100, description="所在城市")
|
||||
address: Optional[str] = Field(None, max_length=200, description="详细地址")
|
||||
bio: Optional[str] = Field(None, description="个人简介")
|
||||
is_active: bool = Field(default=True, description="是否激活")
|
||||
dept_id: Optional[str] = Field(None, description="所属部门ID")
|
||||
manager_id: Optional[str] = Field(None, description="直属上级ID")
|
||||
role_ids: Optional[List[str]] = Field(default=None, description="角色ID列表")
|
||||
sort: int = Field(default=0, description="排序")
|
||||
|
||||
@field_validator("mobile")
|
||||
@classmethod
|
||||
def validate_mobile(cls, v):
|
||||
"""验证手机号"""
|
||||
if v:
|
||||
if not v.isdigit():
|
||||
raise ValueError("手机号只能包含数字")
|
||||
if len(v) != 11:
|
||||
raise ValueError("手机号必须为11位")
|
||||
return v
|
||||
|
||||
@field_validator("user_status")
|
||||
@classmethod
|
||||
def validate_user_status(cls, v):
|
||||
"""验证用户状态"""
|
||||
if v not in [0, 1, 2]:
|
||||
raise ValueError("用户状态必须为 0(禁用)、1(正常) 或 2(锁定)")
|
||||
return v
|
||||
|
||||
@field_validator("user_type")
|
||||
@classmethod
|
||||
def validate_user_type(cls, v):
|
||||
"""验证用户类型"""
|
||||
if v not in [0, 1, 2]:
|
||||
raise ValueError("用户类型必须为 0(系统用户)、1(普通用户) 或 2(外部用户)")
|
||||
return v
|
||||
|
||||
@field_validator("gender")
|
||||
@classmethod
|
||||
def validate_gender(cls, v):
|
||||
"""验证性别"""
|
||||
if v not in [0, 1, 2]:
|
||||
raise ValueError("性别必须为 0(未知)、1(男) 或 2(女)")
|
||||
return v
|
||||
|
||||
|
||||
class UserCreate(UserBase):
|
||||
"""用户创建Schema"""
|
||||
# password: str = Field(..., min_length=6, max_length=20, description="密码")
|
||||
pass
|
||||
|
||||
|
||||
class UserUpdate(BaseModel):
|
||||
"""用户更新Schema - 所有字段可选"""
|
||||
username: Optional[str] = Field(None, min_length=3, max_length=150, description="用户名")
|
||||
email: Optional[str] = Field(None, max_length=255, description="邮箱")
|
||||
mobile: Optional[str] = Field(None, max_length=11, description="手机号")
|
||||
avatar: Optional[str] = Field(None, max_length=36, description="头像UUID")
|
||||
name: Optional[str] = Field(None, max_length=64, description="真实姓名")
|
||||
gender: Optional[int] = Field(None, ge=0, le=2, description="性别")
|
||||
user_type: Optional[int] = Field(None, ge=0, le=2, description="用户类型")
|
||||
user_status: Optional[int] = Field(None, ge=0, le=2, description="用户状态")
|
||||
birthday: Optional[date] = Field(None, description="生日")
|
||||
city: Optional[str] = Field(None, max_length=100, description="所在城市")
|
||||
address: Optional[str] = Field(None, max_length=200, description="详细地址")
|
||||
bio: Optional[str] = Field(None, description="个人简介")
|
||||
is_active: Optional[bool] = Field(None, description="是否激活")
|
||||
dept_id: Optional[str] = Field(None, description="所属部门ID")
|
||||
manager_id: Optional[str] = Field(None, description="直属上级ID")
|
||||
role_ids: Optional[List[str]] = Field(None, description="角色ID列表")
|
||||
sort: Optional[int] = Field(None, description="排序")
|
||||
|
||||
@field_validator("mobile")
|
||||
@classmethod
|
||||
def validate_mobile(cls, v):
|
||||
"""验证手机号"""
|
||||
if v is not None and v:
|
||||
if not v.isdigit():
|
||||
raise ValueError("手机号只能包含数字")
|
||||
if len(v) != 11:
|
||||
raise ValueError("手机号必须为11位")
|
||||
return v
|
||||
|
||||
@field_validator("user_status")
|
||||
@classmethod
|
||||
def validate_user_status(cls, v):
|
||||
"""验证用户状态"""
|
||||
if v is not None and v not in [0, 1, 2]:
|
||||
raise ValueError("用户状态必须为 0(禁用)、1(正常) 或 2(锁定)")
|
||||
return v
|
||||
|
||||
@field_validator("user_type")
|
||||
@classmethod
|
||||
def validate_user_type(cls, v):
|
||||
"""验证用户类型"""
|
||||
if v is not None and v not in [0, 1, 2]:
|
||||
raise ValueError("用户类型必须为 0(系统用户)、1(普通用户) 或 2(外部用户)")
|
||||
return v
|
||||
|
||||
@field_validator("gender")
|
||||
@classmethod
|
||||
def validate_gender(cls, v):
|
||||
"""验证性别"""
|
||||
if v is not None and v not in [0, 1, 2]:
|
||||
raise ValueError("性别必须为 0(未知)、1(男) 或 2(女)")
|
||||
return v
|
||||
|
||||
|
||||
class UserResponse(BaseModel):
|
||||
"""用户响应Schema"""
|
||||
id: str
|
||||
username: str
|
||||
email: Optional[str] = None
|
||||
mobile: Optional[str] = None
|
||||
avatar: Optional[str] = None
|
||||
name: Optional[str] = None
|
||||
gender: int = 0
|
||||
gender_display: Optional[str] = None
|
||||
user_type: int = 1
|
||||
user_type_display: Optional[str] = None
|
||||
user_status: int = 1
|
||||
user_status_display: Optional[str] = None
|
||||
birthday: Optional[date] = None
|
||||
city: Optional[str] = None
|
||||
address: Optional[str] = None
|
||||
bio: Optional[str] = None
|
||||
is_superuser: bool = False
|
||||
is_active: bool = True
|
||||
dept_id: Optional[str] = None
|
||||
post_id: Optional[str] = None
|
||||
post_name: Optional[str] = None
|
||||
manager_id: Optional[str] = None
|
||||
role_ids: Optional[List[str]] = None
|
||||
last_login: Optional[CSTDatetime] = None
|
||||
last_login_ip: Optional[str] = None
|
||||
last_login_type: Optional[str] = None
|
||||
sort: int = 0
|
||||
is_deleted: bool = False
|
||||
sys_create_datetime: Optional[CSTDatetime] = None
|
||||
sys_update_datetime: Optional[CSTDatetime] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class UserSimple(BaseModel):
|
||||
"""用户简单输出（用于选择器）"""
|
||||
id: str
|
||||
name: Optional[str] = None
|
||||
username: str
|
||||
avatar: Optional[str] = None
|
||||
email: Optional[str] = None
|
||||
mobile: Optional[str] = None
|
||||
dept_name: Optional[str] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class OrgChartNode(BaseModel):
|
||||
"""组织架构节点"""
|
||||
id: str
|
||||
name: Optional[str] = None
|
||||
username: str
|
||||
avatar: Optional[str] = None
|
||||
dept_name: Optional[str] = None
|
||||
post_name: Optional[str] = None
|
||||
subordinate_count: int = 0
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class OrgChartChainNode(BaseModel):
|
||||
"""组织架构汇报链节点（嵌套结构）"""
|
||||
id: str
|
||||
name: Optional[str] = None
|
||||
username: str
|
||||
avatar: Optional[str] = None
|
||||
dept_name: Optional[str] = None
|
||||
post_name: Optional[str] = None
|
||||
subordinate_count: int = 0
|
||||
children: List["OrgChartChainNode"] = []
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class UserPasswordResetIn(BaseModel):
|
||||
"""重置密码输入"""
|
||||
old_password: str = Field(..., description="旧密码")
|
||||
new_password: str = Field(..., min_length=6, max_length=20, description="新密码")
|
||||
confirm_password: str = Field(..., description="确认新密码")
|
||||
|
||||
@field_validator("confirm_password")
|
||||
@classmethod
|
||||
def validate_confirm_password(cls, v, info):
|
||||
"""验证确认密码"""
|
||||
if info.data.get("new_password") and v != info.data.get("new_password"):
|
||||
raise ValueError("两次输入的密码不一致")
|
||||
return v
|
||||
|
||||
|
||||
class UserPasswordSetIn(BaseModel):
|
||||
"""管理员设置密码输入"""
|
||||
new_password: str = Field(..., min_length=6, max_length=20, description="新密码")
|
||||
|
||||
|
||||
class UserBatchDeleteIn(BaseModel):
|
||||
"""批量删除用户输入"""
|
||||
ids: List[str] = Field(..., description="要删除的用户ID列表")
|
||||
|
||||
|
||||
class UserBatchDeleteOut(BaseModel):
|
||||
"""批量删除用户输出"""
|
||||
success_count: int = Field(..., description="删除的记录数")
|
||||
fail_count: int = Field(..., description="删除失败的ID列表")
|
||||
|
||||
|
||||
class UserBatchUpdateStatusIn(BaseModel):
|
||||
"""批量更新用户状态输入"""
|
||||
ids: List[str] = Field(..., description="用户ID列表")
|
||||
user_status: int = Field(..., ge=0, le=2, description="用户状态：0-禁用，1-正常，2-锁定")
|
||||
|
||||
@field_validator("user_status")
|
||||
@classmethod
|
||||
def validate_user_status(cls, v):
|
||||
"""验证用户状态"""
|
||||
if v not in [0, 1, 2]:
|
||||
raise ValueError("用户状态必须为 0(禁用)、1(正常) 或 2(锁定)")
|
||||
return v
|
||||
|
||||
|
||||
class UserBatchUpdateStatusOut(BaseModel):
|
||||
"""批量更新用户状态输出"""
|
||||
count: int = Field(..., description="更新的记录数")
|
||||
|
||||
|
||||
class UserProfileUpdateIn(BaseModel):
|
||||
"""用户个人信息更新输入"""
|
||||
name: Optional[str] = Field(None, max_length=64, description="真实姓名")
|
||||
email: Optional[str] = Field(None, max_length=255, description="邮箱")
|
||||
mobile: Optional[str] = Field(None, max_length=11, description="手机号")
|
||||
avatar: Optional[str] = Field(None, max_length=36, description="头像UUID")
|
||||
gender: Optional[int] = Field(None, ge=0, le=2, description="性别")
|
||||
birthday: Optional[date] = Field(None, description="生日")
|
||||
city: Optional[str] = Field(None, max_length=100, description="所在城市")
|
||||
address: Optional[str] = Field(None, max_length=200, description="详细地址")
|
||||
bio: Optional[str] = Field(None, description="个人简介")
|
||||
|
||||
@field_validator("birthday", mode="before")
|
||||
@classmethod
|
||||
def validate_birthday(cls, v):
|
||||
"""验证生日，将空字符串转换为None"""
|
||||
if v == "" or v is None:
|
||||
return None
|
||||
return v
|
||||
|
||||
@field_validator("mobile")
|
||||
@classmethod
|
||||
def validate_mobile(cls, v):
|
||||
"""验证手机号"""
|
||||
if v is not None and v:
|
||||
if not v.isdigit():
|
||||
raise ValueError("手机号只能包含数字")
|
||||
if len(v) != 11:
|
||||
raise ValueError("手机号必须为11位")
|
||||
return v
|
||||
|
||||
@field_validator("gender")
|
||||
@classmethod
|
||||
def validate_gender(cls, v):
|
||||
"""验证性别"""
|
||||
if v is not None and v not in [0, 1, 2]:
|
||||
raise ValueError("性别必须为 0(未知)、1(男) 或 2(女)")
|
||||
return v
|
||||
Reference in New Issue
Block a user