数据看板

data_dashboard_service.py

@@ -0,0 +1,59 @@
+from typing import Any
+
+import requests
+
+from config import Config
+from task_service import TaskError
+
+
+def _headers() -> dict[str, str]:
+    if not Config.SEARCH_QA_API_KEY:
+        raise TaskError(
+            "SEARCH_QA_NOT_CONFIGURED",
+            "搜索问一问开放 API 尚未配置",
+            503,
+        )
+    return {"X-API-Key": Config.SEARCH_QA_API_KEY}
+
+
+def _request(method: str, path: str, **kwargs) -> requests.Response:
+    try:
+        session = requests.Session()
+        # 搜索问一问开放 API 必须直连；本机或服务环境的全局代理可能
+        # 指向不可用端口，且不应影响这一条受控的服务端对服务端链路。
+        session.trust_env = False
+        response = session.request(
+            method,
+            f"{Config.SEARCH_QA_API_BASE_URL}{path}",
+            headers={**_headers(), **kwargs.pop("headers", {})},
+            timeout=Config.SEARCH_QA_TIMEOUT_SECONDS,
+            **kwargs,
+        )
+    except requests.RequestException as exc:
+        raise TaskError("SEARCH_QA_UNAVAILABLE", "搜索问一问服务暂时不可用", 502) from exc
+    if response.status_code == 429:
+        raise TaskError("SEARCH_QA_RATE_LIMITED", "搜索问一问接口请求过于频繁，请稍后重试", 429)
+    if response.status_code >= 400:
+        message = "搜索问一问服务请求失败"
+        try:
+            payload = response.json()
+            message = str(payload.get("message") or payload.get("error") or message)
+        except ValueError:
+            pass
+        raise TaskError("SEARCH_QA_UPSTREAM_ERROR", message, 502)
+    return response
+
+
+def list_runs(task_id: str = "", xhs_account_id: str = "", limit: int = 20) -> dict[str, Any]:
+    limit = max(1, min(int(limit), 100))
+    params: dict[str, Any] = {"limit": limit}
+    if task_id:
+        params = {"task_id": task_id}
+    elif xhs_account_id:
+        params["xhs_account_id"] = xhs_account_id
+    payload = _request("GET", "/api/open/v1/search-qa/runs", params=params).json()
+    if not payload.get("ok"):
+        raise TaskError("SEARCH_QA_UPSTREAM_ERROR", "搜索问一问服务返回失败", 502)
+    runs = payload.get("runs") or ([payload["run"]] if payload.get("run") else [])
+    return {"items": runs, "total": len(runs), "query": params}
+