Files
FastapiAdmin/backend/app/api/v1/module_example/demo/schema.py
T
zhangtao 1ee6400ac7 feat: 重构代码生成模块并添加示例模块
refactor(menu): 优化菜单删除逻辑,自动删除子菜单
refactor(dept): 优化部门删除逻辑,自动删除子部门
feat(redis): 添加分布式锁功能
feat(validator): 新增DateStr和TimeStr验证类型
refactor(logger): 简化日志配置,移除开发环境同步策略
refactor(console): 重命名并简化控制台输出函数
feat(example): 添加示例模块,包含完整数据类型示例
refactor(gencode): 重构代码生成工具,移除SQL生成
feat(ai): 添加MCP客户端和服务端实现
chore: 更新依赖版本
style: 清理无用代码和文件
2026-01-03 04:28:18 +08:00

102 lines
4.1 KiB
Python

# -*- coding: utf-8 -*-
from datetime import date, time, datetime
from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
from fastapi import Query
from app.core.base_schema import BaseSchema, UserBySchema
from app.core.validator import DateTimeStr, DateStr, TimeStr
class DemoCreateSchema(BaseModel):
"""新增模型"""
name: str = Field(..., description='名称')
status: str = Field(default="0", description="是否启用(0:启用 1:禁用)")
description: str | None = Field(default=None, description="描述")
a: int | None = Field(default=None, description="整数")
b: int | None = Field(default=None, description="大整数")
c: float | None = Field(default=None, description="浮点数")
d: bool = Field(default=True, description="布尔型")
e: date | None = Field(default=None, description="日期")
f: time | None = Field(default=None, description="时间")
g: datetime | None = Field(default=None, description="日期时间")
h: str | None = Field(default=None, description="长文本")
i: dict | None = Field(default=None, description="元数据(JSON格式)")
@field_validator('name')
@classmethod
def validate_name(cls, v: str) -> str:
"""验证名称字段的格式和内容"""
# 去除首尾空格
v = v.strip()
if not v:
raise ValueError('名称不能为空')
return v
@model_validator(mode='after')
def _after_validation(self):
"""
核心业务规则校验
"""
# 长度校验:名称最小长度
if len(self.name) < 2 or len(self.name) > 50:
raise ValueError('名称长度必须在2-50个字符之间')
# 格式校验:名称只能包含字母、数字、下划线和中划线
if not self.name.isalnum() and not all(c in '-_' for c in self.name):
raise ValueError('名称只能包含字母、数字、下划线和中划线')
if self.status not in ["0", "1"]:
raise ValueError('是否启用必须为0或1')
# 描述校验:描述最大长度
if self.description and len(self.description) > 255:
raise ValueError('描述长度不能超过255个字符')
return self
class DemoUpdateSchema(DemoCreateSchema):
"""更新模型"""
...
class DemoOutSchema(DemoCreateSchema, BaseSchema, UserBySchema):
"""响应模型"""
model_config = ConfigDict(from_attributes=True)
# 覆盖日期时间字段,使用可序列化的类型
e: DateStr | None = Field(default=None, description="日期")
f: TimeStr | None = Field(default=None, description="时间")
g: DateTimeStr | None = Field(default=None, description="日期时间")
class DemoQueryParam:
"""示例查询参数"""
def __init__(
self,
name: str | None = Query(None, description="名称"),
description: str | None = Query(None, description="描述"),
status: str | None = Query(None, description="是否启用"),
created_time: list[DateTimeStr] | None = Query(None, description="创建时间范围", examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"]),
updated_time: list[DateTimeStr] | None = Query(None, description="更新时间范围", examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"]),
created_id: int | None = Query(None, description="创建人"),
updated_id: int | None = Query(None, description="更新人")
) -> None:
# 模糊查询字段
self.name = ("like", name)
if description:
self.description = ("like", description)
# 精确查询字段
if status:
self.status = ("eq", status)
# 时间范围查询
if created_time and len(created_time) == 2:
self.created_time = ("between", (created_time[0], created_time[1]))
if updated_time and len(updated_time) == 2:
self.updated_time = ("between", (updated_time[0], updated_time[1]))
# 关联查询字段
if created_id:
self.created_id = ("eq", created_id)
if updated_id:
self.updated_id = ("eq", updated_id)