import re from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator from app.common.enums import QueueEnum from app.core.base_schema import BaseQueryParam, BaseSchema, TenantByQueryParam, TenantBySchema, UserByQueryParam, UserBySchema class ParamsCreateSchema(BaseModel): """参数创建模型 """ config_name: str = Field(..., min_length=1, max_length=64, description="参数名称") config_key: str = Field(..., min_length=1, max_length=500, description="参数键名(小写字母开头,仅允许字母数字_.-)") config_value: str | None = Field(default=None, description="参数键值") config_type: bool = Field(default=False, description="是否系统内置(True:是 False:否)") status: int = Field(default=0, ge=0, le=1, description="状态(0:正常 1:停用)") description: str | None = Field(default=None, max_length=500, description="参数描述") @field_validator("config_key") @classmethod def _validate_config_key(cls, v: str) -> str: """校验参数键名:小写字母开头,仅含字母/数字/_ . -""" v = v.strip().lower() if not re.match(r"^[a-z][a-z0-9_.-]*$", v): raise ValueError("参数键名必须以小写字母开头,仅允许小写字母、数字、_ . -") return v @field_validator("status") @classmethod def _validate_status(cls, v: int) -> int: """校验状态:仅支持 0(正常) 或 1(停用)""" if v not in {0, 1}: raise ValueError("状态仅支持 0(正常) 或 1(停用)") return v class ParamsUpdateSchema(ParamsCreateSchema): """参数更新模型 """ class ParamsOutSchema(ParamsCreateSchema, BaseSchema, UserBySchema, TenantBySchema): """参数响应模型 """ model_config = ConfigDict(from_attributes=True) class ParamsQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam): """参数管理查询参数 """ config_name: str | tuple[str, str] | None = Field(None, description="参数名称") config_key: str | tuple[str, str] | None = Field(None, description="参数键名") config_type: bool | tuple[str, bool] | None = Field(None, description="是否系统内置(True:是 False:否)") status: int | tuple[str, int] | None = Field(None, ge=0, le=1, description="状态(0:启动 1:停用)") @model_validator(mode="after") def validate_query_params(self) -> "ParamsQueryParam": if isinstance(self.config_name, str): self.config_name = (QueueEnum.like.value, self.config_name) if isinstance(self.config_key, str): self.config_key = (QueueEnum.like.value, self.config_key) if isinstance(self.config_type, bool): self.config_type = (QueueEnum.eq.value, self.config_type) if isinstance(self.status, int): self.status = (QueueEnum.eq.value, self.status) return self