Create ontology_reasoner.py

ontology_reasoner.py

@@ -1,4 +1,7 @@
-# ontology_reasoner.py
+"""
+Owlready2 wrapper for a small infrastructure ontology.
+Provides classification and consistency checking.
+"""
 import logging
 logger = logging.getLogger(__name__)
 
@@ -11,20 +14,82 @@ except ImportError:
 class InfraOntology:
     def __init__(self, path="infra.owl"):
         self.available = OWLREADY_AVAILABLE
+        self.onto = None
         if self.available:
             try:
                 self.onto = get_ontology(path).load()
-            except:
+                logger.info(f"Ontology loaded from {path}")
+            except Exception as e:
+                logger.warning(f"Failed to load ontology: {e}")
                 self.available = False
-                logger.warning("Ontology load failed, using mock")
         else:
             logger.info("Owlready2 not installed, ontology disabled")
+    
+    def classify(self, component_type: str) -> dict:
+        """
+        Run reasoner and return inferred classes for a given component type.
+        Returns dict with 'inferred' list and 'consistent' bool.
+        """
+        if not self.available or self.onto is None:
+            return {"inferred": [], "consistent": True}
+        
+        try:
+            with self.onto:
+                sync_reasoner()
+            # Find all individuals of the given type (simplified)
+            inferred = []
+            # Example: if component_type == "server", get all Server instances
+            if hasattr(self.onto, component_type.capitalize()):
+                cls = getattr(self.onto, component_type.capitalize())
+                inferred = [str(i) for i in cls.instances()]
+            return {"inferred": inferred, "consistent": True}
+        except Exception as e:
+            logger.error(f"Reasoning error: {e}")
+            return {"inferred": [], "consistent": False, "error": str(e)}"""
+Owlready2 wrapper for a small infrastructure ontology.
+Provides classification and consistency checking.
+"""
+import logging
+logger = logging.getLogger(__name__)
+
+try:
+    from owlready2 import *
+    OWLREADY_AVAILABLE = True
+except ImportError:
+    OWLREADY_AVAILABLE = False
 
-    def classify(self, component_type):
-        if not self.available:
+class InfraOntology:
+    def __init__(self, path="infra.owl"):
+        self.available = OWLREADY_AVAILABLE
+        self.onto = None
+        if self.available:
+            try:
+                self.onto = get_ontology(path).load()
+                logger.info(f"Ontology loaded from {path}")
+            except Exception as e:
+                logger.warning(f"Failed to load ontology: {e}")
+                self.available = False
+        else:
+            logger.info("Owlready2 not installed, ontology disabled")
+    
+    def classify(self, component_type: str) -> dict:
+        """
+        Run reasoner and return inferred classes for a given component type.
+        Returns dict with 'inferred' list and 'consistent' bool.
+        """
+        if not self.available or self.onto is None:
             return {"inferred": [], "consistent": True}
-        # Run reasoner and return results
-        with self.onto:
-            sync_reasoner()
-        # ... extract inferences
-        return {"inferred": ["NetworkDevice"], "consistent": True}
+        
+        try:
+            with self.onto:
+                sync_reasoner()
+            # Find all individuals of the given type (simplified)
+            inferred = []
+            # Example: if component_type == "server", get all Server instances
+            if hasattr(self.onto, component_type.capitalize()):
+                cls = getattr(self.onto, component_type.capitalize())
+                inferred = [str(i) for i in cls.instances()]
+            return {"inferred": inferred, "consistent": True}
+        except Exception as e:
+            logger.error(f"Reasoning error: {e}")
+            return {"inferred": [], "consistent": False, "error": str(e)}