major refactor

tests/detect/manual/list_scenarios.py

@@ -35,9 +35,11 @@ logger = logging.getLogger(__name__)
 
 
 def main():
-    from core.db.detect import list_scenarios
+    from core.db import list_scenarios
+    from core.db.connection import get_session
 
-    scenarios = list_scenarios()
+    with get_session() as session:
+        scenarios = list_scenarios(session)
 
     if not scenarios:
         logger.info("No scenarios found. Create one with:")

tests/detect/manual/seed_scenario.py

@@ -121,15 +121,14 @@ def main():
     )
 
     # Mark as scenario
-    from core.db.detect import get_latest_checkpoint
+    from core.db import get_latest_checkpoint
     from core.db.connection import get_session
 
-    checkpoint = get_latest_checkpoint(branch_id)
-    if checkpoint:
-        checkpoint.is_scenario = True
-        checkpoint.scenario_label = args.label
-        with get_session() as session:
-            session.add(checkpoint)
+    with get_session() as session:
+        checkpoint = get_latest_checkpoint(session, branch_id)
+        if checkpoint:
+            checkpoint.is_scenario = True
+            checkpoint.scenario_label = args.label
             session.commit()
 
     logger.info("")