Create infra_simulator.py

infra_simulator.py

@@ -0,0 +1,34 @@
+# infra_simulator.py
+import random
+from typing import Dict, List, Optional
+
+class InfraSimulator:
+    """Simulates a dynamic infrastructure. Maintains per‑session state."""
+    def __init__(self):
+        self.reset()
+
+    def reset(self):
+        self.components = {
+            "switch-1": {"type": "switch", "status": "up", "connections": ["server-1", "server-2"]},
+            "server-1": {"type": "server", "status": "up", "connections": ["switch-1"], "services": ["db"]},
+            "server-2": {"type": "server", "status": "up", "connections": ["switch-1"], "services": ["web"]},
+            "service-db": {"type": "service", "status": "up", "runs_on": "server-1"},
+            "service-web": {"type": "service", "status": "up", "runs_on": "server-2"},
+        }
+        self.fault_mode: Optional[str] = None
+
+    def set_fault(self, fault: str):
+        self.reset()  # start fresh
+        self.fault_mode = fault
+        if fault == "switch_down":
+            self.components["switch-1"]["status"] = "down"
+        elif fault == "server_overload":
+            self.components["server-1"]["status"] = "overloaded"
+        elif fault == "cascade":
+            self.components["switch-1"]["status"] = "down"
+            self.components["server-1"]["status"] = "down"
+            self.components["service-db"]["status"] = "down"
+        # etc.
+
+    def read_state(self) -> Dict:
+        return self.components