Files
ai-exchange/scripts/run_alert_dispatcher.py
T
2026-05-12 01:17:52 +08:00

46 lines
1.7 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)}")
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())