refac: do not use subprocess

95426fc6 · Timothy J. Baek · 39d3dcd0 · 95426fc6
Commit 95426fc6 authored Jul 08, 2024 by Timothy J. Baek
Show 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()
+    from alembic.config import Config
-    env["DATABASE_URL"] = DATABASE_URL
+    from alembic import command
-    migration_task = subprocess.run(
-        ["alembic", f"-c{BACKEND_DIR}/alembic.ini", "upgrade", "head"], env=env
+    alembic_cfg = Config("alembic.ini")
-    )
+    command.upgrade(alembic_cfg, "head")
-    if migration_task.returncode > 0:
-        raise ValueError("Error running migrations")
 @asynccontextmanager