Initial commit

libs/neo/queries.py

@@ -0,0 +1,78 @@
+"""Neo4j Cypher queries for coverage policy system"""
+
+from datetime import datetime
+from typing import Any
+
+import structlog
+
+logger = structlog.get_logger()
+
+
+class TemporalQueries:
+    """Helper class for temporal queries"""
+
+    @staticmethod
+    def get_current_state_query(
+        label: str, filters: dict[str, Any] | None = None
+    ) -> str:
+        """Get query for current state of nodes"""
+        where_clause = "n.retracted_at IS NULL"
+
+        if filters:
+            filter_conditions = []
+            for key, value in filters.items():
+                if isinstance(value, str):
+                    filter_conditions.append(f"n.{key} = '{value}'")
+                else:
+                    filter_conditions.append(f"n.{key} = {value}")
+
+            if filter_conditions:
+                where_clause += " AND " + " AND ".join(filter_conditions)
+
+        return f"""
+        MATCH (n:{label})
+        WHERE {where_clause}
+        RETURN n
+        ORDER BY n.asserted_at DESC
+        """
+
+    @staticmethod
+    def get_historical_state_query(
+        label: str, as_of_time: datetime, filters: dict[str, Any] | None = None
+    ) -> str:
+        """Get query for historical state at specific time"""
+        where_clause = f"""
+        n.asserted_at <= datetime('{as_of_time.isoformat()}')
+        AND (n.retracted_at IS NULL OR n.retracted_at > datetime('{as_of_time.isoformat()}'))
+        """
+
+        if filters:
+            filter_conditions = []
+            for key, value in filters.items():
+                if isinstance(value, str):
+                    filter_conditions.append(f"n.{key} = '{value}'")
+                else:
+                    filter_conditions.append(f"n.{key} = {value}")
+
+            if filter_conditions:
+                where_clause += " AND " + " AND ".join(filter_conditions)
+
+        return f"""
+        MATCH (n:{label})
+        WHERE {where_clause}
+        RETURN n
+        ORDER BY n.asserted_at DESC
+        """
+
+    @staticmethod
+    def get_audit_trail_query(node_id: str) -> str:
+        """Get complete audit trail for a node"""
+        return f"""
+        MATCH (n {{id: '{node_id}'}})
+        RETURN n.asserted_at as asserted_at,
+               n.retracted_at as retracted_at,
+               n.source as source,
+               n.extractor_version as extractor_version,
+               properties(n) as properties
+        ORDER BY n.asserted_at ASC
+        """