Files
FastapiAdmin/backend/app/api/v1/module_system/notice/schema.py
T
zhangtao 4e2b668d7b chore: 批量优化项目代码,修复多处细节问题
本次提交包含多项优化和修复:
1. 修复CRUD初始化参数传递、搜索参数处理逻辑
2. 更新环境配置中的大模型相关参数
3. 重构部分服务方法命名,统一代码风格
4. 新增多个枚举类型,补充模型关联关系和加载选项
5. 优化查询参数类实现,完善字段校验逻辑
6. 调整Pydantic模型字段注释和类型定义
7. 简化并移除冗余的CRUD方法实现
8. 新增超级管理员权限装饰器
9. 修复邮件日志模型的租户关联和字段定义
2026-06-20 23:37:58 +08:00

101 lines
3.3 KiB
Python

from dataclasses import dataclass
from fastapi import Query
from pydantic import (
BaseModel,
ConfigDict,
Field,
field_validator,
model_validator,
)
from app.common.enums import QueueEnum
from app.core.base_params import BaseQueryParam, TenantByQueryParam, UserByQueryParam
from app.core.base_schema import BaseSchema, TenantBySchema, UserBySchema
from app.utils.xss_util import sanitize_html
class NoticeCreateSchema(BaseModel):
"""公告通知创建模型"""
notice_title: str = Field(..., min_length=1, max_length=64, description="公告标题")
notice_type: str = Field(..., max_length=1, description="公告类型(1:通知 2:公告)")
notice_content: str | None = Field(default=None, max_length=65535, description="公告内容")
status: int = Field(default=0, ge=0, le=1, description="状态(0:启动 1:停用)")
description: str | None = Field(default=None, max_length=255, description="描述")
@field_validator("notice_type")
@classmethod
def _validate_notice_type(cls, value: str):
if value not in {"1", "2"}:
raise ValueError("公告类型仅支持 1(通知) 或 2(公告)")
return value
@field_validator("status")
@classmethod
def _validate_status(cls, value: int):
if value not in {0, 1}:
raise ValueError("状态仅支持 0(正常) 或 1(禁用)")
return value
@field_validator("notice_content")
@classmethod
def _sanitize_notice_content(cls, value: str | None) -> str | None:
if value is None:
return value
return sanitize_html(value)
@model_validator(mode="after")
def _validate_after(self):
if not self.notice_title.strip():
raise ValueError("公告标题不能为空")
if self.notice_content and not self.notice_content.strip():
raise ValueError("公告内容不能为空")
return self
class NoticeUpdateSchema(NoticeCreateSchema):
"""公告通知更新模型"""
class NoticeOutSchema(NoticeCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
"""公告通知响应模型"""
model_config = ConfigDict(from_attributes=True)
@dataclass
class NoticeQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
"""公告通知查询参数"""
def __init__(
self,
notice_title: str | None = Query(None, description="公告标题"),
notice_type: str | None = Query(None, description="公告类型"),
*args,
**kwargs,
) -> None:
super().__init__(*args, **kwargs)
if notice_title:
self.notice_title = (QueueEnum.like.value, notice_title)
if notice_type:
self.notice_type = (QueueEnum.eq.value, notice_type)
class PanelMessageItem(BaseModel):
"""面板-消息项"""
id: int = Field(..., description="消息ID")
title: str = Field(..., description="标题")
content: str = Field(..., description="内容")
time: str = Field(..., description="时间")
type: str = Field(..., description="类型")
class PanelDataOut(BaseModel):
"""通知面板聚合数据"""
notices: list[NoticeOutSchema] = Field(default_factory=list, description="通知列表")
messages: list[PanelMessageItem] = Field(default_factory=list, description="消息列表")
pendings: list[dict] = Field(default_factory=list, description="待办列表")