schema.py 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. from fastapi import Query
  2. from pydantic import BaseModel, ConfigDict, Field, field_validator
  3. from app.common.enums import QueueEnum
  4. from app.core.base_schema import BaseSchema
  5. from app.core.validator import DateTimeStr
  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. class ParamsOutSchema(ParamsCreateSchema, BaseSchema):
  25. """配置响应模型"""
  26. model_config = ConfigDict(from_attributes=True)
  27. class ParamsQueryParam:
  28. """配置管理查询参数"""
  29. def __init__(
  30. self,
  31. config_name: str | None = Query(None, description="配置名称"),
  32. config_key: str | None = Query(None, description="配置键名"),
  33. config_type: bool | None = Query(None, description="系统内置((True:是 False:否))"),
  34. description: str | None = Query(None, description="描述"),
  35. status: str | None = Query(None, description="是否启用"),
  36. created_time: list[DateTimeStr] | None = Query(
  37. None,
  38. description="创建时间范围",
  39. examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
  40. ),
  41. updated_time: list[DateTimeStr] | None = Query(
  42. None,
  43. description="更新时间范围",
  44. examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
  45. ),
  46. ) -> None:
  47. # 模糊查询字段
  48. # 模糊查询字段
  49. self.config_name = (QueueEnum.like.value, config_name)
  50. self.config_key = (QueueEnum.like.value, config_key)
  51. # 精确查询字段
  52. self.config_type = config_type
  53. if description:
  54. self.description = (QueueEnum.like.value, description)
  55. # 精确查询字段
  56. if status:
  57. self.status = (QueueEnum.eq.value, status)
  58. # 时间范围查询
  59. if created_time and len(created_time) == 2:
  60. self.created_time = (QueueEnum.between.value, (created_time[0], created_time[1]))
  61. if updated_time and len(updated_time) == 2:
  62. self.updated_time = (QueueEnum.between.value, (updated_time[0], updated_time[1]))