Add support for 'None' on response data

trinodb · Jan 30, 2024 · 0dc4427 · 0dc4427
1 parent ca38d29
commit 0dc4427
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/trino/client.py b/trino/client.py
@@ -633,6 +633,9 @@ def process(self, http_response) -> TrinoStatus:
 
         self._next_uri = response.get("nextUri")
 
+        data = response.get("data", [])
+        if not data:
+            data = []
         return TrinoStatus(
             id=response["id"],
             stats=response["stats"],
@@ -641,7 +644,7 @@ def process(self, http_response) -> TrinoStatus:
             next_uri=self._next_uri,
             update_type=response.get("updateType"),
             update_count=response.get("updateCount"),
-            rows=response.get("data", []),
+            rows=data,
             columns=response.get("columns"),
         )