Commit 070d9083 authored by Jonathan Rohde's avatar Jonathan Rohde
Browse files

feat(sqlalchemy): use subprocess to do migrations

parent bee835cb
...@@ -58,7 +58,7 @@ version_path_separator = os # Use os.pathsep. Default configuration used for ne ...@@ -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]
......
...@@ -173,13 +173,11 @@ https://github.com/open-webui/open-webui ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment