move staging allowed local path to connection props

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

src/databricks/sql/backend/databricks_client.py

@@ -334,30 +334,6 @@ def get_columns(
         """
         pass
 
-    # == Properties ==
-    @property
-    @abstractmethod
-    def staging_allowed_local_path(self) -> Union[None, str, List[str]]:
-        """
-        Gets the allowed local paths for staging operations.
-
-        Returns:
-            Union[None, str, List[str]]: The allowed local paths for staging operations,
-            or None if staging is not allowed
-        """
-        pass
-
-    @property
-    @abstractmethod
-    def ssl_options(self) -> SSLOptions:
-        """
-        Gets the SSL options for this client.
-
-        Returns:
-            SSLOptions: The SSL configuration options
-        """
-        pass
-
     @property
     @abstractmethod
     def max_download_threads(self) -> int:

src/databricks/sql/backend/thrift_backend.py

@@ -101,7 +101,6 @@ def __init__(
         http_headers,
         auth_provider: AuthProvider,
         ssl_options: SSLOptions,
-        staging_allowed_local_path: Union[None, str, List[str]] = None,
         **kwargs,
     ):
         # Internal arguments in **kwargs:
@@ -160,7 +159,6 @@ def __init__(
         else:
             raise ValueError("No valid connection settings.")
 
-        self._staging_allowed_local_path = staging_allowed_local_path
         self._initialize_retry_args(kwargs)
         self._use_arrow_native_complex_types = kwargs.get(
             "_use_arrow_native_complex_types", True
@@ -234,14 +232,6 @@ def __init__(
 
         self._request_lock = threading.RLock()
 
-    @property
-    def staging_allowed_local_path(self) -> Union[None, str, List[str]]:
-        return self._staging_allowed_local_path
-
-    @property
-    def ssl_options(self) -> SSLOptions:
-        return self._ssl_options
-
     @property
     def max_download_threads(self) -> int:
         return self._max_download_threads

src/databricks/sql/client.py

@@ -247,6 +247,7 @@ def read(self) -> Optional[OAuthToken]:
         self.use_inline_params = self._set_use_inline_params_with_warning(
             kwargs.get("use_inline_params", False)
         )
+        self.staging_allowed_local_path = kwargs.get("staging_allowed_local_path", None)
 
     def _set_use_inline_params_with_warning(self, value: Union[bool, str]):
         """Valid values are True, False, and "silent"
@@ -793,7 +794,7 @@ def execute(
 
         if execute_response.is_staging_operation:
             self._handle_staging_operation(
-                staging_allowed_local_path=self.backend.staging_allowed_local_path
+                staging_allowed_local_path=self.connection.staging_allowed_local_path
             )
 
         return self
@@ -900,7 +901,7 @@ def get_async_execution_result(self):
 
             if execute_response.is_staging_operation:
                 self._handle_staging_operation(
-                    staging_allowed_local_path=self.backend.staging_allowed_local_path
+                    staging_allowed_local_path=self.connection.staging_allowed_local_path
                 )
 
             return self