You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
from dataclasses import dataclass
|
|
|
|
from fastapi import Query
|
|
from pydantic import BaseModel, ConfigDict, Field, field_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
|
|
|
|
|
|
class RuleCreateSchema(BaseModel):
|
|
"""规则创建模型"""
|
|
|
|
name: str = Field(..., min_length=1, max_length=128, description="规则名称")
|
|
category_id: int | None = Field(default=None, ge=0, description="规则分类ID")
|
|
description: str | None = Field(default=None, max_length=2000, description="规则描述")
|
|
prompt_template: str = Field(..., min_length=1, description="提示词模板")
|
|
status: int = Field(default=0, ge=0, le=1, description="状态 0:启用 1:停用")
|
|
|
|
@field_validator("name")
|
|
@classmethod
|
|
def validate_name(cls, value: str) -> str:
|
|
value = value.strip()
|
|
if not value:
|
|
raise ValueError("规则名称不能为空")
|
|
return value
|
|
|
|
@field_validator("category_id")
|
|
@classmethod
|
|
def normalize_category_id(cls, value: int | None) -> int | None:
|
|
return None if value == 0 else value
|
|
|
|
@field_validator("prompt_template")
|
|
@classmethod
|
|
def validate_prompt_template(cls, value: str) -> str:
|
|
value = value.strip()
|
|
if not value:
|
|
raise ValueError("提示词模板不能为空")
|
|
return value
|
|
|
|
@field_validator("status")
|
|
@classmethod
|
|
def validate_status(cls, value: int) -> int:
|
|
if value not in {0, 1}:
|
|
raise ValueError("状态仅支持 0(启用) 或 1(停用)")
|
|
return value
|
|
|
|
|
|
class RuleUpdateSchema(RuleCreateSchema):
|
|
"""规则更新模型"""
|
|
|
|
|
|
class RuleOutSchema(RuleCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
|
|
"""规则响应模型"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
category_name_snapshot: str | None = Field(default=None, max_length=64, description="规则分类名称快照")
|
|
category_name: str | None = Field(default=None, max_length=64, description="规则分类名称")
|
|
|
|
|
|
@dataclass
|
|
class RuleQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
|
|
"""规则查询参数"""
|
|
|
|
name: str | None = Query(None, description="规则名称")
|
|
category_id: int | None = Query(None, ge=0, description="规则分类ID")
|
|
status: int | None = Query(None, ge=0, le=1, description="状态 0:启用 1:停用")
|
|
|
|
def __post_init__(self) -> None:
|
|
self.name = (QueueEnum.like.value, self.name)
|
|
if isinstance(self.category_id, int):
|
|
self.category_id = (QueueEnum.eq.value, None if self.category_id == 0 else self.category_id)
|
|
if isinstance(self.status, int):
|
|
self.status = (QueueEnum.eq.value, self.status)
|