added db

db/repositories/runs.py

@@ -0,0 +1,82 @@
+"""Repository for the `runs` table."""
+from __future__ import annotations
+
+import json
+from typing import Any
+
+from db.connection import get_db
+
+
+class RunsRepository:
+    async def insert(
+        self,
+        scenario: str,
+        identifier: str,
+        success: bool,
+        failure_reason: str | None,
+        steps: list[dict[str, Any]],
+        total_ms: int,
+    ) -> int:
+        async with get_db() as db:
+            cur = await db.execute(
+                """INSERT INTO runs
+                       (scenario, identifier, success, failure_reason, steps_json, total_ms)
+                   VALUES (?, ?, ?, ?, ?, ?)""",
+                (scenario, identifier, int(success), failure_reason, json.dumps(steps), total_ms),
+            )
+            await db.commit()
+            return int(cur.lastrowid or 0)
+
+    async def stats(self) -> dict[str, Any]:
+        async with get_db() as db:
+            total      = (await (await db.execute("SELECT COUNT(*) FROM runs")).fetchone())[0]
+            unique_ids = (await (await db.execute("SELECT COUNT(DISTINCT identifier) FROM runs")).fetchone())[0]
+            successes  = (await (await db.execute("SELECT COUNT(*) FROM runs WHERE success=1")).fetchone())[0]
+            failures   = (await (await db.execute("SELECT COUNT(*) FROM runs WHERE success=0")).fetchone())[0]
+
+            failure_reasons = [
+                {"reason": r["failure_reason"], "count": r["cnt"]}
+                for r in await (await db.execute(
+                    """SELECT failure_reason, COUNT(*) as cnt
+                       FROM runs WHERE success=0 AND failure_reason IS NOT NULL
+                       GROUP BY failure_reason ORDER BY cnt DESC LIMIT 10"""
+                )).fetchall()
+            ]
+
+            scenario_breakdown = [
+                {
+                    "scenario": r["scenario"],
+                    "total":    r["total"],
+                    "success":  r["ok"],
+                    "failure":  r["fail"],
+                }
+                for r in await (await db.execute(
+                    """SELECT scenario,
+                              COUNT(*) as total,
+                              SUM(success) as ok,
+                              COUNT(*) - SUM(success) as fail
+                       FROM runs GROUP BY scenario"""
+                )).fetchall()
+            ]
+
+            recent = [
+                dict(r)
+                for r in await (await db.execute(
+                    """SELECT id, created_at, scenario, identifier, success, failure_reason, total_ms
+                       FROM runs ORDER BY id DESC LIMIT 50"""
+                )).fetchall()
+            ]
+
+        return {
+            "total_runs":          total,
+            "unique_identifiers":  unique_ids,
+            "successes":           successes,
+            "failures":            failures,
+            "success_rate":        round(successes / total * 100, 1) if total else 0.0,
+            "failure_reasons":     failure_reasons,
+            "scenario_breakdown":  scenario_breakdown,
+            "recent_runs":         recent,
+        }
+
+
+runs_repo = RunsRepository()