Skip to content

Commit fa6274e

Browse files
committed
changed driver system config
Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>
1 parent 8a871e3 commit fa6274e

File tree

3 files changed

+16
-19
lines changed

3 files changed

+16
-19
lines changed

examples/persistent_oauth.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,4 +60,4 @@ def read(self, hostname: str) -> Optional[OAuthToken]:
6060
print(row)
6161
cursor.close()
6262

63-
connection.close()
63+
connection.close()

src/databricks/sql/client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -312,6 +312,7 @@ def read(self) -> Optional[OAuthToken]:
312312
discovery_url=TelemetryHelper.get_discovery_url(auth_provider),
313313
socket_timeout=kwargs.get("_socket_timeout", None),
314314
)
315+
315316
self.telemetry_client = telemetry_client_factory.initialize_telemetry_client(
316317
telemetry_enabled=self.telemetry_enabled,
317318
connection_uuid=self.get_session_id_hex(),

src/databricks/sql/telemetry/telemetry_client.py

Lines changed: 14 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -24,32 +24,28 @@
2424
import uuid
2525
import locale
2626
from abc import ABC, abstractmethod
27+
from databricks.sql import __version__
2728

2829

2930
class TelemetryHelper:
3031
"""Helper class for getting telemetry related information."""
3132

32-
_DRIVER_SYSTEM_CONFIGURATION = None
33+
_DRIVER_SYSTEM_CONFIGURATION = DriverSystemConfiguration(
34+
driver_name="Databricks SQL Python Connector",
35+
driver_version=__version__,
36+
runtime_name=f"Python {sys.version.split()[0]}",
37+
runtime_vendor=platform.python_implementation(),
38+
runtime_version=platform.python_version(),
39+
os_name=platform.system(),
40+
os_version=platform.release(),
41+
os_arch=platform.machine(),
42+
client_app_name=None, # TODO: Add client app name
43+
locale_name=locale.getlocale()[0] or locale.getdefaultlocale()[0],
44+
char_set_encoding=sys.getdefaultencoding(),
45+
)
3346

3447
@classmethod
3548
def getDriverSystemConfiguration(cls) -> DriverSystemConfiguration:
36-
"""Get the driver system configuration."""
37-
if cls._DRIVER_SYSTEM_CONFIGURATION is None:
38-
from databricks.sql import __version__
39-
40-
cls._DRIVER_SYSTEM_CONFIGURATION = DriverSystemConfiguration(
41-
driver_name="Databricks SQL Python Connector",
42-
driver_version=__version__,
43-
runtime_name=f"Python {sys.version.split()[0]}",
44-
runtime_vendor=platform.python_implementation(),
45-
runtime_version=platform.python_version(),
46-
os_name=platform.system(),
47-
os_version=platform.release(),
48-
os_arch=platform.machine(),
49-
client_app_name=None, # TODO: Add client app name
50-
locale_name=locale.getlocale()[0] or locale.getdefaultlocale()[0],
51-
char_set_encoding=sys.getdefaultencoding(),
52-
)
5349
return cls._DRIVER_SYSTEM_CONFIGURATION
5450

5551
@staticmethod

0 commit comments

Comments
 (0)