fix: add alembic reinit to hard reset

rivenmedia · Jul 27, 2024 · 91ba58b · 91ba58b
1 parent e3366a6
commit 91ba58b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/makefile b/makefile
@@ -80,6 +80,9 @@ clean:
 	@rm -rf data/alembic/
 	@rm -rf data/*.db
 
+hard_reset:
+	@poetry run python src/main.py --hard_reset_db
+
 install:
 	@poetry install --with dev
 

diff --git a/src/program/db/db_functions.py b/src/program/db/db_functions.py
@@ -12,7 +12,7 @@
 from utils.logger import logger
 from utils import alembic_dir
 
-from .db import db
+from .db import db, alembic
 
 
 def _ensure_item_exists_in_db(item: MediaItem) -> bool:
@@ -143,6 +143,8 @@ def hard_reset_database():
     logger.debug("Removing Alembic Directory")
     shutil.rmtree(alembic_dir, ignore_errors=True)
     os.makedirs(alembic_dir, exist_ok=True)
+    alembic.init(alembic_dir)
+    logger.debug("Alembic reinitialized")
 
     logger.debug("Hard Reset Complete")