OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-23 02:16:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-08-22 16:11:19 +02:00
parent d8d2f3529f
commit 673b893a8a
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/apps/webui/internal/db.py
View File

@ -55,17 +55,15 @@ else:
# This is required to ensure the peewee migration is handled before the alembic migration # This is required to ensure the peewee migration is handled before the alembic migration
def handle_peewee_migration(DATABASE_URL): def handle_peewee_migration(DATABASE_URL):
try: try:
# Replace the postgresql:// with postgres:// and %40 with @ in the DATABASE_URL # Replace the postgresql:// with postgres:// to handle the peewee migration
db = register_connection( db = register_connection(
DATABASE_URL.replace("postgresql://", "postgres://").replace("%40", "@") DATABASE_URL.replace("postgresql://", "postgres://"), unquote_password=True
) )
migrate_dir = BACKEND_DIR / "apps" / "webui" / "internal" / "migrations" migrate_dir = BACKEND_DIR / "apps" / "webui" / "internal" / "migrations"
router = Router(db, logger=log, migrate_dir=migrate_dir) router = Router(db, logger=log, migrate_dir=migrate_dir)
router.run() router.run()
db.close() db.close()
# check if db connection has been closed
except Exception as e: except Exception as e:
log.error(f"Failed to initialize the database connection: {e}") log.error(f"Failed to initialize the database connection: {e}")
raise raise