allow DELETE in retries

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

src/databricks/sql/auth/retry.py

@@ -127,7 +127,7 @@ def __init__(
             total=_attempts_remaining,
             respect_retry_after_header=True,
             backoff_factor=self.delay_min,
-            allowed_methods=["POST"],
+            allowed_methods=["POST", "DELETE"],
             status_forcelist=[429, 503, *self.force_dangerous_codes],
         )
 

src/databricks/sql/backend/sea/backend.py

@@ -180,7 +180,14 @@ def __init__(
         # Extract retry policy parameters
         self._initialize_retry_args(kwargs)
         self._auth_provider = auth_provider
+
         self.enable_v3_retries = kwargs.get("_enable_v3_retries", True)
+        if not self.enable_v3_retries:
+            logger.warning(
+                "Legacy retry behavior is enabled for this connection."
+                " This behaviour is deprecated and will be removed in a future release."
+            )
+
         self.force_dangerous_codes = kwargs.get("_retry_dangerous_codes", [])
 
         additional_transport_args = {}

src/databricks/sql/backend/sea/utils/http_client_adapter.py

@@ -92,10 +92,6 @@ def get(
         Returns:
             Response data parsed from JSON
         """
-        # Set the command type for retry policy
-        command_type = self._determine_command_type(path, "GET", None)
-        self.thrift_client.set_retry_command_type(command_type)
-
         return self.thrift_client.make_rest_request(
             "GET", path, params=params, headers=headers
         )
@@ -119,10 +115,6 @@ def post(
         Returns:
             Response data parsed from JSON
         """
-        # Set the command type for retry policy
-        command_type = self._determine_command_type(path, "POST", data)
-        self.thrift_client.set_retry_command_type(command_type)
-
         response = self.thrift_client.make_rest_request(
             "POST", path, data=data, params=params, headers=headers
         )
@@ -147,10 +139,6 @@ def delete(
         Returns:
             Response data parsed from JSON
         """
-        # Set the command type for retry policy
-        command_type = self._determine_command_type(path, "DELETE", data)
-        self.thrift_client.set_retry_command_type(command_type)
-
         return self.thrift_client.make_rest_request(
             "DELETE", path, data=data, params=params, headers=headers
         )