remove excess logs, assertions, instantiations

large merge artifacts

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

Author: varun-edachali-dbx

src/databricks/sql/backend/thrift_backend.py

@@ -946,24 +946,6 @@ def execute_command(
             thrift_handle,
         )
 
-        logger.debug(
-            "ThriftBackend.execute_command(operation=%s, session_handle=%s)",
-            operation,
-            thrift_handle,
-        )
-
-        logger.debug(
-            "ThriftBackend.execute_command(operation=%s, session_handle=%s)",
-            operation,
-            session_handle,
-        )
-
-        logger.debug(
-            "ThriftBackend.execute_command(operation=%s, session_handle=%s)",
-            operation,
-            session_handle,
-        )
-
         spark_arrow_types = ttypes.TSparkArrowTypes(
             timestampAsArrow=self._use_arrow_native_timestamps,
             decimalAsArrow=self._use_arrow_native_decimals,

src/databricks/sql/client.py

@@ -232,7 +232,6 @@ def read(self) -> Optional[OAuthToken]:
         self.use_cloud_fetch = kwargs.get("use_cloud_fetch", True)
         self._cursors = []  # type: List[Cursor]
 
-        # Create the session
         self.session = Session(
             server_hostname,
             http_path,
@@ -245,23 +244,6 @@ def read(self) -> Optional[OAuthToken]:
         )
         self.session.open()
 
-        logger.info(
-            "Successfully opened connection with session "
-            + str(self.get_session_id_hex())
-        )
-
-        # Create the session
-        self.session = Session(
-            server_hostname,
-            http_path,
-            http_headers,
-            session_configuration,
-            catalog,
-            schema,
-            _use_arrow_native_complex_types,
-            **kwargs
-        )
-
         self.use_inline_params = self._set_use_inline_params_with_warning(
             kwargs.get("use_inline_params", False)
         )

src/databricks/sql/session.py

@@ -84,6 +84,8 @@ def __init__(
             **kwargs,
         )
 
+        self.protocol_version = None 
+
     def open(self):
         self._session_id = self.backend.open_session(
             session_configuration=self.session_configuration,

tests/unit/test_client.py

@@ -218,15 +218,6 @@ def test_context_manager_closes_cursor(self):
         finally:
             cursor.close.assert_called()
 
-        connection = databricks.sql.connect(**self.DUMMY_CONNECTION_ARGS)
-        connection.close = Mock()
-        try:
-            with self.assertRaises(KeyboardInterrupt):
-                with connection:
-                    raise KeyboardInterrupt("Simulated interrupt")
-        finally:
-            connection.close.assert_called()
-
     def dict_product(self, dicts):
         """
         Generate cartesion product of values in input dictionary, outputting a dictionary