Create infra_graph.py

infra_graph.py

@@ -0,0 +1,41 @@
+# infra_graph.py
+import logging
+import networkx as nx
+from typing import Dict, Optional
+
+logger = logging.getLogger(__name__)
+
+class InfraGraph:
+    def __init__(self, uri: Optional[str] = None, user: Optional[str] = None, password: Optional[str] = None):
+        self.neo4j_driver = None
+        if uri and user and password:
+            try:
+                from neo4j import GraphDatabase
+                self.neo4j_driver = GraphDatabase.driver(uri, auth=(user, password))
+                logger.info("Connected to Neo4j")
+            except Exception as e:
+                logger.warning(f"Neo4j connection failed, using mock: {e}")
+        # Always create a NetworkX graph as fallback
+        self.nx_graph = nx.DiGraph()
+
+    def update_from_state(self, components: Dict):
+        """Populate graph from simulator state."""
+        self.nx_graph.clear()
+        for cid, props in components.items():
+            self.nx_graph.add_node(cid, **props)
+            for conn in props.get("connections", []):
+                self.nx_graph.add_edge(cid, conn, relation="connects_to")
+        if self.neo4j_driver:
+            self._update_neo4j(components)
+
+    def _update_neo4j(self, components):
+        # Implementation to sync with Neo4j (Cypher queries)
+        pass
+
+    def get_failing_components(self):
+        """Return list of components with status != 'up'."""
+        failing = []
+        for node, data in self.nx_graph.nodes(data=True):
+            if data.get("status") != "up":
+                failing.append(node)
+        return failing