Fix cookie setting (#379)

* fix cookie setting Signed-off-by: Ben Cassell <ben.cassell@databricks.com> * Removing cookie code Signed-off-by: Ben Cassell <ben.cassell@databricks.com> --------- Signed-off-by: Ben Cassell <ben.cassell@databricks.com>
databricks · Mar 27, 2024 · 7fade51 · 7fade51
1 parent 64c71fb
commit 7fade51
Show file tree

Hide file tree

Showing 2 changed files with 424 additions and 264 deletions.
diff --git a/src/databricks/sql/auth/thrift_http_client.py b/src/databricks/sql/auth/thrift_http_client.py
@@ -189,10 +189,6 @@ def flush(self):
         self.message = self.__resp.reason
         self.headers = self.__resp.headers
 
-        # Saves the cookie sent by the server response
-        if "Set-Cookie" in self.headers:
-            self.setCustomHeaders(dict("Cookie", self.headers["Set-Cookie"]))
-
     @staticmethod
     def basic_proxy_auth_header(proxy):
         if proxy is None or not proxy.username: