mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-10-15 05:19:17 +00:00
zhangtao
07fa789d2e
本批为 t10 → t13 → t15 三轮实现 + 一轮复核驳回整改的最终形态（pytest 83 passed / ruff clean，
已部署并线上验证）。
异常语义
- CustomException 默认状态码 500 → 400（约 433 处裸调用不再被计为服务端故障），
且「包装底层异常」的 130 处显式恢复 500；get_or_404 → 404；响应体结构不变
代码生成注入面（复核 blocker F1，二次修复）
- column_type 不再做字符串拼接：改为常量白名单派生（类型名 ∈ DB_TO_SQLALCHEMY ∪ {Boolean}，
长度仅取数字/正则捕获），彻底移除 split("(")[1]
- 列类型加形状白名单；字符串/注释/文档字符串用 \uXXXX，标识符过 safe_ident（含 Python/JS 关键字），
.vue 文本与静态属性改 HTML 实体
- 落盘闸门由 ast.parse 升级为 assert_python_structure（语法 + 顶层语句白名单 + 与哨兵骨架逐节点比对）
与 assert_text_structure（.vue/.ts/.toml 行结构）；U+2028/2029 与 C1 控制符纳入转义
安全默认值
- 生产 CORS 不再回落 ["*"]（未配置即拒绝跨域，可选 CORS_STRICT_STARTUP 直接启动失败）
- OAUTH_ALLOWED_HOSTS 不再默认 "*"；新增 ALLOW_LOCALHOST_HOSTS 控制 Host 校验范围
- ALLOWED_HOSTS 收窄为按配置派生；健康检查暴露 code_exec_enabled / cors_origins_configured
定时任务
- SCHEDULER_ALLOW_CODE_EXEC 默认 False；新增 builtin:<模块名>[.<函数名>] 内置处理器路线（不经 exec）
- 判定收敛到 describe_schedulability；列表返回 schedulable / unschedulable_reason；
启动 reconcile 与状态查询对「已启用但不可调度」打 WARN（去重、恢复后清除）
- 加固：内置处理器必须在本模块内定义或经 HANDLERS 声明，拒绝模块内导入的外部对象（复核 R1）
- 顺带修复：_WARNED_UNSCHEDULABLE 的 set.pop TypeError、job.next_run_time 缺失导致的列表 500
测试：新增 tests/test_generator_template_safety.py、test_security_defaults.py、
test_scheduler_schedulability.py（含恶意载荷、闸门红绿、静态护栏）
101 lines
4.6 KiB
Python
101 lines
4.6 KiB
Python
from pydantic import (
|
|
BaseModel,
|
|
ConfigDict,
|
|
Field,
|
|
field_validator,
|
|
)
|
|
|
|
from app.core.base_schema import BaseQueryParam, BaseSchema, UserByQueryParam, UserBySchema
|
|
|
|
|
|
class SchedulerStatusSchema(BaseModel):
|
|
"""调度器状态"""
|
|
|
|
status: str = Field(..., description="调度器状态(运行中/已暂停/已停止/未知)")
|
|
is_running: bool = Field(..., description="是否运行中")
|
|
job_count: int = Field(..., description="调度器任务数量")
|
|
code_exec_enabled: bool = Field(default=False, description="是否允许执行用户提交的代码块(SCHEDULER_ALLOW_CODE_EXEC)")
|
|
unschedulable_count: int = Field(default=0, description="已注册但当前不可执行的任务数（>0 时列表页会给出原因）")
|
|
|
|
|
|
class SchedulerJobSchema(BaseModel):
|
|
"""调度器任务列表项"""
|
|
|
|
id: str = Field(..., description="任务ID")
|
|
name: str = Field(..., description="任务名称")
|
|
trigger: str = Field(..., description="触发器")
|
|
next_run_time: str | None = Field(default=None, description="下次运行时间")
|
|
status: int = Field(..., description="任务状态(0:运行中 1:暂停中 2:已停止 3:未知)")
|
|
schedulable: bool = Field(default=True, description="当前是否可执行；false 表示已注册但执行时会被拒绝")
|
|
unschedulable_reason: str | None = Field(default=None, description="不可执行原因（含修复建议）")
|
|
|
|
|
|
class SchedulerJobModifySchema(BaseModel):
|
|
"""调度器任务属性修改模型（仅允许白名单属性）"""
|
|
|
|
name: str | None = Field(default=None, min_length=1, max_length=128, description="任务名称")
|
|
coalesce: bool | None = Field(default=None, description="是否合并错过的执行")
|
|
max_instances: int | None = Field(default=None, ge=1, le=100, description="最大并发实例数")
|
|
misfire_grace_time: int | None = Field(default=None, ge=0, le=86400, description="错过执行的宽限时间(秒)")
|
|
|
|
def to_changes(self) -> dict:
|
|
"""转换为调度器属性参数（忽略未传字段）"""
|
|
return self.model_dump(exclude_none=True)
|
|
|
|
|
|
class JobCreateSchema(BaseModel):
|
|
"""执行日志创建模型"""
|
|
|
|
job_id: str = Field(..., max_length=64, description="任务ID")
|
|
job_name: str | None = Field(default=None, max_length=128, description="任务名称")
|
|
trigger_type: str | None = Field(default=None, max_length=32, description="触发方式")
|
|
status: int = Field(default=0, ge=0, le=5, description="执行状态(0:待执行 1:执行中 2:成功 3:失败 4:超时 5:已取消)")
|
|
next_run_time: str | None = Field(default=None, description="下次执行时间")
|
|
job_state: str | None = Field(default=None, description="任务状态信息")
|
|
result: str | None = Field(default=None, description="执行结果")
|
|
error: str | None = Field(default=None, description="错误信息")
|
|
|
|
@field_validator("job_id")
|
|
@classmethod
|
|
def validate_job_id(cls, v: str) -> str:
|
|
v = v.strip()
|
|
if len(v) < 1 or len(v) > 64:
|
|
raise ValueError("任务ID长度必须在1-64个字符之间")
|
|
return v
|
|
|
|
@field_validator("trigger_type")
|
|
@classmethod
|
|
def validate_trigger_type(cls, v: str | None) -> str | None:
|
|
if v is None:
|
|
return v
|
|
allowed = {"cron", "interval", "date", "manual"}
|
|
v = v.strip()
|
|
if v not in allowed:
|
|
raise ValueError(f"触发方式必须为 {allowed}")
|
|
return v
|
|
|
|
|
|
class JobUpdateSchema(BaseModel):
|
|
"""执行日志更新模型"""
|
|
|
|
status: int | None = Field(default=None, ge=0, le=5, description="执行状态(0:待执行 1:执行中 2:成功 3:失败 4:超时 5:已取消)")
|
|
next_run_time: str | None = Field(default=None, description="下次执行时间")
|
|
job_state: str | None = Field(default=None, description="任务状态信息")
|
|
result: str | None = Field(default=None, description="执行结果")
|
|
error: str | None = Field(default=None, description="错误信息")
|
|
|
|
|
|
class JobOutSchema(JobCreateSchema, BaseSchema, UserBySchema):
|
|
"""执行日志响应模型"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
class JobQueryParam(BaseQueryParam, UserByQueryParam):
|
|
"""执行日志查询参数"""
|
|
|
|
job_id: str | None = Field(None, description="任务ID", json_schema_extra={"q": "eq"})
|
|
job_name: str | None = Field(None, description="任务名称", json_schema_extra={"q": "like"})
|
|
trigger_type: str | None = Field(None, description="触发方式", json_schema_extra={"q": "eq"})
|
|
status: int | None = Field(None, ge=0, le=5, description="执行状态(0:待执行 1:执行中 2:成功 3:失败 4:超时 5:已取消)", json_schema_extra={"q": "eq"})
|