UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-04 21:03:18 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge pull request #4768 from zhaofengli/db-url-fixes

Browse Source 
fix: Fix parameter handling for PostgreSQL URLs
This commit is contained in:
Timothy Jaeryang Baek 2024-08-22 14:01:28 +02:00 committed by GitHub
commit 11aecbe79a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/apps/webui/internal/wrappers.py
View File

@ -43,7 +43,7 @@ class ReconnectingPostgresqlDatabase(CustomReconnectMixin, PostgresqlDatabase):
def register_connection(db_url): def register_connection(db_url):
db = connect(db_url) db = connect(db_url, unquote_password=True)
if isinstance(db, PostgresqlDatabase): if isinstance(db, PostgresqlDatabase):
# Enable autoconnect for SQLite databases, managed by Peewee # Enable autoconnect for SQLite databases, managed by Peewee
db.autoconnect = True db.autoconnect = True
@ -51,16 +51,10 @@ def register_connection(db_url):
log.info("Connected to PostgreSQL database") log.info("Connected to PostgreSQL database")
# Get the connection details # Get the connection details
connection = parse(db_url) connection = parse(db_url, unquote_password=True)
# Use our custom database class that supports reconnection # Use our custom database class that supports reconnection
db = ReconnectingPostgresqlDatabase( db = ReconnectingPostgresqlDatabase(**connection)
connection["database"],
user=connection["user"],
password=connection["password"],
host=connection["host"],
port=connection["port"],
)
db.connect(reuse_if_open=True) db.connect(reuse_if_open=True)
elif isinstance(db, SqliteDatabase): elif isinstance(db, SqliteDatabase):
# Enable autoconnect for SQLite databases, managed by Peewee # Enable autoconnect for SQLite databases, managed by Peewee