formatting

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

Author: saishreeeee

src/databricks/sql/client.py

@@ -49,7 +49,10 @@
     TSparkParameter,
     TOperationState,
 )
-from databricks.sql.telemetry.telemetry_client import telemetry_client, NoopTelemetryClient
+from databricks.sql.telemetry.telemetry_client import (
+    telemetry_client,
+    NoopTelemetryClient,
+)
 from databricks.sql.telemetry.latency_logger import log_latency
 from databricks.sql.telemetry.models.enums import DriverVolumeOperationType
 
@@ -240,7 +243,9 @@ def read(self) -> Optional[OAuthToken]:
         self.telemetry_enabled = (
             self.client_telemetry_enabled and self.server_telemetry_enabled
         )
-        telemetry_batch_size = kwargs.get("telemetry_batch_size", 100) # TODO: Decide on batch size
+        telemetry_batch_size = kwargs.get(
+            "telemetry_batch_size", 100
+        )  # TODO: Decide on batch size
 
         user_agent_entry = kwargs.get("user_agent_entry")
         if user_agent_entry is None:
@@ -302,16 +307,16 @@ def read(self) -> Optional[OAuthToken]:
                 host=self.host,
                 connection_uuid=self.get_session_id_hex(),
                 auth_provider=auth_provider,
-                is_authenticated=True, # TODO: Add authentication logic later
+                is_authenticated=True,  # TODO: Add authentication logic later
                 batch_size=telemetry_batch_size,
                 user_agent=useragent_header,
             )
-            
+
             telemetry_client.export_initial_telemetry_log(
-                http_path, 
-                self.port, 
+                http_path,
+                self.port,
                 kwargs.get("_socket_timeout", None),
-                self.get_session_id_hex()
+                self.get_session_id_hex(),
             )
         else:
             self.telemetry_client = NoopTelemetryClient()
@@ -512,7 +517,10 @@ def __iter__(self):
             for row in self.active_result_set:
                 yield row
         else:
-            raise Error("There is no active result set", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "There is no active result set",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
     def _determine_parameter_approach(
         self, params: Optional[TParameterCollection]
@@ -649,7 +657,10 @@ def _close_and_clear_active_result_set(self):
 
     def _check_not_closed(self):
         if not self.open:
-            raise Error("Attempting operation on closed cursor", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "Attempting operation on closed cursor",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
     def _handle_staging_operation(
         self, staging_allowed_local_path: Union[None, str, List[str]]
@@ -668,7 +679,7 @@ def _handle_staging_operation(
         else:
             raise Error(
                 "You must provide at least one staging_allowed_local_path when initialising a connection to perform ingestion commands",
-                connection_uuid=self.connection.get_session_id_hex()
+                connection_uuid=self.connection.get_session_id_hex(),
             )
 
         abs_staging_allowed_local_paths = [
@@ -698,7 +709,7 @@ def _handle_staging_operation(
             if not allow_operation:
                 raise Error(
                     "Local file operations are restricted to paths within the configured staging_allowed_local_path",
-                    connection_uuid=self.connection.get_session_id_hex()
+                    connection_uuid=self.connection.get_session_id_hex(),
                 )
 
         # May be real headers, or could be json string
@@ -729,7 +740,7 @@ def _handle_staging_operation(
             raise Error(
                 f"Operation {row.operation} is not supported. "
                 + "Supported operations are GET, PUT, and REMOVE",
-                connection_uuid=self.connection.get_session_id_hex()
+                connection_uuid=self.connection.get_session_id_hex(),
             )
 
     @log_latency()
@@ -742,7 +753,10 @@ def _handle_staging_put(
         """
 
         if local_file is None:
-            raise Error("Cannot perform PUT without specifying a local_file", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "Cannot perform PUT without specifying a local_file",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
         self.volume_operation_type = DriverVolumeOperationType.PUT
         self.volume_path = local_file
@@ -762,7 +776,8 @@ def _handle_staging_put(
 
         if r.status_code not in [OK, CREATED, NO_CONTENT, ACCEPTED]:
             raise Error(
-                f"Staging operation over HTTP was unsuccessful: {r.status_code}-{r.text}", connection_uuid=self.connection.get_session_id_hex()
+                f"Staging operation over HTTP was unsuccessful: {r.status_code}-{r.text}",
+                connection_uuid=self.connection.get_session_id_hex(),
             )
 
         if r.status_code == ACCEPTED:
@@ -781,7 +796,10 @@ def _handle_staging_get(
         """
 
         if local_file is None:
-            raise Error("Cannot perform GET without specifying a local_file", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "Cannot perform GET without specifying a local_file",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
         self.volume_operation_type = DriverVolumeOperationType.GET
         self.volume_path = local_file
@@ -792,7 +810,8 @@ def _handle_staging_get(
         # Any 2xx or 3xx will evaluate r.ok == True
         if not r.ok:
             raise Error(
-                f"Staging operation over HTTP was unsuccessful: {r.status_code}-{r.text}", connection_uuid=self.connection.get_session_id_hex()
+                f"Staging operation over HTTP was unsuccessful: {r.status_code}-{r.text}",
+                connection_uuid=self.connection.get_session_id_hex(),
             )
 
         with open(local_file, "wb") as fp:
@@ -805,7 +824,9 @@ def _handle_staging_remove(
         """Make an HTTP DELETE request to the presigned_url"""
 
         self.volume_operation_type = DriverVolumeOperationType.DELETE
-        self.volume_path = presigned_url  # Using presigned URL as path since there's no local file
+        self.volume_path = (
+            presigned_url  # Using presigned URL as path since there's no local file
+        )
 
         r = requests.delete(url=presigned_url, headers=headers)
 
@@ -1010,7 +1031,8 @@ def get_async_execution_result(self):
             return self
         else:
             raise Error(
-                f"get_execution_result failed with Operation status {operation_state}", connection_uuid=self.connection.get_session_id_hex()
+                f"get_execution_result failed with Operation status {operation_state}",
+                connection_uuid=self.connection.get_session_id_hex(),
             )
 
     def executemany(self, operation, seq_of_parameters):
@@ -1160,7 +1182,10 @@ def fetchall(self) -> List[Row]:
         if self.active_result_set:
             return self.active_result_set.fetchall()
         else:
-            raise Error("There is no active result set", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "There is no active result set",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
     def fetchone(self) -> Optional[Row]:
         """
@@ -1174,7 +1199,10 @@ def fetchone(self) -> Optional[Row]:
         if self.active_result_set:
             return self.active_result_set.fetchone()
         else:
-            raise Error("There is no active result set", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "There is no active result set",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
     def fetchmany(self, size: int) -> List[Row]:
         """
@@ -1196,21 +1224,30 @@ def fetchmany(self, size: int) -> List[Row]:
         if self.active_result_set:
             return self.active_result_set.fetchmany(size)
         else:
-            raise Error("There is no active result set", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "There is no active result set",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
     def fetchall_arrow(self) -> "pyarrow.Table":
         self._check_not_closed()
         if self.active_result_set:
             return self.active_result_set.fetchall_arrow()
         else:
-            raise Error("There is no active result set", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "There is no active result set",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
     def fetchmany_arrow(self, size) -> "pyarrow.Table":
         self._check_not_closed()
         if self.active_result_set:
             return self.active_result_set.fetchmany_arrow(size)
         else:
-            raise Error("There is no active result set", connection_uuid=self.connection.get_session_id_hex())
+            raise Error(
+                "There is no active result set",
+                connection_uuid=self.connection.get_session_id_hex(),
+            )
 
     def cancel(self) -> None:
         """

src/databricks/sql/telemetry/models/event.py

@@ -97,7 +97,7 @@ class DriverSystemConfiguration:
     runtime_name: str
     runtime_version: str
     runtime_vendor: str
-    client_app_name: str
+    client_app_name: Optional[str]
     locale_name: str
     driver_name: str
     char_set_encoding: str

src/databricks/sql/telemetry/telemetry_client.py

@@ -47,7 +47,7 @@ def getDriverSystemConfiguration(cls) -> DriverSystemConfiguration:
                 os_name=platform.system(),
                 os_version=platform.release(),
                 os_arch=platform.machine(),
-                client_app_name="unknown",  # TODO: Add client app name
+                client_app_name=None,  # TODO: Add client app name
                 locale_name=locale.getlocale()[0] or locale.getdefaultlocale()[0],
                 char_set_encoding=sys.getdefaultencoding(),
             )