[SNOW-812398] Add ability to return raw response json from python-connector (#1556)

* [SNOW-812398] Add ability to return raw response json from python-connector.

Author: sfc-gh-skumbham

src/snowflake/connector/cursor.py

@@ -264,6 +264,7 @@ def __init__(
         self._inner_cursor: SnowflakeCursor | None = None
         self._prefetch_hook = None
         self._rownumber: int | None = None
+        self._cached_last_result: dict | None = None
 
         self.reset()
 
@@ -542,6 +543,8 @@ def interrupt_handler(*_):  # pragma: no cover
                 self._timebomb.cancel()
                 logger.debug("cancelled timebomb in finally")
 
+            self._cached_last_result = ret
+
         if "data" in ret and "parameters" in ret["data"]:
             parameters = ret["data"]["parameters"]
             # Set session parameters for cursor object
@@ -1073,6 +1076,7 @@ def query_result(self, qid: str) -> SnowflakeCursor:
             Error.errorhandler_wrapper(
                 self.connection, self, ProgrammingError, errvalue
             )
+        self._cached_last_result = ret
         return self
 
     def fetch_arrow_batches(self) -> Iterator[Table]:
@@ -1122,6 +1126,7 @@ def fetch_pandas_all(self, **kwargs: Any) -> DataFrame:
     def abort_query(self, qid: str) -> bool:
         url = f"/queries/{qid}/abort-request"
         ret = self._connection.rest.request(url=url, method="post")
+        self._cached_last_result = ret
         return ret.get("success")
 
     def executemany(
@@ -1457,6 +1462,7 @@ def wait_until_ready() -> None:
             ):
                 url = f"/queries/{sfqid}/result"
                 ret = self._connection.rest.request(url=url, method="get")
+                self._cached_last_result = ret
                 if "data" in ret and "resultIds" in ret["data"]:
                     self._init_multi_statement_results(ret["data"])
 

test/integ/test_cursor.py

@@ -1088,6 +1088,34 @@ def test_rownumber(conn):
             assert cur.rownumber == 1
 
 
+def test_last_result(conn):
+    """Checks whether get_last_request_json is returned as expected."""
+    with conn() as cnx:
+        with cnx.cursor() as cur:
+            assert cur.execute("select * from values (1), (2)")
+            assert cur.rownumber is None
+            return_val = cur._cached_last_result
+
+            assert "data" in return_val
+            assert "code" in return_val
+            assert return_val["success"] is True
+            assert "message" in return_val
+
+
+def test_last_result_for_get_results_api(conn):
+    """Checks whether get_last_request_json for get_results_from_sfqid is returned as expected."""
+    with conn() as cnx:
+        with cnx.cursor() as cur:
+            assert cur.execute("select * from values (4), (5)")
+            cur.get_results_from_sfqid(cur.sfqid)
+            return_val = cur._cached_last_result
+
+            assert "data" in return_val
+            assert "code" in return_val
+            assert return_val["success"] is True
+            assert "message" in return_val
+
+
 def test_values_set(conn):
     """Checks whether a bunch of properties start as Nones, but get set to something else when a query was executed."""
     properties = [