Improve sql ssl options

dymmond · Nov 29, 2023 · 3b9c922 · 3b9c922
1 parent 1a02288
commit 3b9c922
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/databasez/backends/aiopg.py b/databasez/backends/aiopg.py
@@ -56,7 +56,7 @@ def _get_connection_kwargs(self) -> dict:
         if max_size is not None:
             kwargs["maxsize"] = int(max_size)
         if ssl is not None:
-            kwargs["ssl"] = {"true": True, "false": False}[ssl.lower()]
+            kwargs["ssl"] = {"true": True, "false": False}.get(ssl.lower(), ssl.lower())
 
         for key, value in self._options.items():
             # Coerce 'min_size' and 'max_size' for consistency.

diff --git a/databasez/backends/asyncmy.py b/databasez/backends/asyncmy.py
@@ -48,7 +48,7 @@ def _get_connection_kwargs(self) -> dict:
         if pool_recycle is not None:
             kwargs["pool_recycle"] = int(pool_recycle)
         if ssl is not None:
-            kwargs["ssl"] = {"true": True, "false": False}[ssl.lower()]
+            kwargs["ssl"] = {"true": True, "false": False}.get(ssl.lower(), ssl.lower())
 
         for key, value in self._options.items():
             # Coerce 'min_size' and 'max_size' for consistency.

diff --git a/databasez/backends/mssql.py b/databasez/backends/mssql.py
@@ -51,7 +51,7 @@ def _get_connection_kwargs(self) -> dict:
         if pool_recycle is not None:
             kwargs["pool_recycle"] = int(pool_recycle)
         if ssl is not None:
-            kwargs["ssl"] = {"true": True, "false": False}[ssl.lower()]
+            kwargs["ssl"] = {"true": True, "false": False}.get(ssl.lower(), ssl.lower())
 
         kwargs.update(
             {

diff --git a/databasez/backends/mysql.py b/databasez/backends/mysql.py
@@ -48,7 +48,7 @@ def _get_connection_kwargs(self) -> dict:
         if pool_recycle is not None:
             kwargs["pool_recycle"] = int(pool_recycle)
         if ssl is not None:
-            kwargs["ssl"] = {"true": True, "false": False}[ssl.lower()]
+            kwargs["ssl"] = {"true": True, "false": False}.get(ssl.lower(), ssl.lower())
 
         for key, value in self._options.items():
             # Coerce 'min_size' and 'max_size' for consistency.

diff --git a/databasez/backends/postgres.py b/databasez/backends/postgres.py
@@ -51,7 +51,7 @@ def _get_connection_kwargs(self) -> dict:
         if max_size is not None:
             kwargs["max_size"] = int(max_size)
         if ssl is not None:
-            kwargs["ssl"] = {"true": True, "false": False}[ssl.lower()]
+            kwargs["ssl"] = {"true": True, "false": False}.get(ssl.lower(), ssl.lower())
 
         kwargs.update(self._options)