From 393362782516664f1a205ad6cbf959b6a97ec125 Mon Sep 17 00:00:00 2001 From: Mohit Goyal Date: Mon, 20 Mar 2023 14:38:13 +0530 Subject: [PATCH] adds service name env var in py sdk --- pydoctordroid/lib.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pydoctordroid/lib.py b/pydoctordroid/lib.py index 2343fcf..b525b4c 100644 --- a/pydoctordroid/lib.py +++ b/pydoctordroid/lib.py @@ -7,6 +7,8 @@ _DRDROID_AUTH_TOKEN_ENV = 'DRDROID_AUTH_TOKEN' _DRDROID_HOSTNAME_ENV = 'DRDROID_HOSTNAME' +_DRDROID_SERVICE_NAME_ENV = 'DRDROID_SERVICE_NAME' +_DRDROID_SERVICE_NAME_KEY = '$drd_service_name' _DRDROID_DEBUG = 'DRDROID_DEBUG' logger = setup_logger(debug=os.environ.get(_DRDROID_DEBUG, False)) @@ -23,6 +25,7 @@ def __init__( self._setup = False self._token = os.environ.get(_DRDROID_AUTH_TOKEN_ENV, token) self._hostname = os.environ.get(_DRDROID_HOSTNAME_ENV, endpoint) + self._service_name = os.environ.get(_DRDROID_SERVICE_NAME_ENV, "") if logger: self._logger = logger elif os.environ.get(_DRDROID_DEBUG, debug): @@ -39,6 +42,8 @@ def publish(self, name: str, payload: dict = None, event_time: datetime = None): if not self._setup: return try: + if self._service_name is not None: + payload[_DRDROID_SERVICE_NAME_KEY] = self._service_name event = create_event(name, payload, event_time) self._event_exporter.export(event) except Exception as e: