schema.py 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. from urllib.parse import urlparse
  2. from fastapi import Query
  3. from pydantic import BaseModel, ConfigDict, Field, field_validator
  4. from app.common.enums import QueueEnum
  5. from app.core.base_schema import BaseSchema, TenantBySchema, UserBySchema
  6. from app.core.validator import DateTimeStr
  7. class ApplicationCreateSchema(BaseModel):
  8. """应用创建模型"""
  9. name: str = Field(..., max_length=64, description="应用名称")
  10. access_url: str = Field(..., max_length=255, description="访问地址")
  11. icon_url: str | None = Field(None, max_length=300, description="应用图标URL")
  12. status: str = Field("0", description="是否启用(0:启用 1:禁用)")
  13. description: str | None = Field(default=None, max_length=255, description="描述")
  14. @field_validator("access_url")
  15. @classmethod
  16. def _validate_access_url(cls, v: str) -> str:
  17. v = v.strip()
  18. if not v:
  19. raise ValueError("访问地址不能为空")
  20. parsed = urlparse(v)
  21. if parsed.scheme not in ("http", "https"):
  22. raise ValueError("访问地址必须为 http/https URL")
  23. return v
  24. @field_validator("icon_url")
  25. @classmethod
  26. def _validate_icon_url(cls, v: str | None) -> str | None:
  27. if v is None:
  28. return v
  29. v = v.strip()
  30. if not v:
  31. return None
  32. parsed = urlparse(v)
  33. if parsed.scheme not in ("http", "https"):
  34. raise ValueError("应用图标URL必须为 http/https URL")
  35. return v
  36. class ApplicationUpdateSchema(ApplicationCreateSchema):
  37. """应用更新模型"""
  38. class ApplicationOutSchema(ApplicationCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
  39. """应用响应模型"""
  40. model_config = ConfigDict(from_attributes=True)
  41. class PluginInfoOut(BaseModel):
  42. """``app/plugin/module_*`` 插件目录与可选 ``plugin.toml`` 的汇总信息。"""
  43. module_dir: str = Field(..., description="目录名,如 module_example")
  44. route_prefix: str = Field(..., description="动态路由容器前缀,如 /example")
  45. has_manifest: bool = Field(..., description="是否存在 plugin.toml")
  46. name: str | None = Field(None, description="manifest 中的 name,应与目录段一致")
  47. title: str | None = None
  48. version: str | None = None
  49. description: str | None = None
  50. optional: bool | None = Field(None, description="语义:是否可关闭该子系统")
  51. tags: list[str] | None = None
  52. manifest_name_mismatch: bool | None = Field(
  53. None, description="manifest 的 name 与目录 module_<name> 不一致时为 True"
  54. )
  55. class ApplicationQueryParam:
  56. """应用系统查询参数"""
  57. def __init__(
  58. self,
  59. name: str | None = Query(None, description="应用名称"),
  60. status: str | None = Query(None, description="是否启用"),
  61. created_time: list[DateTimeStr] | None = Query(
  62. None,
  63. description="创建时间范围",
  64. examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
  65. ),
  66. updated_time: list[DateTimeStr] | None = Query(
  67. None,
  68. description="更新时间范围",
  69. examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
  70. ),
  71. created_id: int | None = Query(None, description="创建人"),
  72. updated_id: int | None = Query(None, description="更新人"),
  73. ) -> None:
  74. # 模糊查询字段
  75. self.name = (QueueEnum.like.value, name)
  76. # 精确查询字段
  77. self.status = (QueueEnum.eq.value, status)
  78. self.created_id = (QueueEnum.eq.value, created_id)
  79. self.updated_id = (QueueEnum.eq.value, updated_id)
  80. # 时间范围查询
  81. if created_time and len(created_time) == 2:
  82. self.created_time = (QueueEnum.between.value, (created_time[0], created_time[1]))
  83. if updated_time and len(updated_time) == 2:
  84. self.updated_time = (QueueEnum.between.value, (updated_time[0], updated_time[1]))