schema.py 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. # -*- coding: utf-8 -*-
  2. from pydantic import BaseModel, ConfigDict, Field, field_validator
  3. from fastapi import Query
  4. from app.core.validator import DateTimeStr
  5. from app.core.base_schema import BaseSchema
  6. class ParamsCreateSchema(BaseModel):
  7. """配置创建模型"""
  8. config_name: str = Field(..., max_length=64, description="参数名称")
  9. config_key: str = Field(..., max_length=500, description="参数键名")
  10. config_value: str | None = Field(default=None, description="参数键值")
  11. config_type: bool = Field(default=False, description="系统内置(True:是 False:否)")
  12. status: str = Field(default="0", description="状态(True:正常 False:停用)")
  13. description: str | None = Field(default=None, max_length=500, description="描述")
  14. @field_validator('config_key')
  15. @classmethod
  16. def _validate_config_key(cls, v: str) -> str:
  17. v = v.strip().lower()
  18. import re
  19. if not re.match(r'^[a-z][a-z0-9_.-]*$', v):
  20. raise ValueError('参数键名必须以小写字母开头,仅包含小写字母/数字/_.-')
  21. return v
  22. class ParamsUpdateSchema(ParamsCreateSchema):
  23. """配置更新模型"""
  24. ...
  25. class ParamsOutSchema(ParamsCreateSchema, BaseSchema):
  26. """配置响应模型"""
  27. model_config = ConfigDict(from_attributes=True)
  28. class ParamsQueryParam:
  29. """配置管理查询参数"""
  30. def __init__(
  31. self,
  32. config_name: str | None = Query(None, description="配置名称"),
  33. config_key: str | None = Query(None, description="配置键名"),
  34. config_type: bool | None = Query(None, description="系统内置((True:是 False:否))"),
  35. created_time: list[DateTimeStr] | None = Query(None, description="创建时间范围", examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"]),
  36. updated_time: list[DateTimeStr] | None = Query(None, description="更新时间范围", examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"]),
  37. ) -> None:
  38. # 模糊查询字段
  39. self.config_name = ("like", config_name)
  40. self.config_key = ("like", config_key)
  41. # 精确查询字段
  42. self.config_type = config_type
  43. # 时间范围查询
  44. if created_time and len(created_time) == 2:
  45. self.created_time = ("between", (created_time[0], created_time[1]))
  46. if updated_time and len(updated_time) == 2:
  47. self.updated_time = ("between", (updated_time[0], updated_time[1]))