formatting

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/client.py

@@ -253,7 +253,7 @@ def read(self) -> Optional[OAuthToken]:
                 host_url=server_hostname,
                 http_path=http_path,
                 port=self.port,
-                original_exception=e
+                original_exception=e,
             ) from e
 
         self.server_telemetry_enabled = True
@@ -314,7 +314,7 @@ def read(self) -> Optional[OAuthToken]:
                 http_path=http_path,
                 port=self.port,
                 user_agent=useragent_header,
-                original_exception=e
+                original_exception=e,
             ) from e
 
         self._session_handle = self._open_session_resp.sessionHandle

src/databricks/sql/exc.py

@@ -27,20 +27,23 @@ def __init__(
                 session_id_hex
             )
             telemetry_client.export_failure_log(error_name, self.message)
-        elif isinstance(self, (ConnectionError, AuthenticationError)) and 'host_url' in self.context:
+        elif (
+            isinstance(self, (ConnectionError, AuthenticationError))
+            and "host_url" in self.context
+        ):
             # Connection error case: no session but we should still send telemetry
             self._send_connection_error_telemetry(error_name)
 
     def _send_connection_error_telemetry(self, error_name):
         """Send connection error telemetry to unauthenticated endpoint"""
-        
+
         TelemetryClientFactory.send_connection_error_telemetry(
             error_name=error_name,
             error_message=self.message or str(self),
-            host_url=self.context['host_url'],
-            http_path=self.context.get('http_path', ''),
-            port=self.context.get('port', 443),
-            user_agent=self.context.get('user_agent'),
+            host_url=self.context["host_url"],
+            http_path=self.context.get("http_path", ""),
+            port=self.context.get("port", 443),
+            user_agent=self.context.get("user_agent"),
         )
 
     def __str__(self):
@@ -146,38 +149,42 @@ class CursorAlreadyClosedError(RequestError):
 
 class ConnectionError(OperationalError):
     """Thrown when connection to Databricks fails during initial setup"""
-    
+
     def __init__(
-        self, 
-        message=None, 
-        host_url=None, 
-        http_path=None, 
+        self,
+        message=None,
+        host_url=None,
+        http_path=None,
         port=443,
         user_agent=None,
         original_exception=None,
         **kwargs
     ):
         # Set up context for connection error telemetry
-        context = kwargs.get('context', {})
+        context = kwargs.get("context", {})
         if host_url:
-            context.update({
-                'host_url': host_url,
-                'http_path': http_path or '',
-                'port': port,
-                'user_agent': user_agent,
-                'original_exception': str(original_exception) if original_exception else None,
-            })
-        
+            context.update(
+                {
+                    "host_url": host_url,
+                    "http_path": http_path or "",
+                    "port": port,
+                    "user_agent": user_agent,
+                    "original_exception": str(original_exception)
+                    if original_exception
+                    else None,
+                }
+            )
+
         super().__init__(message=message, context=context, **kwargs)
 
 
 class AuthenticationError(ConnectionError):
     """Thrown when authentication to Databricks fails"""
-    
+
     def __init__(self, message=None, auth_type=None, **kwargs):
-        context = kwargs.get('context', {})
+        context = kwargs.get("context", {})
         if auth_type:
-            context['auth_type'] = auth_type
-        kwargs['context'] = context
-        
+            context["auth_type"] = auth_type
+        kwargs["context"] = context
+
         super().__init__(message=message, **kwargs)

src/databricks/sql/telemetry/telemetry_client.py

@@ -18,7 +18,11 @@
     FrontendLogContext,
     FrontendLogEntry,
 )
-from databricks.sql.telemetry.models.enums import AuthMech, AuthFlow, DatabricksClientType
+from databricks.sql.telemetry.models.enums import (
+    AuthMech,
+    AuthFlow,
+    DatabricksClientType,
+)
 from databricks.sql.auth.authenticators import (
     AccessTokenAuthProvider,
     DatabricksOAuthProvider,
@@ -462,8 +466,7 @@ def send_connection_error_telemetry(
             )
 
             error_info = DriverErrorInfo(
-                error_name=error_name, 
-                stack_trace=error_message
+                error_name=error_name, stack_trace=error_message
             )
 
             telemetry_frontend_log = TelemetryFrontendLog(
@@ -512,4 +515,4 @@ def send_connection_error_telemetry(
                 logger.debug("Failed to send connection error telemetry: %s", e)
 
         except Exception as e:
-            logger.debug("Failed to create connection error telemetry: %s", e)
+            logger.debug("Failed to create connection error telemetry: %s", e)