Datadog TCP logging

1 files changed, 12 insertions, 2 deletions

diff --git a/pysite/__init__.py b/pysite/__init__.py
index f807eaf4..cc80b075 100644
--- a/pysite/__init__.py
+++ b/pysite/__init__.py
@@ -5,22 +5,32 @@ from logging import StreamHandler
 from logging.handlers import SysLogHandler
 import sys
 
-from pysite.constants import PAPERTRAIL_ADDRESS, PAPERTRAIL_PORT
+from logmatic import JsonFormatter
+
+from pysite.constants import DATADOG_ADDRESS, DATADOG_PORT, PAPERTRAIL_ADDRESS, PAPERTRAIL_PORT
+from pysite.logs import NonPicklingSocketHandler
 
 # region Logging
 # Get the log level from environment
+
 log_level = os.environ.get("LOG_LEVEL", "info").upper()
 
 if hasattr(logging, log_level):
     log_level = getattr(logging, log_level)
 else:
-    raise RuntimeError("LOG_LEVEL environment variable has an invalid value.")
+    raise RuntimeError(f"LOG_LEVEL environment variable has invalid value: {log_level}")
 
 logging_handlers = []
 
 if PAPERTRAIL_ADDRESS:
     logging_handlers.append(SysLogHandler(address=(PAPERTRAIL_ADDRESS, PAPERTRAIL_PORT)))
 
+if DATADOG_ADDRESS:
+    datadog_handler = NonPicklingSocketHandler(host=DATADOG_ADDRESS, port=DATADOG_PORT)
+    datadog_handler.formatter = JsonFormatter(datefmt="%b %d %H:%M:%S")
+
+    logging_handlers.append(datadog_handler)
+
 logging_handlers.append(StreamHandler(stream=sys.stderr))
 
 logging.basicConfig(