fix: alembic

4b6ee584 · Timothy J. Baek · d3ef3a74 · 4b6ee584 · 4b6ee584
Commit 4b6ee584 authored Jul 08, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 5 deletions

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

backend/migrations/env.py backend/migrations/env.py +1 -1

No files found.
--- a/backend/main.py
+++ b/backend/main.py
@@ -173,11 +173,14 @@ https://github.com/open-webui/open-webui
 def run_migrations():
-    from alembic.config import Config
+    try:
-    from alembic import command
+        from alembic.config import Config
+        from alembic import command
-    alembic_cfg = Config("alembic.ini")
+        alembic_cfg = Config("alembic.ini")
-    command.upgrade(alembic_cfg, "head")
+        command.upgrade(alembic_cfg, "head")
+    except Exception as e:
+        print(f"Error: {e}")
 @asynccontextmanager

--- a/backend/migrations/env.py
+++ b/backend/migrations/env.py
@@ -43,7 +43,7 @@ target_metadata = Auth.metadata
 DB_URL = DATABASE_URL
 if DB_URL:
-    config.set_main_option("sqlalchemy.url", DB_URL)
+    config.set_main_option("sqlalchemy.url", DB_URL.replace("%", "%%"))
 def run_migrations_offline() -> None: