68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
"""Workspace UI extension API. No model or device actions are dispatched."""
|
|
|
|
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from apps.admin_console.services import batch_archive, report_export, task_extensions
|
|
from apps.admin_console.services.automation_store import store
|
|
|
|
router = APIRouter(prefix="/api/task-extensions", tags=["task-extensions"])
|
|
|
|
|
|
class ExtensionUpdate(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
values: dict[str, str] = Field(max_length=20)
|
|
expected_revision: int = Field(ge=0)
|
|
request_id: UUID
|
|
|
|
|
|
@router.get("/{session_id}")
|
|
def read_extensions(session_id: str):
|
|
try:
|
|
return {
|
|
**task_extensions.task_context(session_id, store),
|
|
"plugins": task_extensions.contributions(session_id, store),
|
|
}
|
|
except ValueError as exc:
|
|
raise HTTPException(404, str(exc)) from exc
|
|
|
|
|
|
@router.put("/{session_id}/{plugin_id}")
|
|
def update_extension(session_id: str, plugin_id: str, value: ExtensionUpdate):
|
|
job = store.job(session_id) or {}
|
|
batch_id = job.get("batch_id")
|
|
if (
|
|
batch_id
|
|
and batch_archive.status(batch_id, store.get("batch", batch_id) or {})["status"]
|
|
== "preparing"
|
|
):
|
|
raise HTTPException(409, "报告正在打包，请稍后保存评审")
|
|
with (
|
|
report_export.export_lock(
|
|
*(["batch:" + batch_id] if batch_id else []), "session:" + session_id
|
|
),
|
|
task_extensions.plugin_store().mutation_lock,
|
|
):
|
|
context = read_extensions(session_id)
|
|
if not context["editable"]:
|
|
raise HTTPException(409, "请在任务结束后填写评审")
|
|
manifest = next((m for m in task_extensions.active_manifests() if m.id == plugin_id), None)
|
|
if manifest is None:
|
|
raise HTTPException(409, "插件未安装或未启用，请刷新页面")
|
|
try:
|
|
task_extensions.save_values(
|
|
store,
|
|
session_id,
|
|
manifest,
|
|
value.values,
|
|
value.expected_revision,
|
|
str(value.request_id),
|
|
)
|
|
except ValueError as exc:
|
|
raise HTTPException(409, str(exc)) from exc
|
|
if batch_id:
|
|
batch_archive.forget(batch_id)
|
|
return read_extensions(session_id)
|