Codex
296 lines
11 KiB
Python
296 lines
11 KiB
Python
#!/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
|