refac: do not use subprocess

95426fc6 · Timothy J. Baek · 39d3dcd0 · 95426fc6
Commit 95426fc6 authored Jul 08, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 7 deletions

backend/main.py backend/main.py +5 -7

No files found.
--- a/backend/main.py
+++ b/backend/main.py
@@ -173,13 +173,11 @@ https://github.com/open-webui/open-webui


 def run_migrations():
-    env = os.environ.copy()
-    env["DATABASE_URL"] = DATABASE_URL
-    migration_task = subprocess.run(
-        ["alembic", f"-c{BACKEND_DIR}/alembic.ini", "upgrade", "head"], env=env
-    )
-    if migration_task.returncode > 0:
-        raise ValueError("Error running migrations")
+    from alembic.config import Config
+    from alembic import command
+
+    alembic_cfg = Config("alembic.ini")
+    command.upgrade(alembic_cfg, "head")


 @asynccontextmanager