Implement Manticore fully and re-theme

core/management/commands/event_ledger_smoke.py

@@ -5,12 +5,46 @@ import time
 
 from django.core.management.base import BaseCommand, CommandError
 
+from core.events.manticore import get_recent_event_rows
 from core.models import ConversationEvent
 
 
 class Command(BaseCommand):
     help = "Quick non-mutating sanity check for recent canonical event writes."
 
+    def _recent_rows(self, *, minutes: int, service: str, user_id: str, limit: int):
+        cutoff_ts = int(time.time() * 1000) - (minutes * 60 * 1000)
+        queryset = ConversationEvent.objects.filter(ts__gte=cutoff_ts).order_by("-ts")
+        if service:
+            queryset = queryset.filter(origin_transport=service)
+        if user_id:
+            queryset = queryset.filter(user_id=user_id)
+
+        rows = list(
+            queryset.values(
+                "id",
+                "user_id",
+                "session_id",
+                "ts",
+                "event_type",
+                "direction",
+                "origin_transport",
+                "trace_id",
+            )[:limit]
+        )
+        if rows:
+            return rows, "django"
+        try:
+            manticore_rows = get_recent_event_rows(
+                minutes=minutes,
+                service=service,
+                user_id=user_id,
+                limit=limit,
+            )
+        except Exception:
+            manticore_rows = []
+        return manticore_rows, "manticore" if manticore_rows else "django"
+
     def add_arguments(self, parser):
         parser.add_argument("--minutes", type=int, default=120)
         parser.add_argument("--service", default="")
@@ -34,24 +68,11 @@ class Command(BaseCommand):
             if item.strip()
         ]
 
-        cutoff_ts = int(time.time() * 1000) - (minutes * 60 * 1000)
-        queryset = ConversationEvent.objects.filter(ts__gte=cutoff_ts).order_by("-ts")
-        if service:
-            queryset = queryset.filter(origin_transport=service)
-        if user_id:
-            queryset = queryset.filter(user_id=user_id)
-
-        rows = list(
-            queryset.values(
-                "id",
-                "user_id",
-                "session_id",
-                "ts",
-                "event_type",
-                "direction",
-                "origin_transport",
-                "trace_id",
-            )[:limit]
+        rows, data_source = self._recent_rows(
+            minutes=minutes,
+            service=service,
+            user_id=user_id,
+            limit=limit,
         )
         event_type_counts = {}
         for row in rows:
@@ -67,6 +88,7 @@ class Command(BaseCommand):
             "minutes": minutes,
             "service": service,
             "user_id": user_id,
+            "data_source": data_source,
             "count": len(rows),
             "event_type_counts": event_type_counts,
             "required_types": required_types,
@@ -79,7 +101,7 @@ class Command(BaseCommand):
             return
 
         self.stdout.write(
-            f"event-ledger-smoke minutes={minutes} service={service or '-'} user={user_id or '-'} count={len(rows)}"
+            f"event-ledger-smoke minutes={minutes} service={service or '-'} user={user_id or '-'} source={data_source} count={len(rows)}"
         )
         self.stdout.write(f"event_type_counts={event_type_counts}")
         if required_types:
@@ -88,7 +110,7 @@ class Command(BaseCommand):
             )
 
         if fail_if_empty and len(rows) == 0:
-            raise CommandError("No recent ConversationEvent rows found.")
+            raise CommandError("No recent canonical event rows found.")
         if missing_required_types:
             raise CommandError(
                 "Missing required event types: " + ", ".join(missing_required_types)