fix: alembic upgrade head path is now independent of working dir

Author: MiNeves00

libs/tracker/llmstudio_tracker/db/migrations/utils.py

@@ -1,61 +1,24 @@
+import logging
 import subprocess
+from pathlib import Path
 
+from alembic import command
+from alembic.config import Config
 
-def run_alembic_upgrade():
-    try:
-        if _alembic_is_up_to_date():
-            print("Alembic: DB is already up-to-date.")
-            return
-
-        print("Alembic: DB is not up-to-date. Running Alembic upgrade...")
-        subprocess.run(["poetry", "run", "alembic", "upgrade", "head"], check=True)
-        print("Alembic: Upgrade successful.")
-    except subprocess.CalledProcessError as e:
-        print(f"Alembic: Upgrade failed: {e}")
-        raise
+logging.basicConfig(level=logging.INFO)
 
 
-def _alembic_is_up_to_date() -> bool:
-    """Returns True if the DB is already at the latest revision."""
+def run_alembic_upgrade():
     try:
-        current = (
-            subprocess.check_output(
-                ["poetry", "run", "alembic", "current"], stderr=subprocess.DEVNULL
-            )
-            .decode()
-            .strip()
-        )
-
-        head = (
-            subprocess.check_output(
-                ["poetry", "run", "alembic", "heads"], stderr=subprocess.DEVNULL
-            )
-            .decode()
-            .strip()
-        )
+        logging.info("Running LLMstudio Tracker Migrations Alembic upgrade...")
 
-        current_rev_line = next(
-            (
-                line
-                for line in current.splitlines()
-                if line.strip() and not line.startswith("DB URL")
-            ),
-            "",
-        )
-        head_rev_line = next(
-            (
-                line
-                for line in head.splitlines()
-                if line.strip() and not line.startswith("DB URL")
-            ),
-            "",
-        )
+        alembic_cfg = Config()
+        this_dir = Path(__file__).resolve().parent
+        alembic_cfg.set_main_option("script_location", str(this_dir))
 
-        current_rev = current_rev_line.split(" ")[0]
-        head_rev = head_rev_line.split(" ")[0]
-
-        return current_rev == head_rev
+        command.upgrade(alembic_cfg, "head")
 
+        logging.info("Alembic: LLMstudio Tracker Migrations upgrade successful.")
     except subprocess.CalledProcessError as e:
-        print("Alembic: Check if up-to-date failed. Upgrading head to be safe.")
-        return False
+        logging.error(f"Alembic: LLMstudio Tracker Migrations upgrade failed: {e}")
+        raise