Add raw query response to cursor class

python-pinot-dbapi · Jul 4, 2024 · bc2d6e6 · bc2d6e6
1 parent 556da24
commit bc2d6e6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pinotdb/db.py b/pinotdb/db.py
@@ -314,6 +314,7 @@ def __init__(
         self.schema = None
         self.rowcount = -1
         self._results = None
+        self.raw_query_response = None
         self.timeUsedMs = -1
         self._debug = debug
         self._preserve_types = preserve_types
@@ -395,6 +396,7 @@ def finalize_query_payload(
     def normalize_query_response(self, input_query, query_response):
         try:
             payload = query_response.json()
+            self.raw_query_response = payload
         except Exception as e:
             raise exceptions.DatabaseError(
                 f"Error when querying {input_query} from {self.url}, "