first commit

simulators/scenarios/runner.py

@@ -0,0 +1,88 @@
+"""
+DemoBMS Scenario Runner
+Triggers demo scenarios by publishing a control message to the MQTT broker.
+
+Usage:
+    python scenarios/runner.py --scenario COOLING_FAILURE --target crac-01
+    python scenarios/runner.py --scenario FAN_DEGRADATION --target crac-02
+    python scenarios/runner.py --scenario HIGH_TEMPERATURE --target hall-a
+    python scenarios/runner.py --scenario HUMIDITY_SPIKE --target hall-b
+    python scenarios/runner.py --scenario UPS_MAINS_FAILURE --target ups-01
+    python scenarios/runner.py --scenario POWER_SPIKE --target hall-a
+    python scenarios/runner.py --scenario RACK_OVERLOAD --target rack-A05/pdu
+    python scenarios/runner.py --scenario LEAK_DETECTED --target leak-01
+    python scenarios/runner.py --scenario RESET
+    python scenarios/runner.py --scenario RESET --target ups-01
+"""
+import sys
+import os
+import asyncio
+import json
+import argparse
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
+import aiomqtt
+from config import MQTT_HOST, MQTT_PORT
+
+SCENARIOS = {
+    # Cooling
+    "COOLING_FAILURE":    "CRAC unit goes offline — rack temperatures rise rapidly (target: crac-01 / crac-02)",
+    "FAN_DEGRADATION":    "CRAC fan bearing wear — fan speed drops, ΔT rises over ~25 min (target: crac-01 / crac-02)",
+    "COMPRESSOR_FAULT":   "Compressor trips — unit drops to fan-only, cooling capacity collapses to ~8% (target: crac-01 / crac-02)",
+    "DIRTY_FILTER":       "Filter fouling — ΔP rises, airflow and capacity degrade over time (target: crac-01 / crac-02)",
+    "HIGH_TEMPERATURE":   "Gradual ambient heat rise, e.g. summer overload — slower than COOLING_FAILURE (target: hall-a / hall-b)",
+    "HUMIDITY_SPIKE":     "Humidity climbs — condensation / humidifier fault risk (target: hall-a / hall-b)",
+    "CHILLER_FAULT":      "Chiller plant trips — chilled water supply lost (target: chiller-01)",
+    # Power
+    "UPS_MAINS_FAILURE":  "Mains power lost — UPS switches to battery and drains (target: ups-01 / ups-02)",
+    "POWER_SPIKE":        "PDU load surges across a room by up to 50% (target: hall-a / hall-b / rack-XXX/pdu)",
+    "RACK_OVERLOAD":      "Single rack redlines at ~85-95% of rated 10 kW capacity (target: rack-XXX/pdu)",
+    "PHASE_IMBALANCE":    "PDU phase A overloads, phase C drops — imbalance flag triggers (target: rack-XXX/pdu)",
+    "ATS_TRANSFER":       "Utility feed lost — ATS transfers load to generator (target: ats-01)",
+    "GENERATOR_FAILURE":  "Generator starts and runs under load following utility failure (target: gen-01)",
+    "GENERATOR_LOW_FUEL": "Generator fuel level drains to critical low (target: gen-01)",
+    "GENERATOR_FAULT":    "Generator fails to start — fault state, no output (target: gen-01)",
+    # Environmental / Life Safety
+    "LEAK_DETECTED":      "Water leak sensor triggers a critical alarm (target: leak-01 / leak-02 / leak-03)",
+    "VESDA_ALERT":        "Smoke obscuration rises into Alert/Action band (target: vesda-hall-a / vesda-hall-b)",
+    "VESDA_FIRE":         "Smoke obscuration escalates to Fire level (target: vesda-hall-a / vesda-hall-b)",
+    # Control
+    "RESET":              "Return all bots (or a specific target) to normal operation",
+    # ── Compound (multi-bot, time-sequenced) ─────────────────────────────────
+    "HOT_NIGHT":                 "CRAC-01 trips → temps rise → power spikes → CRAC-02 degrades → VESDA alert  [~10 min]",
+    "GENERATOR_TEST_GONE_WRONG": "ATS transfers → generator runs low on fuel → faults → UPS takes over  [~16 min]",
+    "SLOW_BURN":                 "Dirty filter → creeping temp+humidity → VESDA alert → CRAC protection trips  [~30 min]",
+    "LAST_RESORT":               "Utility fails → generator starts → faults at 2 min → UPS overloads → VESDA fire  [~9 min]",
+}
+
+
+async def trigger(scenario: str, target: str | None) -> None:
+    async with aiomqtt.Client(MQTT_HOST, port=MQTT_PORT) as client:
+        payload = json.dumps({"scenario": scenario, "target": target})
+        await client.publish("bms/control/scenario", payload, qos=1)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="DemoBMS Scenario Runner")
+    parser.add_argument("--scenario", choices=list(SCENARIOS.keys()), required=False)
+    parser.add_argument(
+        "--target",
+        help="Device ID to target (e.g. rack-A05/pdu, crac-01, ups-01, hall-a, leak-01). Omit for all.",
+        default=None,
+    )
+    parser.add_argument("--list", action="store_true", help="List all available scenarios and exit")
+    args = parser.parse_args()
+
+    if args.list or not args.scenario:
+        print("\nAvailable scenarios:\n")
+        for name, desc in SCENARIOS.items():
+            print(f"  {name:<22}  {desc}")
+        print()
+        sys.exit(0)
+
+    print(f"\n  Scenario : {args.scenario}")
+    print(f"  Target   : {args.target or 'ALL bots'}")
+    print(f"  Effect   : {SCENARIOS[args.scenario]}\n")
+
+    asyncio.run(trigger(args.scenario, args.target))
+    print("Done. Watch the dashboard for changes.")