Merge pull request #5 from meet86/v1.1.0

V1.1.0

Author: ivaltryek

README.md

@@ -1,13 +1,14 @@
 # Datadog Custom Logger
-[![PyPI version](https://badge.fury.io/py/datadog-custom-logger.svg)](https://badge.fury.io/py/datadog-custom-logger)  [![Open in VS Code](https://open.vscode.dev/badges/open-in-vscode.svg)](https://open.vscode.dev/meet86/datadog-custom-logger)
+[![PyPI](https://img.shields.io/pypi/v/datadog-custom-logger)](https://pypi.org/project/datadog-custom-logger/)
+  [![Open in VS Code](https://open.vscode.dev/badges/open-in-vscode.svg)](https://open.vscode.dev/meet86/datadog-custom-logger)
 
 ## Usage
 
 ### Steps
 - Install pip package
 
   ```shell
-  pip install datadog-custom-logger==0.1.0
+  pip install datadog-custom-logger
   ```
 - Import package
 

datadog_custom_logger/handler.py

@@ -1,5 +1,6 @@
 import os
 import logging
+import datetime
 from dateutil.parser import parse as dateutil_parser
 
 # Datadog python client dependencies.
@@ -18,10 +19,11 @@
 
 class DatadogCustomLogHandler(logging.Handler):
 
-    def __init__(self, tags=None, ** kwargs):
+    def __init__(self, tags=None, service="default", ** kwargs):
         super(DatadogCustomLogHandler, self).__init__(**kwargs)
 
         self.tags = tags
+        self.service = service
 
     def emit(self, record):
         text = self.format(record)
@@ -39,13 +41,16 @@ def emit(self, record):
 
         # configuration.api_key['apiKeyAuth'] = os.environ['DD_CLIENT_API_KEY']
 
+        # Get UTC time and format it into default Datadog Python log pipeline standards.
+        time = datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S,%f')[:-3]
+        
         with ApiClient(configuration) as api_client:
             api_instance = logs_api.LogsApi(api_client)
             body = HTTPLog([
                 HTTPLogItem(
                     ddsource="python",
-                    message=f"{record.levelname.upper()}:: {create_args['text']}",
-                    service="AZF",
+                    message=f"{time} {record.levelname.upper()} :: {create_args['text']}",
+                    service=self.service,
                     ddtags=self.tags +
                     f",level:{record.levelname}" if self.tags is not None else f"level:{record.levelname}"
                 )
@@ -54,4 +59,4 @@ def emit(self, record):
             api_response = api_instance.submit_log(body)
         except ApiException as e:
             logging.info(
-                "Exception when calling LogsApi->submit_log: %s\n" % e)
+                "Exception when calling LogsApi->submit_log: %s\n" % e)

setup.py

@@ -12,7 +12,7 @@
 
 setuptools.setup(
     name="datadog-custom-logger",
-    version="1.0.0",
+    version="1.1.0",
     author="Meet Vasani",
     author_email="meet.vasani86@gmail.com",
     description="Logger Tool for Datadog logs monitoring",