102 lines
3.8 KiB
Python
102 lines
3.8 KiB
Python
"""Utility helpers for signal source repair service."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from .shared import Any, Optional, SignalSourceRegistry, os, re
|
||
|
||
|
||
class SignalSourceRepairUtilsMixin:
|
||
"""Common utility helpers for repair workflow."""
|
||
|
||
@staticmethod
|
||
def _safe_int(value: Any) -> Optional[int]:
|
||
try:
|
||
if value is None:
|
||
return None
|
||
return int(value)
|
||
except (TypeError, ValueError):
|
||
return None
|
||
|
||
@staticmethod
|
||
def _resolve_timeout_seconds() -> int:
|
||
raw = (
|
||
os.getenv("SIGNAL_SOURCE_REPAIR_TIMEOUT_SECONDS")
|
||
or os.getenv("LLM_REQUEST_TIMEOUT")
|
||
or "60"
|
||
)
|
||
try:
|
||
timeout = int(float(raw))
|
||
except (TypeError, ValueError):
|
||
timeout = 60
|
||
return max(10, min(timeout, 300))
|
||
|
||
@staticmethod
|
||
def _build_repair_message(*, project_id: int, registry_id: int, source_name: str) -> str:
|
||
return (
|
||
f"给项目ID: {project_id} 补录历史接入记录 registry#{registry_id}（{source_name}），"
|
||
"请仅执行入库，不需要二次处理。"
|
||
)
|
||
|
||
@staticmethod
|
||
def _build_explicit_repair_message(*, project_id: int, row: SignalSourceRegistry) -> str:
|
||
source_kind = str(row.source_kind or "").strip()[:40]
|
||
source_name = str(row.source_name or "历史信号源").strip()[:80]
|
||
source_identifier = str(row.source_identifier or "").strip()[:180]
|
||
return (
|
||
f"项目 {int(project_id)} 新增信号源：{source_name}。"
|
||
f"source_kind={source_kind}; source_identifier={source_identifier}。"
|
||
"这是历史脏数据补录，请执行 signal_source.onboard 入库并生成消息台记录。"
|
||
)
|
||
|
||
@staticmethod
|
||
def _build_capture_message_id(*, project_id: int, registry_id: int, stage: str) -> str:
|
||
phase = "explicit" if str(stage or "").strip().lower() == "explicit" else "base"
|
||
return f"om_registry_repair_{int(project_id)}_{int(registry_id)}_{phase}"[:120]
|
||
|
||
@staticmethod
|
||
def _build_group_chat_id(row: SignalSourceRegistry) -> str:
|
||
raw = (row.config or {}).get("chat_id") or row.source_identifier or f"repair_registry_{int(row.id)}"
|
||
return str(raw)[:120]
|
||
|
||
@staticmethod
|
||
def _extract_route_status(capture_result: dict[str, Any]) -> str:
|
||
rows = capture_result.get("results")
|
||
if not isinstance(rows, list) or not rows:
|
||
return ""
|
||
first = rows[0] if isinstance(rows[0], dict) else {}
|
||
return str(first.get("route_status") or "")
|
||
|
||
@classmethod
|
||
def _extract_signal_id(cls, capture_result: dict[str, Any]) -> Optional[int]:
|
||
candidates: list[Any] = [capture_result.get("signal_id")]
|
||
for key in ("signal_source_workflow", "signal_source_orchestration", "ai_execution"):
|
||
payload = capture_result.get(key)
|
||
if isinstance(payload, dict):
|
||
candidates.append(payload.get("signal_id"))
|
||
rows = capture_result.get("results")
|
||
if isinstance(rows, list) and rows and isinstance(rows[0], dict):
|
||
candidates.append(rows[0].get("signal_id"))
|
||
for item in candidates:
|
||
value = cls._safe_int(item)
|
||
if value is not None:
|
||
return value
|
||
return None
|
||
|
||
@staticmethod
|
||
def _normalize_identifier_key(value: Any) -> str:
|
||
key = re.sub(r"\s+", "", str(value or "").strip().lower())
|
||
if not key:
|
||
return ""
|
||
for prefix in (
|
||
"feishu:chat:",
|
||
"wecom:chat:",
|
||
"feishu_chat_",
|
||
"wecom_chat_",
|
||
"feishu:",
|
||
"wecom:",
|
||
):
|
||
if key.startswith(prefix):
|
||
key = key[len(prefix) :]
|
||
break
|
||
return key[:180]
|