updates :)

db/repositories/batch_runs.py

@@ -0,0 +1,55 @@
+"""Repository for the `batch_runs` table."""
+from __future__ import annotations
+
+from typing import Any
+
+from db.connection import get_db
+
+
+class BatchRunsRepository:
+    async def insert(
+        self,
+        batch_id: str,
+        config_id: int | None,
+        config_name: str,
+        workers: int,
+        total_runs: int,
+        stagger_ms: int,
+        headless: bool,
+    ) -> None:
+        async with get_db() as db:
+            await db.execute(
+                """INSERT INTO batch_runs
+                   (batch_id, config_id, config_name, workers, total_runs, stagger_ms, headless)
+                   VALUES (?, ?, ?, ?, ?, ?, ?)""",
+                (batch_id, config_id, config_name, workers, total_runs, stagger_ms, int(headless)),
+            )
+            await db.commit()
+
+    async def finish(
+        self,
+        batch_id: str,
+        succeeded: int,
+        failed: int,
+        stopped: bool,
+        error: str | None,
+    ) -> None:
+        async with get_db() as db:
+            await db.execute(
+                """UPDATE batch_runs
+                   SET finished_at = strftime('%Y-%m-%dT%H:%M:%SZ', 'now'),
+                       succeeded = ?, failed = ?, stopped = ?, error = ?
+                   WHERE batch_id = ?""",
+                (succeeded, failed, int(stopped), error, batch_id),
+            )
+            await db.commit()
+
+    async def list(self, limit: int = 100) -> list[dict[str, Any]]:
+        async with get_db() as db:
+            rows = await (await db.execute(
+                "SELECT * FROM batch_runs ORDER BY id DESC LIMIT ?", (limit,)
+            )).fetchall()
+        return [dict(r) for r in rows]
+
+
+batch_runs_repo = BatchRunsRepository()