From 0483a7978205711ba8833337dfd5b690826b187a Mon Sep 17 00:00:00 2001 From: disheng Date: Sun, 29 Dec 2024 10:18:47 +0800 Subject: [PATCH] Added SSL connection info for MySQL --- ibis-server/app/model/__init__.py | 3 +++ ibis-server/app/model/data_source.py | 1 + 2 files changed, 4 insertions(+) diff --git a/ibis-server/app/model/__init__.py b/ibis-server/app/model/__init__.py index 3798e690f..98e3a081f 100644 --- a/ibis-server/app/model/__init__.py +++ b/ibis-server/app/model/__init__.py @@ -95,6 +95,9 @@ class MySqlConnectionInfo(BaseModel): database: SecretStr user: SecretStr password: SecretStr + kwargs: dict[str, str] | None = Field( + description="Additional keyword arguments to pass to PyMySQL", default=None + ) class ConnectionUrl(BaseModel): diff --git a/ibis-server/app/model/data_source.py b/ibis-server/app/model/data_source.py index ea271f6aa..ba282e973 100644 --- a/ibis-server/app/model/data_source.py +++ b/ibis-server/app/model/data_source.py @@ -131,6 +131,7 @@ def get_mysql_connection(info: MySqlConnectionInfo) -> BaseBackend: database=info.database.get_secret_value(), user=info.user.get_secret_value(), password=info.password.get_secret_value(), + **info.kwargs if info.kwargs else dict(), ) @staticmethod