OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-25 15:15:28 +00:00
Code Issues Actions 4 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/alembic.ini
View File

@ -58,7 +58,7 @@ version_path_separator = os # Use os.pathsep. Default configuration used for ne
# are written from script.py.mako # are written from script.py.mako
# output_encoding = utf-8 # output_encoding = utf-8
sqlalchemy.url = REPLACE_WITH_DATABASE_URL # sqlalchemy.url = REPLACE_WITH_DATABASE_URL
[post_write_hooks] [post_write_hooks]

12
backend/main.py
View File

@ -173,13 +173,11 @@ https://github.com/open-webui/open-webui
def run_migrations(): def run_migrations():
from alembic.config import Config env = os.environ.copy()
from alembic import command env["DATABASE_URL"] = DATABASE_URL
migration_task = subprocess.run(["alembic", f"-c{BACKEND_DIR}/alembic.ini", "upgrade", "head"], env=env)
alembic_cfg = Config(f"{BACKEND_DIR}/alembic.ini") if migration_task.returncode > 0:
alembic_cfg.set_main_option("sqlalchemy.url", DATABASE_URL) raise ValueError("Error running migrations")
alembic_cfg.set_main_option("script_location", f"{BACKEND_DIR}/migrations")
command.upgrade(alembic_cfg, "head")
@asynccontextmanager @asynccontextmanager