@josephyan/qingflow-cli 0.2.0-beta.55 → 0.2.0-beta.56

package/src/qingflow_mcp/ops/tasks.py

@@ -0,0 +1,171 @@
+from __future__ import annotations
+
+from typing import Any
+
+from .base import normalize_exception, success_result, tool_payload
+
+
+class TaskOperations:
+    def __init__(self, tools: Any) -> None:
+        self._tools = tools
+
+    def list(
+        self,
+        *,
+        profile: str,
+        task_box: str,
+        flow_status: str,
+        app_key: str | None,
+        workflow_node_id: int | None,
+        query: str | None,
+        page: int,
+        page_size: int,
+    ) -> dict:
+        try:
+            raw = self._tools.task.task_list(
+                profile=profile,
+                task_box=task_box,
+                flow_status=flow_status,
+                app_key=app_key,
+                workflow_node_id=workflow_node_id,
+                query=query,
+                page=page,
+                page_size=page_size,
+            )
+        except Exception as error:  # noqa: BLE001
+            return normalize_exception(error)
+        data = tool_payload(raw)
+        if isinstance(data, dict) and isinstance(data.get("items"), list):
+            items = []
+            for item in data["items"]:
+                if not isinstance(item, dict):
+                    continue
+                title = item.get("title") or item.get("task_name") or item.get("node_name")
+                if not title:
+                    app_name = item.get("app_name") or item.get("app_key") or "任务"
+                    record_id = item.get("record_id")
+                    title = f"{app_name}#{record_id}" if record_id is not None else str(app_name)
+                normalized = dict(item)
+                normalized["title"] = title
+                items.append(normalized)
+            data = dict(data)
+            data["items"] = items
+        return success_result(
+            "TASKS_LISTED",
+            "已读取待办列表",
+            data=data if isinstance(data, dict) else {"items": data},
+            warnings=raw.get("warnings") if isinstance(raw, dict) and isinstance(raw.get("warnings"), list) else [],
+            meta={"profile": profile},
+            legacy=raw,
+        )
+
+    def show(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        record_id: int,
+        workflow_node_id: int,
+        include_candidates: bool,
+        include_associated_reports: bool,
+    ) -> dict:
+        try:
+            raw = self._tools.task.task_get(
+                profile=profile,
+                app_key=app_key,
+                record_id=record_id,
+                workflow_node_id=workflow_node_id,
+                include_candidates=include_candidates,
+                include_associated_reports=include_associated_reports,
+            )
+        except Exception as error:  # noqa: BLE001
+            return normalize_exception(error)
+        return success_result(
+            "TASK_READY",
+            "已读取待办详情",
+            data=tool_payload(raw),
+            warnings=raw.get("warnings") if isinstance(raw, dict) and isinstance(raw.get("warnings"), list) else [],
+            meta={"profile": profile},
+            legacy=raw,
+        )
+
+    def act(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        record_id: int,
+        workflow_node_id: int,
+        action: str,
+        payload: dict,
+    ) -> dict:
+        try:
+            raw = self._tools.task.task_action_execute(
+                profile=profile,
+                app_key=app_key,
+                record_id=record_id,
+                workflow_node_id=workflow_node_id,
+                action=action,
+                payload=payload,
+            )
+        except Exception as error:  # noqa: BLE001
+            return normalize_exception(error)
+        return success_result(
+            "TASK_ACTION_EXECUTED",
+            "已执行待办动作",
+            data=tool_payload(raw),
+            warnings=raw.get("warnings") if isinstance(raw, dict) and isinstance(raw.get("warnings"), list) else [],
+            meta={"profile": profile, "verification": raw.get("verification") if isinstance(raw, dict) else None},
+            legacy=raw,
+        )
+
+    def log(self, *, profile: str, app_key: str, record_id: int, workflow_node_id: int) -> dict:
+        try:
+            raw = self._tools.task.task_workflow_log_get(
+                profile=profile,
+                app_key=app_key,
+                record_id=record_id,
+                workflow_node_id=workflow_node_id,
+            )
+        except Exception as error:  # noqa: BLE001
+            return normalize_exception(error)
+        return success_result(
+            "TASK_LOG_READY",
+            "已读取流程日志",
+            data=tool_payload(raw),
+            warnings=raw.get("warnings") if isinstance(raw, dict) and isinstance(raw.get("warnings"), list) else [],
+            meta={"profile": profile},
+            legacy=raw,
+        )
+
+    def associated_report_detail(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        record_id: int,
+        workflow_node_id: int,
+        report_id: int,
+        page: int,
+        page_size: int,
+    ) -> dict:
+        try:
+            raw = self._tools.task.task_associated_report_detail_get(
+                profile=profile,
+                app_key=app_key,
+                record_id=record_id,
+                workflow_node_id=workflow_node_id,
+                report_id=report_id,
+                page=page,
+                page_size=page_size,
+            )
+        except Exception as error:  # noqa: BLE001
+            return normalize_exception(error)
+        return success_result(
+            "TASK_ASSOCIATED_REPORT_READY",
+            "已读取关联报表详情",
+            data=tool_payload(raw),
+            warnings=raw.get("warnings") if isinstance(raw, dict) and isinstance(raw.get("warnings"), list) else [],
+            meta={"profile": profile},
+            legacy=raw,
+        )

package/src/qingflow_mcp/ops/workspace.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+
+from typing import Any
+
+from .base import normalize_exception, success_result
+
+
+class WorkspaceOperations:
+    def __init__(self, tools: Any) -> None:
+        self._tools = tools
+
+    def list(
+        self,
+        *,
+        profile: str,
+        page_num: int,
+        page_size: int,
+        include_external: bool,
+    ) -> dict:
+        try:
+            raw = self._tools.workspace.workspace_list(
+                profile=profile,
+                page_num=page_num,
+                page_size=page_size,
+                include_external=include_external,
+            )
+        except Exception as error:  # noqa: BLE001
+            return normalize_exception(error)
+        page = raw.get("page") if isinstance(raw.get("page"), dict) else {}
+        items = page.get("list") if isinstance(page.get("list"), list) else []
+        normalized = [
+            {
+                "ws_id": item.get("wsId"),
+                "name": item.get("workspaceName") or item.get("wsName"),
+                "remark": item.get("remark"),
+                "system_version": item.get("systemVersion"),
+            }
+            for item in items
+            if isinstance(item, dict)
+        ]
+        return success_result(
+            "WORKSPACES_LISTED",
+            "已读取工作区列表",
+            data={
+                "items": normalized,
+                "page_num": page_num,
+                "page_size": page_size,
+                "include_external": include_external,
+                "total": page.get("total"),
+            },
+            meta={"profile": profile},
+            legacy=raw,
+        )
+
+    def use(self, *, profile: str, ws_id: int) -> dict:
+        try:
+            raw = self._tools.workspace.workspace_select(profile=profile, ws_id=ws_id)
+        except Exception as error:  # noqa: BLE001
+            return normalize_exception(error)
+        workspace = raw.get("workspace") if isinstance(raw.get("workspace"), dict) else {}
+        return success_result(
+            "WORKSPACE_SELECTED",
+            "已切换工作区",
+            data={
+                "workspace": {
+                    "ws_id": raw.get("selected_ws_id", ws_id),
+                    "name": raw.get("selected_ws_name"),
+                    "remark": workspace.get("remark"),
+                    "system_version": workspace.get("systemVersion"),
+                },
+                "qf_version": raw.get("qf_version"),
+                "qf_version_source": raw.get("qf_version_source"),
+            },
+            meta={"profile": profile, "request_route": raw.get("request_route")},
+            legacy=raw,
+        )