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.
73 lines
3.0 KiB
Python
73 lines
3.0 KiB
Python
from dataclasses import dataclass
|
|
|
|
from fastapi import Query
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
from app.api.v1.module_common.file.schema import FileOutSchema
|
|
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 EnergyReportTemplateCreateSchema(BaseModel):
|
|
"""能源报告模板创建模型。"""
|
|
|
|
template_name: str = Field(..., min_length=1, max_length=128, description="模板名称")
|
|
template_type: str = Field(..., min_length=1, max_length=64, description="模板类型")
|
|
version: str | None = Field(default=None, max_length=32, description="模板版本")
|
|
file_id: int = Field(..., ge=1, description="文件记录ID")
|
|
status: int = Field(default=0, ge=0, le=1, description="状态 0:启用 1:停用")
|
|
order: int = Field(default=999, ge=0, description="显示排序")
|
|
description: str | None = Field(default=None, max_length=500, description="描述")
|
|
|
|
@field_validator("template_name", "template_type")
|
|
@classmethod
|
|
def validate_required_text(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 EnergyReportTemplateUpdateSchema(EnergyReportTemplateCreateSchema):
|
|
"""能源报告模板更新模型。"""
|
|
|
|
|
|
class EnergyReportTemplateOutSchema(EnergyReportTemplateCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
|
|
"""能源报告模板响应模型。"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
file: FileOutSchema | None = Field(default=None, description="文件记录")
|
|
file_url: str | None = Field(default=None, description="文件访问地址")
|
|
original_name: str | None = Field(default=None, description="原始文件名")
|
|
object_name: str | None = Field(default=None, description="MinIO对象名称")
|
|
|
|
|
|
@dataclass
|
|
class EnergyReportTemplateQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
|
|
"""能源报告模板查询参数。"""
|
|
|
|
template_name: str | None = Query(None, description="模板名称")
|
|
template_type: str | None = Query(None, description="模板类型")
|
|
version: str | None = Query(None, description="模板版本")
|
|
status: int | None = Query(None, ge=0, le=1, description="状态 0:启用 1:停用")
|
|
|
|
def __post_init__(self) -> None:
|
|
if self.template_name:
|
|
self.template_name = (QueueEnum.like.value, self.template_name)
|
|
if self.template_type:
|
|
self.template_type = (QueueEnum.eq.value, self.template_type)
|
|
if self.version:
|
|
self.version = (QueueEnum.like.value, self.version)
|
|
if isinstance(self.status, int):
|
|
self.status = (QueueEnum.eq.value, self.status)
|
|
|