Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 20:07:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat(sqlalchemy): use subprocess to do migrations

Browse Source
This commit is contained in:
Jonathan Rohde
2024-06-24 09:50:14 +02:00
parent bee835cb65
commit 070d9083d5
2 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
backend/main.py
View File

@@ -173,13 +173,11 @@ https://github.com/open-webui/open-webui
def run_migrations():
from alembic.config import Config
from alembic import command
alembic_cfg = Config(f"{BACKEND_DIR}/alembic.ini")
alembic_cfg.set_main_option("sqlalchemy.url", DATABASE_URL)
alembic_cfg.set_main_option("script_location", f"{BACKEND_DIR}/migrations")
command.upgrade(alembic_cfg, "head")
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")
@asynccontextmanager