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.
70 lines
2.6 KiB
Python
70 lines
2.6 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, CommonSchema, TenantBySchema, UserBySchema
|
|
|
|
|
|
class ChapterCreateSchema(BaseModel):
|
|
"""章节创建模型。"""
|
|
|
|
template_id: int = Field(..., ge=1, description="能源报告模板ID")
|
|
chapter_name: str = Field(..., min_length=1, max_length=128, description="章节名称")
|
|
chapter_order: int = Field(default=999, ge=0, description="章节排序")
|
|
rule_id: int = Field(..., ge=1, description="关联规则ID")
|
|
description: str | None = Field(default=None, max_length=500, description="描述")
|
|
|
|
@field_validator("chapter_name")
|
|
@classmethod
|
|
def validate_chapter_name(cls, value: str) -> str:
|
|
value = value.strip()
|
|
if not value:
|
|
raise ValueError("章节名称不能为空")
|
|
return value
|
|
|
|
|
|
class ChapterUpdateSchema(BaseModel):
|
|
"""章节更新模型。"""
|
|
|
|
chapter_name: str = Field(..., min_length=1, max_length=128, description="章节名称")
|
|
chapter_order: int = Field(default=999, ge=0, description="章节排序")
|
|
rule_id: int = Field(..., ge=1, description="关联规则ID")
|
|
description: str | None = Field(default=None, max_length=500, description="描述")
|
|
|
|
@field_validator("chapter_name")
|
|
@classmethod
|
|
def validate_chapter_name(cls, value: str) -> str:
|
|
value = value.strip()
|
|
if not value:
|
|
raise ValueError("章节名称不能为空")
|
|
return value
|
|
|
|
|
|
class ChapterOutSchema(ChapterCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
|
|
"""章节响应模型。"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
rule_name: str | None = Field(default=None, description="关联规则名称")
|
|
|
|
|
|
@dataclass
|
|
class ChapterQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
|
|
"""章节查询参数。"""
|
|
|
|
template_id: int | None = Query(None, ge=1, description="模板ID")
|
|
chapter_name: str | None = Query(None, description="章节名称")
|
|
rule_id: int | None = Query(None, ge=1, description="规则ID")
|
|
|
|
def __post_init__(self) -> None:
|
|
if isinstance(self.template_id, int):
|
|
self.template_id = (QueueEnum.eq.value, self.template_id)
|
|
if self.chapter_name:
|
|
self.chapter_name = (QueueEnum.like.value, self.chapter_name)
|
|
if isinstance(self.rule_id, int):
|
|
self.rule_id = (QueueEnum.eq.value, self.rule_id)
|
|
|