Add raw response to cursor class

python-pinot-dbapi · Jul 4, 2024 · 4e81b23 · 4e81b23
1 parent bc2d6e6
commit 4e81b23
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pinotdb/db.py b/pinotdb/db.py
@@ -396,8 +396,15 @@ def finalize_query_payload(
     def normalize_query_response(self, input_query, query_response):
         try:
             payload = query_response.json()
-            self.raw_query_response = payload
+            self.raw_query_response = {
+                "response" : payload, 
+                "response_status" : query_response.status_code
+                }
         except Exception as e:
+            self.raw_query_response = {
+                "response" : query_response.text, 
+                "response_status" : query_response.status_code
+                }
             raise exceptions.DatabaseError(
                 f"Error when querying {input_query} from {self.url}, "
                 f"raw response is:\n{query_response.text}"