moves load_statuses to the aim cli

mlibrary · Sep 25, 2024 · 7d37185 · 7d37185
1 parent 12d5bbb
commit 7d37185
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 12 deletions.
diff --git a/aim/cli/digifeeds.py b/aim/cli/digifeeds.py
@@ -1,8 +1,15 @@
 import typer
+from aim.digifeeds.database import models, main
 
 app = typer.Typer()
 
 
 @app.command()
 def add_to_db(barcode: str):
     print(f'Adding barcode "{barcode}" to database')
+
+
+@app.command()
+def load_statuses():
+    with main.SessionLocal() as db_session:
+        models.load_statuses(session=db_session)
diff --git a/aim/digifeeds/bin/load_statuses.py b/aim/digifeeds/bin/load_statuses.py