Added classes required for telemetry

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

Author: saishreeeee

src/databricks/sql/telemetry/DriverConnectionParameters.py

@@ -9,16 +9,33 @@
 @dataclass
 class DriverConnectionParameters:
     http_path: str
-    driver_mode: DatabricksClientType
-    host_details: HostDetails
+    mode: DatabricksClientType
+    host_info: HostDetails
     auth_mech: AuthMech
     auth_flow: AuthFlow
     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",
+#     driverMode = "THRIFT",
+#     hostDetails = new HostDetails(
+#         hostUrl = "https://my-workspace.cloud.databricks.com",
+#         port = 443
+#     ),
+#     authMech = "OAUTH",
+#     authFlow = "AZURE_MANAGED_IDENTITIES",
+#     authScope = "sql",
+#     discoveryUrl = "https://example-url",
+#     allowedVolumeIngestionPaths = "[]",
+#     azureTenantId = "1234567890abcdef",
+#     socketTimeout = 10000
+# )

src/databricks/sql/telemetry/DriverErrorInfo.py

@@ -9,3 +9,15 @@ class DriverErrorInfo:
 
     def to_json(self):
         return json.dumps(asdict(self))
+
+
+# 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" +
+#               "databricks.sql.exc.OperationalError: Connection refused: connect\n" +
+#               "at databricks.sql.thrift_backend.ThriftBackend.make_request(ThriftBackend.py:329)\n" +
+#               "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)")

src/databricks/sql/telemetry/DriverSystemConfiguration.py

@@ -1,8 +1,5 @@
 import json
 from dataclasses import dataclass, asdict
-import platform
-import sys
-import locale
 from databricks.sql import __version__
 
 
@@ -20,18 +17,21 @@ class DriverSystemConfiguration:
     driver_name: str
     char_set_encoding: str
 
-    def __init__(self):
-        self.driver_version = __version__
-        self.os_name = platform.system()
-        self.os_version = platform.version()
-        self.os_arch = platform.machine()
-        self.runtime_name = platform.python_implementation()
-        self.runtime_version = platform.python_version()
-        self.runtime_vendor = sys.implementation.name
-        self.client_app_name = "databricks-sql-python"
-        self.locale_name = locale.getdefaultlocale()[0]
-        self.driver_name = "databricks-sql-python"
-        self.char_set_encoding = "UTF-8"
-
     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",
+#     runtime_vendor = "cpython",
+#     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

@@ -12,3 +12,10 @@ class DriverVolumeOperation:
 
     def to_json(self):
         return json.dumps(asdict(self))
+
+
+# Part of TelemetryEvent
+# DriverVolumeOperation volumeOperation = new DriverVolumeOperation(
+#     volumeOperationType = "LIST",
+#     volumePath = "/path/to/volume"
+# )

src/databricks/sql/telemetry/FrontendLogContext.py

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

src/databricks/sql/telemetry/HostDetails.py

@@ -9,3 +9,10 @@ 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",
+#     port = 443
+# )

src/databricks/sql/telemetry/SqlExecutionEvent.py

@@ -13,3 +13,12 @@ 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,3 +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"
+# )

src/databricks/sql/telemetry/TelemetryEvent.py

@@ -19,7 +19,7 @@ class TelemetryEvent:
     vol_operation: DriverVolumeOperation
     sql_operation: SqlExecutionEvent
     error_info: DriverErrorInfo
-    latency: int
+    operation_latency_ms: int
 
     def to_json(self):
         return json.dumps(asdict(self))