mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-23 05:10:57 +00:00
zhangtao
4e2b668d7b
本次提交包含多项优化和修复： 1. 修复CRUD初始化参数传递、搜索参数处理逻辑 2. 更新环境配置中的大模型相关参数 3. 重构部分服务方法命名，统一代码风格 4. 新增多个枚举类型，补充模型关联关系和加载选项 5. 优化查询参数类实现，完善字段校验逻辑 6. 调整Pydantic模型字段注释和类型定义 7. 简化并移除冗余的CRUD方法实现 8. 新增超级管理员权限装饰器 9. 修复邮件日志模型的租户关联和字段定义
337 lines
13 KiB
Python
337 lines
13 KiB
Python
from dataclasses import dataclass
|
||
from urllib.parse import urlparse
|
||
|
||
from fastapi import Query
|
||
from pydantic import (
|
||
BaseModel,
|
||
ConfigDict,
|
||
EmailStr,
|
||
Field,
|
||
field_validator,
|
||
model_validator,
|
||
)
|
||
|
||
from app.api.v1.module_platform.menu.schema import MenuOutSchema
|
||
from app.api.v1.module_system.role.schema import RoleOutSchema
|
||
from app.common.enums import QueueEnum
|
||
from app.core.base_params import BaseQueryParam, TenantByQueryParam, UserByQueryParam
|
||
from app.core.base_schema import BaseSchema, CommonSchema, TenantBySchema, UserBySchema
|
||
from app.core.validator import email_validator, mobile_validator
|
||
|
||
|
||
class CurrentUserUpdateSchema(BaseModel):
|
||
"""基础用户信息"""
|
||
|
||
name: str | None = Field(default=None, max_length=32, description="名称")
|
||
mobile: str | None = Field(default=None, max_length=11, description="手机号")
|
||
email: EmailStr | None = Field(default=None, description="邮箱")
|
||
gender: str | None = Field(default=None, max_length=1, description="性别(0:男 1:女 2:未知)")
|
||
avatar: str | None = Field(default=None, max_length=255, description="头像")
|
||
|
||
@field_validator("mobile")
|
||
@classmethod
|
||
def validate_mobile(cls, value: str | None):
|
||
"""校验手机号格式"""
|
||
return mobile_validator(value)
|
||
|
||
@field_validator("email")
|
||
@classmethod
|
||
def validate_email(cls, value: str | None):
|
||
"""校验邮箱格式"""
|
||
if not value:
|
||
return value
|
||
return email_validator(value)
|
||
|
||
@field_validator("gender")
|
||
@classmethod
|
||
def validate_gender(cls, value: str | None):
|
||
"""校验性别：仅支持 0(男)、1(女)、2(未知)"""
|
||
if value and value not in {"0", "1", "2"}:
|
||
raise ValueError("性别仅支持 0(男)、1(女)、2(未知)")
|
||
return value
|
||
|
||
@field_validator("avatar")
|
||
@classmethod
|
||
def validate_avatar(cls, value: str | None):
|
||
"""校验头像地址为合法的 HTTP/HTTPS URL"""
|
||
if not value:
|
||
return value
|
||
parsed = urlparse(value)
|
||
if parsed.scheme in ("http", "https") and parsed.netloc:
|
||
return value
|
||
raise ValueError("头像地址需为有效的 HTTP/HTTPS URL")
|
||
|
||
@model_validator(mode="after")
|
||
def check_model(self):
|
||
"""校验基础用户信息长度约束"""
|
||
if self.name and len(self.name) > 32:
|
||
raise ValueError("名称长度不能超过 32 个字符")
|
||
return self
|
||
|
||
|
||
class UserRegisterSchema(BaseModel):
|
||
"""注册"""
|
||
|
||
name: str | None = Field(default=None, max_length=32, description="姓名")
|
||
mobile: str | None = Field(default=None, max_length=11, description="手机号")
|
||
username: str = Field(..., min_length=3, max_length=32, description="账号")
|
||
password: str = Field(..., min_length=6, max_length=128, description="密码")
|
||
role_ids: list[int] | None = Field(default=[1], description="角色ID列表")
|
||
created_id: int | None = Field(default=1, description="创建人ID")
|
||
description: str | None = Field(default=None, max_length=255, description="备注")
|
||
|
||
@field_validator("mobile")
|
||
@classmethod
|
||
def validate_mobile(cls, value: str | None):
|
||
"""校验手机号格式"""
|
||
return mobile_validator(value)
|
||
|
||
@field_validator("username")
|
||
@classmethod
|
||
def validate_username(cls, value: str):
|
||
"""校验账号：字母开头，3-32 位，仅含字母/数字/_ . -"""
|
||
v = value.strip()
|
||
if not v:
|
||
raise ValueError("账号不能为空")
|
||
import re
|
||
|
||
if not re.match(r"^[A-Za-z][A-Za-z0-9_.-]{2,31}$", v):
|
||
raise ValueError("账号需以字母开头，3-32 位，仅允许字母、数字、_ . -")
|
||
return v
|
||
|
||
@field_validator("password")
|
||
@classmethod
|
||
def validate_password(cls, value: str):
|
||
"""校验密码：6-128 位"""
|
||
if len(value) < 6:
|
||
raise ValueError("密码长度不能少于 6 位")
|
||
if len(value) > 128:
|
||
raise ValueError("密码长度不能超过 128 位")
|
||
return value
|
||
|
||
@model_validator(mode="after")
|
||
def check_model(self):
|
||
"""校验注册信息长度约束"""
|
||
if self.name and len(self.name) > 32:
|
||
raise ValueError("姓名长度不能超过 32 个字符")
|
||
if self.username and len(self.username) > 32:
|
||
raise ValueError("账号长度不能超过 32 个字符")
|
||
if self.description and len(self.description) > 255:
|
||
raise ValueError("备注长度不能超过 255 个字符")
|
||
return self
|
||
|
||
|
||
class UserForgetPasswordSchema(BaseModel):
|
||
"""忘记密码"""
|
||
|
||
username: str = Field(..., min_length=3, max_length=32, description="用户名")
|
||
new_password: str = Field(..., min_length=6, max_length=128, description="新密码")
|
||
mobile: str | None = Field(default=None, max_length=11, description="手机号")
|
||
|
||
@field_validator("username")
|
||
@classmethod
|
||
def validate_username(cls, value: str):
|
||
"""校验账号：字母开头，3-32 位"""
|
||
v = value.strip()
|
||
if not v:
|
||
raise ValueError("账号不能为空")
|
||
import re
|
||
|
||
if not re.match(r"^[A-Za-z][A-Za-z0-9_.-]{2,31}$", v):
|
||
raise ValueError("账号需以字母开头，3-32 位，仅允许字母、数字、_ . -")
|
||
return v
|
||
|
||
@field_validator("new_password")
|
||
@classmethod
|
||
def validate_new_password(cls, value: str):
|
||
"""校验密码：6-128 位"""
|
||
if len(value) < 6:
|
||
raise ValueError("密码长度不能少于 6 位")
|
||
if len(value) > 128:
|
||
raise ValueError("密码长度不能超过 128 位")
|
||
return value
|
||
|
||
@field_validator("mobile")
|
||
@classmethod
|
||
def validate_mobile(cls, value: str | None):
|
||
"""校验手机号格式"""
|
||
return mobile_validator(value)
|
||
|
||
|
||
class UserChangePasswordSchema(BaseModel):
|
||
"""修改密码"""
|
||
|
||
old_password: str = Field(..., min_length=6, max_length=128, description="旧密码")
|
||
new_password: str = Field(..., min_length=6, max_length=128, description="新密码")
|
||
|
||
@field_validator("new_password")
|
||
@classmethod
|
||
def validate_new_password(cls, value: str):
|
||
"""校验新密码：6-128 位"""
|
||
if len(value) < 6:
|
||
raise ValueError("新密码长度不能少于 6 位")
|
||
if len(value) > 128:
|
||
raise ValueError("新密码长度不能超过 128 位")
|
||
return value
|
||
|
||
|
||
class ResetPasswordSchema(BaseModel):
|
||
"""重置密码"""
|
||
|
||
id: int = Field(default=0, description="主键ID（已弃用，由路径参数传入）")
|
||
password: str = Field(..., min_length=6, max_length=128, description="新密码")
|
||
|
||
@field_validator("password")
|
||
@classmethod
|
||
def validate_password(cls, value: str):
|
||
"""校验新密码：6-128 位"""
|
||
if len(value) < 6:
|
||
raise ValueError("新密码长度不能少于 6 位")
|
||
if len(value) > 128:
|
||
raise ValueError("新密码长度不能超过 128 位")
|
||
return value
|
||
|
||
|
||
class UserCreateSchema(CurrentUserUpdateSchema):
|
||
"""
|
||
新增用户
|
||
"""
|
||
|
||
username: str | None = Field(default=None, max_length=32, description="用户名")
|
||
password: str | None = Field(default=None, min_length=6, max_length=128, description="密码")
|
||
status: int = Field(default=0, ge=0, le=1, description="状态(0:启动 1:停用)")
|
||
description: str | None = Field(default=None, max_length=255, description="备注")
|
||
is_superuser: bool | None = Field(default=False, description="是否超管")
|
||
dept_id: int | None = Field(default=None, description="部门ID")
|
||
tenant_id: int | None = Field(default=None, description="租户ID，仅平台管理员创建时可指定")
|
||
role_ids: list[int] | None = Field(default=[], description="角色ID列表")
|
||
position_ids: list[int] | None = Field(default=[], description="岗位ID列表")
|
||
|
||
@field_validator("status")
|
||
@classmethod
|
||
def validate_status(cls, value: int):
|
||
"""校验状态：仅支持 0(正常)、1(禁用)"""
|
||
if value not in {0, 1}:
|
||
raise ValueError("状态仅支持 0(正常) 或 1(禁用)")
|
||
return value
|
||
|
||
@field_validator("username")
|
||
@classmethod
|
||
def validate_username(cls, value: str | None):
|
||
"""校验账号：字母开头，2-32 位"""
|
||
if not value:
|
||
return value
|
||
v = value.strip()
|
||
import re
|
||
|
||
if not re.match(r"^[A-Za-z][A-Za-z0-9_.-]{1,31}$", v):
|
||
raise ValueError("账号需以字母开头，2-32 位，仅允许字母、数字、_ . -")
|
||
return v
|
||
|
||
@field_validator("password")
|
||
@classmethod
|
||
def validate_password(cls, value: str | None):
|
||
"""校验密码：6-128 位"""
|
||
if value and len(value) < 6:
|
||
raise ValueError("密码长度不能少于 6 位")
|
||
if value and len(value) > 128:
|
||
raise ValueError("密码长度不能超过 128 位")
|
||
return value
|
||
|
||
|
||
class UserUpdateSchema(CurrentUserUpdateSchema):
|
||
"""更新"""
|
||
|
||
model_config = ConfigDict(from_attributes=True)
|
||
|
||
username: str | None = Field(default=None, max_length=32, description="用户名")
|
||
status: int | None = Field(default=None, ge=0, le=1, description="状态(0:启动 1:停用)")
|
||
description: str | None = Field(default=None, max_length=255, description="备注")
|
||
dept_id: int | None = Field(default=None, description="部门ID")
|
||
role_ids: list[int] | None = Field(default=[], description="角色ID列表")
|
||
position_ids: list[int] | None = Field(default=[], description="岗位ID列表")
|
||
|
||
@field_validator("status")
|
||
@classmethod
|
||
def validate_status(cls, value: int | None):
|
||
"""校验状态：仅支持 0(正常)、1(禁用)"""
|
||
if value is not None and value not in {0, 1}:
|
||
raise ValueError("状态仅支持 0(正常) 或 1(禁用)")
|
||
return value
|
||
|
||
@field_validator("username")
|
||
@classmethod
|
||
def validate_username(cls, value: str | None):
|
||
"""校验账号：字母开头，2-32 位"""
|
||
if not value:
|
||
return value
|
||
v = value.strip()
|
||
import re
|
||
|
||
if not re.match(r"^[A-Za-z][A-Za-z0-9_.-]{1,31}$", v):
|
||
raise ValueError("账号需以字母开头，2-32 位，仅允许字母、数字、_ . -")
|
||
return v
|
||
|
||
|
||
class UserOutSchema(UserUpdateSchema, BaseSchema, UserBySchema, TenantBySchema):
|
||
"""响应"""
|
||
|
||
model_config = ConfigDict(arbitrary_types_allowed=True, from_attributes=True)
|
||
|
||
username: str | None = Field(default=None, max_length=32, description="用户名")
|
||
|
||
tenant_id: int | None = Field(
|
||
default=None,
|
||
exclude=True,
|
||
description="创建入参使用；列表/详情出参见 tenant",
|
||
)
|
||
gitee_login: str | None = Field(default=None, max_length=32, description="Gitee登录")
|
||
github_login: str | None = Field(default=None, max_length=32, description="Github登录")
|
||
wx_login: str | None = Field(default=None, max_length=32, description="微信登录")
|
||
qq_login: str | None = Field(default=None, max_length=32, description="QQ登录")
|
||
dept_name: str | None = Field(default=None, description="部门名称")
|
||
dept: CommonSchema | None = Field(default=None, description="部门")
|
||
positions: list[CommonSchema] | None = Field(default=[], description="岗位")
|
||
roles: list[RoleOutSchema] | None = Field(default=[], description="角色")
|
||
menus: list[MenuOutSchema] | None = Field(default=[], description="菜单")
|
||
|
||
|
||
@dataclass
|
||
class UserQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
|
||
"""
|
||
用户管理查询参数（继承标准 Mixin）
|
||
|
||
支持：
|
||
- 时间范围（BaseQueryParam）
|
||
- 创建人/更新人筛选（UserByQueryParam）
|
||
- 租户筛选（TenantByQueryParam）
|
||
- 业务字段：用户名、名称、手机号、邮箱、部门、状态
|
||
"""
|
||
|
||
def __init__(
|
||
self,
|
||
username: str | None = Query(None, description="用户名"),
|
||
name: str | None = Query(None, description="名称"),
|
||
mobile: str | None = Query(None, description="手机号", pattern=r"^1[3-9]\d{9}$"),
|
||
email: str | None = Query(
|
||
None,
|
||
description="邮箱",
|
||
pattern=r"^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$",
|
||
),
|
||
dept_id: int | None = Query(None, description="部门ID"),
|
||
status: str | None = Query(None, description="是否可用"),
|
||
*args,
|
||
**kwargs,
|
||
) -> None:
|
||
super().__init__(*args, **kwargs)
|
||
self.username = (QueueEnum.like.value, username)
|
||
self.name = (QueueEnum.like.value, name)
|
||
if mobile:
|
||
self.mobile = (QueueEnum.like.value, mobile)
|
||
if email:
|
||
self.email = (QueueEnum.like.value, email)
|
||
if dept_id:
|
||
self.dept_id = (QueueEnum.eq.value, dept_id)
|
||
if status:
|
||
self.status = (QueueEnum.eq.value, status)
|