Added SSL connection info for MySQL

Canner · Dec 29, 2024 · 0483a79 · 0483a79
1 parent 441743e
commit 0483a79
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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