From c80f609c418391490c9c608eaa1006eb66e4e9f8 Mon Sep 17 00:00:00 2001
From: Gaisberg <None>
Date: Tue, 5 Nov 2024 15:19:53 +0200
Subject: [PATCH] chore: hotfix for db_migration merge

---
 src/program/db/db_functions.py | 2 +-
 src/program/utils/cli.py       | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/program/db/db_functions.py b/src/program/db/db_functions.py
index b05b74a0..b49a6f52 100644
--- a/src/program/db/db_functions.py
+++ b/src/program/db/db_functions.py
@@ -7,7 +7,7 @@
 from sqlalchemy import delete, desc, func, insert, inspect, select, text
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.orm import Session, joinedload, selectinload
-from utils import root_dir
+from program.utils import root_dir
 
 import alembic
 from program.media.stream import Stream, StreamBlacklistRelation, StreamRelation
diff --git a/src/program/utils/cli.py b/src/program/utils/cli.py
index 73d90dc7..f827e9b2 100644
--- a/src/program/utils/cli.py
+++ b/src/program/utils/cli.py
@@ -18,6 +18,11 @@ def handle_args():
         action="store_true",
         help="Hard reset the database.",
     )
+    parser.add_argument(
+        "--hard_reset_db_pre_migration",
+        action="store_true",
+        help="Hard reset the database.",
+    )
     parser.add_argument(
         "--clean_logs",
         action="store_true",