formatting

Signed-off-by: Sai Shree Pradhan <[email protected]>

Author: saishreeeee

src/databricks/sql/telemetry/DriverConnectionParameters.py

@@ -16,13 +16,13 @@ class DriverConnectionParameters:
     auth_scope: str
     discovery_url: str
     allowed_volume_ingestion_paths: str
-    enable_complex_datatype_support: bool
     azure_tenant_id: str
     socket_timeout: int
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
+
 # Part of TelemetryEvent
 # DriverConnectionParameters connectionParams = new DriverConnectionParameters(
 #     httpPath = " /sql/1.0/endpoints/1234567890abcdef",
@@ -36,7 +36,6 @@ def to_json(self):
 #     authScope = "sql",
 #     discoveryUrl = "https://example-url",
 #     allowedVolumeIngestionPaths = "[]",
-#     enableComplexDatatypeSupport = true,
 #     azureTenantId = "1234567890abcdef",
 #     socketTimeout = 10000
-# )     
+# )

src/databricks/sql/telemetry/DriverErrorInfo.py

@@ -10,7 +10,8 @@ class DriverErrorInfo:
     def to_json(self):
         return json.dumps(asdict(self))
 
-# Required for ErrorLogs  
+
+# Required for ErrorLogs
 # DriverErrorInfo errorInfo = new DriverErrorInfo(
 #     errorName="CONNECTION_ERROR",
 #     stackTrace="Connection failure while using the Databricks SQL Python connector. Failed to connect to server: https://my-workspace.cloud.databricks.com\n" +
@@ -19,4 +20,4 @@ def to_json(self):
 #               "at databricks.sql.thrift_backend.ThriftBackend.attempt_request(ThriftBackend.py:366)\n" +
 #               "at databricks.sql.thrift_backend.ThriftBackend.open_session(ThriftBackend.py:575)\n" +
 #               "at databricks.sql.client.Connection.__init__(client.py:69)\n" +
-#               "at databricks.sql.client.connect(connection.py:123)")
+#               "at databricks.sql.client.connect(connection.py:123)")

src/databricks/sql/telemetry/DriverSystemConfiguration.py

@@ -19,18 +19,19 @@ class DriverSystemConfiguration:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
+
 # Part of TelemetryEvent
 # DriverSystemConfiguration systemConfig = new DriverSystemConfiguration(
-#     driver_version = "2.9.3",  
-#     os_name = "Darwin",        
-#     os_version = "24.4.0",     
-#     os_arch = "arm64",         
-#     runtime_name = "CPython",   
-#     runtime_version = "3.13.3", 
+#     driver_version = "2.9.3",
+#     os_name = "Darwin",
+#     os_version = "24.4.0",
+#     os_arch = "arm64",
+#     runtime_name = "CPython",
+#     runtime_version = "3.13.3",
 #     runtime_vendor = "cpython",
-#     client_app_name = "databricks-sql-python", 
-#     locale_name = "en_US", 
-#     driver_name = "databricks-sql-python",     
-#     char_set_encoding = "UTF-8" 
-# )
+#     client_app_name = "databricks-sql-python",
+#     locale_name = "en_US",
+#     driver_name = "databricks-sql-python",
+#     char_set_encoding = "UTF-8"
+# )

src/databricks/sql/telemetry/DriverVolumeOperation.py

@@ -4,14 +4,16 @@
     DriverVolumeOperationType,
 )
 
+
 @dataclass
 class DriverVolumeOperation:
     volume_operation_type: DriverVolumeOperationType
     volume_path: str
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
+
 # Part of TelemetryEvent
 # DriverVolumeOperation volumeOperation = new DriverVolumeOperation(
 #     volumeOperationType = "LIST",

src/databricks/sql/telemetry/FrontendLogContext.py

@@ -9,7 +9,8 @@ class FrontendLogContext:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
+
 # used in TelemetryFrontendLog
 # FrontendLogContext frontendLogContext = new FrontendLogContext(
 #     clientContext = new TelemetryClientContext(

src/databricks/sql/telemetry/HostDetails.py

@@ -10,6 +10,7 @@ class HostDetails:
     def to_json(self):
         return json.dumps(asdict(self))
 
+
 # Part of DriverConnectionParameters
 # HostDetails hostDetails = new HostDetails(
 #     hostUrl = "https://my-workspace.cloud.databricks.com",

src/databricks/sql/telemetry/SqlExecutionEvent.py

@@ -14,10 +14,11 @@ class SqlExecutionEvent:
     def to_json(self):
         return json.dumps(asdict(self))
 
+
 # Part of TelemetryEvent
 # SqlExecutionEvent sqlExecutionEvent = new SqlExecutionEvent(
 #     statementType = "QUERY",
 #     isCompressed = true,
 #     executionResult = "INLINE_ARROW",
 #     retryCount = 0
-# ) 
+# )

src/databricks/sql/telemetry/TelemetryClientContext.py

@@ -9,9 +9,10 @@ class TelemetryClientContext:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
+
 # used in FrontendLogContext
 # TelemetryClientContext clientContext = new TelemetryClientContext(
 #     timestampMillis = 1716489600000,
 #     userAgent = "databricks-sql-python-test"
-# )
+# )