Expose verify parameter on dbapi.connect

This resolves #29
trinodb · Jun 26, 2020 · 836ed92 · 836ed92
1 parent 58c7940
commit 836ed92
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/presto/client.py b/presto/client.py
@@ -209,6 +209,7 @@ def __init__(
         max_attempts=MAX_ATTEMPTS,  # type: int
         request_timeout=constants.DEFAULT_REQUEST_TIMEOUT,  # type: Union[float, Tuple[float, float]]
         handle_retry=exceptions.RetryWithExponentialBackoff(),
+        verify=True     # type: Any
     ):
         # type: (...) -> None
         self._client_session = ClientSession(
@@ -230,6 +231,7 @@ def __init__(
         else:
             # mypy cannot follow module import
             self._http_session = self.http.Session()  # type: ignore
+            self._http_session.verify = verify
         self._http_session.headers.update(self.http_headers)
         self._exceptions = self.HTTP_EXCEPTIONS
         self._auth = auth

diff --git a/presto/dbapi.py b/presto/dbapi.py
@@ -75,6 +75,7 @@ def __init__(
         max_attempts=constants.DEFAULT_MAX_ATTEMPTS,
         request_timeout=constants.DEFAULT_REQUEST_TIMEOUT,
         isolation_level=IsolationLevel.AUTOCOMMIT,
+        verify=True
     ):
         self.host = host
         self.port = port
@@ -85,6 +86,7 @@ def __init__(
         self.session_properties = session_properties
         # mypy cannot follow module import
         self._http_session = presto.client.PrestoRequest.http.Session()
+        self._http_session.verify = verify
         self.http_headers = http_headers
         self.http_scheme = http_scheme
         self.auth = auth