Files
ai-exchange/scripts/run_alert_dispatcher.py
T
2026-05-12 08:30:47 +08:00

55 lines
2.3 KiB
Python

from __future__ import annotations
from pathlib import Path
import json
import os
import sys
ROOT = Path(__file__).resolve().parents[1]
if str(ROOT) not in sys.path:
sys.path.insert(0, str(ROOT))
from src.config.settings import settings
from src.services.alerts import AlertDispatcherService
def main(env=None) -> int:
env = os.environ if env is None else env
limit = int(env.get("ALERT_DISPATCH_LIMIT", str(settings.alert_dispatch_max_per_run)))
output_json = _get_bool(env, "ALERT_DISPATCH_OUTPUT_JSON", False)
payload = AlertDispatcherService().dispatch_pending(limit=limit)
if output_json:
print(json.dumps(payload, ensure_ascii=False, sort_keys=True, default=str))
else:
print(f"Alert dispatch attempted: {payload.get('attempted', 0)}")
items = payload.get("items") or []
delivered = sum(1 for item in items if item.get("status") == "delivered")
failed = sum(1 for item in items if item.get("status") == "failed")
skipped = sum(1 for item in items if item.get("status") == "skipped")
silent_window = dict(payload.get("silent_window") or {})
print(f"Delivered: {delivered} / Failed: {failed} / Skipped: {skipped}")
print(f"Silent window: {'active' if silent_window.get('active') else 'inactive'}")
print(f"Suppressed by silent window: {payload.get('suppressed_silent_window', 0)}")
print(f"Suppressed by noise policy: {payload.get('suppressed_noise_policy', 0)}")
print(f"Workbench-only: {payload.get('workbench_only', 0)}")
print(
"Media attached / linked / unavailable: "
f"{payload.get('media_attached', 0)} / {payload.get('media_linked', 0)} / {payload.get('media_unavailable', 0)}"
)
print(f"Channel disabled: {payload.get('channel_disabled', 0)}")
if payload.get("channel_summary"):
print(f"Channel summary: {json.dumps(payload.get('channel_summary'), ensure_ascii=False, sort_keys=True)}")
items = payload.get("items") or []
return 0 if all(item.get("status") != "failed" for item in items) else 2
def _get_bool(env, key: str, default: bool = False) -> bool:
value = env.get(key)
if value is None or value == "":
return default
return str(value).strip().lower() in {"1", "true", "yes", "on"}
if __name__ == "__main__":
raise SystemExit(main())