Up to date with trino-connection-error

trinodb · Aug 23, 2023 · d716d01 · d716d01
1 parent 1f5cdff
commit d716d01
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/trino/client.py b/trino/client.py
@@ -787,7 +787,11 @@ def execute(self, additional_http_headers=None) -> TrinoResult:
         if self.cancelled:
             raise exceptions.TrinoUserError("Query has been cancelled", self.query_id)
 
-        response = self._request.post(self._query, additional_http_headers)
+        try:
+            response = self._request.post(self._query, additional_http_headers)
+        except requests.exceptions.RequestException as e:
+            raise trino.exceptions.TrinoConnectionError("failed to execute: {}".format(e))
+
         status = self._request.process(response)
         self._info_uri = status.info_uri
         self._query_id = status.id