@josephyan/qingflow-cli 0.2.0-beta.999 → 1.0.6

package/src/qingflow_mcp/tools/import_tools.py

@@ -37,6 +37,13 @@ SAFE_REPAIRS = {
     "normalize_url_cells",
 }
 EMAIL_PATTERN = re.compile(r"^[^@\s]+@[^@\s]+\.[^@\s]+$")
+IMPORT_STATUS_BY_PROCESS_STATUS = {
+    1: "queued",
+    2: "running",
+    3: "succeeded",
+    4: "failed",
+    5: "partially_failed",
+}
 
 
 class ImportTools(ToolBase):
@@ -866,13 +873,26 @@ class ImportTools(ToolBase):
                         "process_id_str": normalized_process,
                     },
                 )
+            raw_process_status = matched_record.get("processStatus")
             total_rows = _coerce_int(matched_record.get("totalNumber") or matched_record.get("total_rows"))
             success_rows = _coerce_int(matched_record.get("successNum") or matched_record.get("success_rows"))
             failed_rows = _coerce_int(matched_record.get("errorNum") or matched_record.get("failed_rows"))
             progress = _coerce_int(matched_record.get("importPercentage") or matched_record.get("progress"))
+            normalized_status = _normalize_import_status(raw_process_status)
+            warnings: list[dict[str, str]] = []
+            if normalized_status in {"succeeded", "failed", "partially_failed"} and all(
+                value is None for value in (total_rows, success_rows, failed_rows)
+            ):
+                warnings.append(
+                    {
+                        "code": "IMPORT_STATUS_COUNTERS_MISSING",
+                        "message": "backend import history returned a terminal process status without row counters",
+                    }
+                )
             return {
                 "ok": True,
-                "status": _normalize_optional_text(matched_record.get("processStatus")) or "unknown",
+                "status": normalized_status,
+                "process_status": _coerce_int(raw_process_status),
                 "app_key": resolved_app_key,
                 "import_id": normalized_import_id or (local_job.get("import_id") if isinstance(local_job, dict) else None),
                 "process_id_str": normalized_process,
@@ -885,7 +905,7 @@ class ImportTools(ToolBase):
                 "error_file_urls": _normalize_error_file_urls(matched_record.get("errorFileUrls")),
                 "operate_time": matched_record.get("operateTime"),
                 "operate_user": matched_record.get("operateUser"),
-                "warnings": [],
+                "warnings": warnings,
                 "verification": {
                     "status_lookup_completed": True,
                     "matched_by": matched_by,
@@ -2217,6 +2237,26 @@ def _coerce_int(value: Any) -> int | None:
         return None
 
 
+def _normalize_import_status(value: Any) -> str:
+    status_code = _coerce_int(value)
+    if status_code is not None:
+        return IMPORT_STATUS_BY_PROCESS_STATUS.get(status_code, "unknown")
+    text = str(value or "").strip().lower()
+    if text in {"queued", "running", "succeeded", "failed", "partially_failed", "unknown"}:
+        return text
+    if text in {"line_up", "lineup"}:
+        return "queued"
+    if text in {"execute", "executing", "processing"}:
+        return "running"
+    if text in {"success", "completed"}:
+        return "succeeded"
+    if text in {"partly_fail", "partial_fail", "partially_fail", "partial_failed"}:
+        return "partially_failed"
+    if text in {"fail", "error"}:
+        return "failed"
+    return "unknown"
+
+
 def _normalize_error_file_urls(value: Any) -> list[str]:
     if isinstance(value, list):
         return [str(item).strip() for item in value if str(item).strip()]