@josephyan/qingflow-app-user-mcp 0.2.0-beta.20 → 0.2.0-beta.21

package/src/qingflow_mcp/tools/approval_tools.py

@@ -18,14 +18,19 @@ class ApprovalTools(ToolBase):
 
     def register(self, mcp: FastMCP) -> None:
         @mcp.tool()
-        def record_comment_add(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
-            return self.record_comment_add(profile=profile, app_key=app_key, apply_id=apply_id, payload=payload or {})
+        def record_comment_write(
+            profile: str = DEFAULT_PROFILE,
+            app_key: str = "",
+            record_id: int = 0,
+            payload: dict[str, Any] | None = None,
+        ) -> dict[str, Any]:
+            return self.record_comment_write(profile=profile, app_key=app_key, record_id=record_id, payload=payload or {})
 
         @mcp.tool()
         def record_comment_list(
             profile: str = DEFAULT_PROFILE,
             app_key: str = "",
-            apply_id: int = 0,
+            record_id: int = 0,
             page_size: int = 20,
             list_type: int | None = None,
             page_num: int | None = 1,
@@ -33,26 +38,26 @@ class ApprovalTools(ToolBase):
             return self.record_comment_list(
                 profile=profile,
                 app_key=app_key,
-                apply_id=apply_id,
+                apply_id=record_id,
                 page_size=page_size,
                 list_type=list_type,
                 page_num=page_num,
             )
 
         @mcp.tool()
-        def record_comment_mention_candidates(
+        def record_comment_mentions(
             profile: str = DEFAULT_PROFILE,
             app_key: str = "",
-            apply_id: int = 0,
+            record_id: int = 0,
             page_size: int = 20,
             page_num: int = 1,
             list_type: int | None = None,
             keyword: str | None = None,
         ) -> dict[str, Any]:
-            return self.record_comment_mention_candidates(
+            return self.record_comment_mentions(
                 profile=profile,
                 app_key=app_key,
-                apply_id=apply_id,
+                record_id=record_id,
                 page_size=page_size,
                 page_num=page_num,
                 list_type=list_type,
@@ -60,84 +65,180 @@ class ApprovalTools(ToolBase):
             )
 
         @mcp.tool()
-        def record_comment_mark_read(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0) -> dict[str, Any]:
-            return self.record_comment_mark_read(profile=profile, app_key=app_key, apply_id=apply_id)
-
-        @mcp.tool()
-        def record_comment_stats(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0) -> dict[str, Any]:
-            return self.record_comment_stats(profile=profile, app_key=app_key, apply_id=apply_id)
+        def record_comment_mark_read(profile: str = DEFAULT_PROFILE, app_key: str = "", record_id: int = 0) -> dict[str, Any]:
+            return self.record_comment_mark_read(profile=profile, app_key=app_key, apply_id=record_id)
 
-        @mcp.tool(description=self._high_risk_tool_description(operation="approve", target="workflow record"))
-        def record_approve(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
-            return self.record_approve(profile=profile, app_key=app_key, apply_id=apply_id, payload=payload or {})
-
-        @mcp.tool(description=self._high_risk_tool_description(operation="reject", target="workflow record"))
-        def record_reject(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
-            return self.record_reject(profile=profile, app_key=app_key, apply_id=apply_id, payload=payload or {})
-
-        @mcp.tool()
-        def record_rollback_candidates(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, audit_node_id: int = 0) -> dict[str, Any]:
-            return self.record_rollback_candidates(profile=profile, app_key=app_key, apply_id=apply_id, audit_node_id=audit_node_id)
-
-        @mcp.tool()
-        def record_rollback(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
-            return self.record_rollback(profile=profile, app_key=app_key, apply_id=apply_id, payload=payload or {})
+        @mcp.tool(description=self._high_risk_tool_description(operation="approve", target="workflow task"))
+        def task_approve(profile: str = DEFAULT_PROFILE, app_key: str = "", record_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
+            return self.task_approve(profile=profile, app_key=app_key, record_id=record_id, payload=payload or {})
 
-        @mcp.tool()
-        def record_transfer(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
-            return self.record_transfer(profile=profile, app_key=app_key, apply_id=apply_id, payload=payload or {})
+        @mcp.tool(description=self._high_risk_tool_description(operation="reject", target="workflow task"))
+        def task_reject(profile: str = DEFAULT_PROFILE, app_key: str = "", record_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
+            return self.task_reject(profile=profile, app_key=app_key, record_id=record_id, payload=payload or {})
 
         @mcp.tool()
-        def record_transfer_candidates(
+        def task_rollback_candidates(
             profile: str = DEFAULT_PROFILE,
             app_key: str = "",
-            apply_id: int = 0,
-            page_size: int = 20,
-            page_num: int = 1,
-            audit_node_id: int = 0,
-            keyword: str | None = None,
+            record_id: int = 0,
+            workflow_node_id: int = 0,
         ) -> dict[str, Any]:
-            return self.record_transfer_candidates(
-                profile=profile,
-                app_key=app_key,
-                apply_id=apply_id,
-                page_size=page_size,
-                page_num=page_num,
-                audit_node_id=audit_node_id,
-                keyword=keyword,
-            )
+            return self.task_rollback_candidates(profile=profile, app_key=app_key, record_id=record_id, workflow_node_id=workflow_node_id)
 
         @mcp.tool()
-        def record_reassign_get(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0) -> dict[str, Any]:
-            return self.record_reassign_get(profile=profile, app_key=app_key, apply_id=apply_id)
+        def task_rollback(profile: str = DEFAULT_PROFILE, app_key: str = "", record_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
+            return self.task_rollback(profile=profile, app_key=app_key, record_id=record_id, payload=payload or {})
 
         @mcp.tool()
-        def record_reassign(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
-            return self.record_reassign(profile=profile, app_key=app_key, apply_id=apply_id, payload=payload or {})
+        def task_transfer(profile: str = DEFAULT_PROFILE, app_key: str = "", record_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
+            return self.task_transfer(profile=profile, app_key=app_key, record_id=record_id, payload=payload or {})
 
         @mcp.tool()
-        def record_countersign_candidates(
+        def task_transfer_candidates(
             profile: str = DEFAULT_PROFILE,
             app_key: str = "",
-            apply_id: int = 0,
+            record_id: int = 0,
             page_size: int = 20,
             page_num: int = 1,
-            audit_node_id: int = 0,
-            search_key: str | None = None,
+            workflow_node_id: int = 0,
+            keyword: str | None = None,
         ) -> dict[str, Any]:
-            return self.record_countersign_candidates(
+            return self.task_transfer_candidates(
                 profile=profile,
                 app_key=app_key,
-                apply_id=apply_id,
+                record_id=record_id,
                 page_size=page_size,
                 page_num=page_num,
-                audit_node_id=audit_node_id,
-                search_key=search_key,
+                workflow_node_id=workflow_node_id,
+                keyword=keyword,
             )
 
-        @mcp.tool()
-        def record_countersign(profile: str = DEFAULT_PROFILE, app_key: str = "", apply_id: int = 0, payload: dict[str, Any] | None = None) -> dict[str, Any]:
-            return self.record_countersign(profile=profile, app_key=app_key, apply_id=apply_id, payload=payload or {})
+    def record_comment_write(self, *, profile: str, app_key: str, record_id: int, payload: dict[str, Any]) -> dict[str, Any]:
+        raw = self.record_comment_add(profile=profile, app_key=app_key, apply_id=record_id, payload=payload)
+        return self._public_action_response(
+            raw,
+            action="record_comment_write",
+            resource={"app_key": app_key, "record_id": record_id},
+            selection={},
+        )
+
+    def record_comment_mentions(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        record_id: int,
+        page_size: int = 20,
+        page_num: int = 1,
+        list_type: int | None = None,
+        keyword: str | None = None,
+    ) -> dict[str, Any]:
+        raw = self.record_comment_mention_candidates(
+            profile=profile,
+            app_key=app_key,
+            apply_id=record_id,
+            page_size=page_size,
+            page_num=page_num,
+            list_type=list_type,
+            keyword=keyword,
+        )
+        items = _approval_page_items(raw.get("page"))
+        return self._public_page_response(
+            raw,
+            items=items,
+            pagination={
+                "page": page_num,
+                "page_size": page_size,
+                "returned_items": len(items),
+                "page_amount": _approval_page_amount(raw.get("page")),
+                "reported_total": _approval_page_total(raw.get("page")),
+            },
+            selection={"app_key": app_key, "record_id": record_id, "list_type": list_type, "keyword": keyword},
+        )
+
+    def task_approve(self, *, profile: str, app_key: str, record_id: int, payload: dict[str, Any]) -> dict[str, Any]:
+        raw = self.record_approve(profile=profile, app_key=app_key, apply_id=record_id, payload=payload)
+        return self._public_action_response(
+            raw,
+            action="task_approve",
+            resource={"app_key": app_key, "record_id": record_id},
+            selection={},
+            human_review=True,
+        )
+
+    def task_reject(self, *, profile: str, app_key: str, record_id: int, payload: dict[str, Any]) -> dict[str, Any]:
+        raw = self.record_reject(profile=profile, app_key=app_key, apply_id=record_id, payload=payload)
+        return self._public_action_response(
+            raw,
+            action="task_reject",
+            resource={"app_key": app_key, "record_id": record_id},
+            selection={},
+            human_review=True,
+        )
+
+    def task_rollback_candidates(self, *, profile: str, app_key: str, record_id: int, workflow_node_id: int) -> dict[str, Any]:
+        raw = self.record_rollback_candidates(profile=profile, app_key=app_key, apply_id=record_id, audit_node_id=workflow_node_id)
+        items = _approval_page_items(raw.get("result"))
+        return self._public_page_response(
+            raw,
+            items=items,
+            pagination={"returned_items": len(items)},
+            selection={"app_key": app_key, "record_id": record_id, "workflow_node_id": workflow_node_id},
+        )
+
+    def task_rollback(self, *, profile: str, app_key: str, record_id: int, payload: dict[str, Any]) -> dict[str, Any]:
+        raw = self.record_rollback(profile=profile, app_key=app_key, apply_id=record_id, payload=payload)
+        return self._public_action_response(
+            raw,
+            action="task_rollback",
+            resource={"app_key": app_key, "record_id": record_id},
+            selection={},
+            human_review=True,
+        )
+
+    def task_transfer(self, *, profile: str, app_key: str, record_id: int, payload: dict[str, Any]) -> dict[str, Any]:
+        raw = self.record_transfer(profile=profile, app_key=app_key, apply_id=record_id, payload=payload)
+        return self._public_action_response(
+            raw,
+            action="task_transfer",
+            resource={"app_key": app_key, "record_id": record_id},
+            selection={},
+            human_review=True,
+        )
+
+    def task_transfer_candidates(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        record_id: int,
+        page_size: int = 20,
+        page_num: int = 1,
+        workflow_node_id: int = 0,
+        keyword: str | None = None,
+    ) -> dict[str, Any]:
+        raw = self.record_transfer_candidates(
+            profile=profile,
+            app_key=app_key,
+            apply_id=record_id,
+            page_size=page_size,
+            page_num=page_num,
+            audit_node_id=workflow_node_id,
+            keyword=keyword,
+        )
+        items = _approval_page_items(raw.get("page"))
+        return self._public_page_response(
+            raw,
+            items=items,
+            pagination={
+                "page": page_num,
+                "page_size": page_size,
+                "returned_items": len(items),
+                "page_amount": _approval_page_amount(raw.get("page")),
+                "reported_total": _approval_page_total(raw.get("page")),
+            },
+            selection={"app_key": app_key, "record_id": record_id, "workflow_node_id": workflow_node_id, "keyword": keyword},
+        )
 
     def record_comment_add(self, *, profile: str, app_key: str, apply_id: int, payload: dict[str, Any]) -> dict[str, Any]:
         self._require_app_and_apply(app_key, apply_id)
@@ -145,7 +246,14 @@ class ApprovalTools(ToolBase):
 
         def runner(session_profile, context):
             result = self.backend.request("POST", context, f"/app/{app_key}/apply/{apply_id}/comment", json_body=payload)
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -167,9 +275,23 @@ class ApprovalTools(ToolBase):
                 "list_type": list_type,
                 "list_type_label": get_record_list_type_label(list_type),
                 "page": result,
+                "request_route": self._request_route_payload(context),
             }
 
-        return self._run(profile, runner)
+        raw = self._run(profile, runner)
+        items = _approval_page_items(raw.get("page"))
+        return self._public_page_response(
+            raw,
+            items=items,
+            pagination={
+                "page": page_num,
+                "page_size": page_size,
+                "returned_items": len(items),
+                "page_amount": _approval_page_amount(raw.get("page")),
+                "reported_total": _approval_page_total(raw.get("page")),
+            },
+            selection={"app_key": app_key, "record_id": apply_id, "list_type": list_type},
+        )
 
     def record_comment_mention_candidates(
         self,
@@ -199,6 +321,7 @@ class ApprovalTools(ToolBase):
                 "list_type": list_type,
                 "list_type_label": get_record_list_type_label(list_type),
                 "page": result,
+                "request_route": self._request_route_payload(context),
             }
 
         return self._run(profile, runner)
@@ -208,16 +331,36 @@ class ApprovalTools(ToolBase):
 
         def runner(session_profile, context):
             result = self.backend.request("POST", context, f"/app/{app_key}/apply/{apply_id}/comment/read")
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
-        return self._run(profile, runner)
+        raw = self._run(profile, runner)
+        return self._public_action_response(
+            raw,
+            action="record_comment_mark_read",
+            resource={"app_key": app_key, "record_id": apply_id},
+            selection={},
+        )
 
     def record_comment_stats(self, *, profile: str, app_key: str, apply_id: int) -> dict[str, Any]:
         self._require_app_and_apply(app_key, apply_id)
 
         def runner(session_profile, context):
             result = self.backend.request("GET", context, f"/app/{app_key}/apply/{apply_id}/comment/statistic")
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -273,7 +416,14 @@ class ApprovalTools(ToolBase):
                 f"/app/{app_key}/apply/{apply_id}/revertNode",
                 params={"auditNodeId": audit_node_id},
             )
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -284,7 +434,14 @@ class ApprovalTools(ToolBase):
 
         def runner(session_profile, context):
             result = self.backend.request("POST", context, f"/app/{app_key}/apply/{apply_id}/rollback", json_body=body)
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -295,7 +452,14 @@ class ApprovalTools(ToolBase):
 
         def runner(session_profile, context):
             result = self.backend.request("POST", context, f"/app/{app_key}/apply/{apply_id}/transfer", json_body=body)
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -319,7 +483,14 @@ class ApprovalTools(ToolBase):
             if keyword:
                 params["keyword"] = keyword
             result = self.backend.request("GET", context, f"/app/{app_key}/apply/{apply_id}/transfer/member", params=params)
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "page": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "page": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -328,7 +499,14 @@ class ApprovalTools(ToolBase):
 
         def runner(session_profile, context):
             result = self.backend.request("GET", context, f"/app/{app_key}/apply/{apply_id}/reassign")
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -338,7 +516,14 @@ class ApprovalTools(ToolBase):
 
         def runner(session_profile, context):
             result = self.backend.request("POST", context, f"/app/{app_key}/apply/{apply_id}/reassign", json_body=body)
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -362,7 +547,14 @@ class ApprovalTools(ToolBase):
             if search_key:
                 params["searchKey"] = search_key
             result = self.backend.request("GET", context, f"/app/{app_key}/apply/{apply_id}/countersign/member", params=params)
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "page": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "page": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
@@ -373,10 +565,29 @@ class ApprovalTools(ToolBase):
 
         def runner(session_profile, context):
             result = self.backend.request("POST", context, f"/app/{app_key}/apply/{apply_id}/countersign", json_body=body)
-            return {"profile": profile, "ws_id": session_profile.selected_ws_id, "app_key": app_key, "apply_id": apply_id, "result": result}
+            return {
+                "profile": profile,
+                "ws_id": session_profile.selected_ws_id,
+                "app_key": app_key,
+                "apply_id": apply_id,
+                "result": result,
+                "request_route": self._request_route_payload(context),
+            }
 
         return self._run(profile, runner)
 
+    def _request_route_payload(self, context) -> JSONObject:  # type: ignore[no-untyped-def]
+        describe_route = getattr(self.backend, "describe_route", None)
+        if callable(describe_route):
+            payload = describe_route(context)
+            if isinstance(payload, dict):
+                return payload
+        return {
+            "base_url": getattr(context, "base_url", None),
+            "qf_version": getattr(context, "qf_version", None),
+            "qf_version_source": getattr(context, "qf_version_source", None) or ("context" if getattr(context, "qf_version", None) else "unknown"),
+        }
+
     def _require_app_and_apply(self, app_key: str, apply_id: int) -> None:
         if not app_key:
             raise_tool_error(QingflowApiError.config_error("app_key is required"))
@@ -485,6 +696,47 @@ class ApprovalTools(ToolBase):
         if payload.get("handSignImageUrl"):
             raise_tool_error(QingflowApiError.not_supported("NOT_SUPPORTED_IN_V1: handSignImageUrl is not supported"))
 
+    def _public_page_response(
+        self,
+        raw: dict[str, Any],
+        *,
+        items: list[dict[str, Any]],
+        pagination: dict[str, Any],
+        selection: dict[str, Any],
+    ) -> dict[str, Any]:
+        response = dict(raw)
+        response["ok"] = bool(raw.get("ok", True))
+        response["warnings"] = []
+        response["output_profile"] = "normal"
+        response["data"] = {
+            "items": items,
+            "pagination": pagination,
+            "selection": selection,
+        }
+        return response
+
+    def _public_action_response(
+        self,
+        raw: dict[str, Any],
+        *,
+        action: str,
+        resource: dict[str, Any],
+        selection: dict[str, Any],
+        human_review: bool = False,
+    ) -> dict[str, Any]:
+        response = dict(raw)
+        response["ok"] = bool(raw.get("ok", True))
+        response["warnings"] = []
+        response["output_profile"] = "normal"
+        response["data"] = {
+            "action": action,
+            "resource": resource,
+            "selection": selection,
+            "result": raw.get("result"),
+            "human_review": human_review,
+        }
+        return response
+
     def _request_route_payload(self, context) -> JSONObject:  # type: ignore[no-untyped-def]
         describe_route = getattr(self.backend, "describe_route", None)
         if callable(describe_route):
@@ -496,3 +748,33 @@ class ApprovalTools(ToolBase):
             "qf_version": context.qf_version,
             "qf_version_source": getattr(context, "qf_version_source", None) or ("context" if getattr(context, "qf_version", None) else "unknown"),
         }
+
+
+def _approval_page_items(payload: Any) -> list[dict[str, Any]]:
+    if isinstance(payload, list):
+        return [item for item in payload if isinstance(item, dict)]
+    if not isinstance(payload, dict):
+        return []
+    for key in ("list", "items", "rows", "result"):
+        value = payload.get(key)
+        if isinstance(value, list):
+            return [item for item in value if isinstance(item, dict)]
+    for container_key in ("data", "page"):
+        nested = payload.get(container_key)
+        if isinstance(nested, dict):
+            nested_items = _approval_page_items(nested)
+            if nested_items:
+                return nested_items
+    return []
+
+
+def _approval_page_amount(payload: Any) -> Any:
+    if isinstance(payload, dict):
+        return payload.get("pageAmount", payload.get("page_amount"))
+    return None
+
+
+def _approval_page_total(payload: Any) -> Any:
+    if isinstance(payload, dict):
+        return payload.get("total", payload.get("count"))
+    return None