Merge pull request #914 from Capsize-Games/devastator

fixes alembic paths
Capsize-Games · Oct 8, 2024 · adef172 · adef172
2 parents 278213e + d89f9f1
commit adef172
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="airunner",
-    version="3.0.7",
+    version="3.0.8",
     author="Capsize LLC",
     description="A Stable Diffusion GUI",
     long_description=open("README.md", "r", encoding="utf-8").read(),

diff --git a/src/airunner/main.py b/src/airunner/main.py
@@ -45,7 +45,14 @@
 from alembic import command
 
 def setup_database():
-    alembic_cfg = Config("alembic.ini")
+    here = os.path.dirname(os.path.abspath(__file__))
+    alembic_file = os.path.join(
+        here,
+        "alembic.ini"
+    )
+    alembic_cfg = Config(alembic_file)
+    # set he script_location to the current directory
+    alembic_cfg.set_main_option("script_location", os.path.join(here, "alembic"))
     command.upgrade(alembic_cfg, "head")