From 8b8cb0dfffe18b6da3380a8329486157eeeab0d8 Mon Sep 17 00:00:00 2001
From: Sean Story <sean.j.story@gmail.com>
Date: Mon, 15 Jul 2024 08:29:34 -0500
Subject: [PATCH] log in utc (#2695)

---
 connectors/logger.py | 64 +++++++++++++++++++++-----------------------
 tests/test_logger.py | 19 +++++++++++++
 2 files changed, 49 insertions(+), 34 deletions(-)

diff --git a/connectors/logger.py b/connectors/logger.py
index eaf265842..44df7f475 100644
--- a/connectors/logger.py
+++ b/connectors/logger.py
@@ -10,10 +10,12 @@
 import inspect
 import logging
 import time
-from functools import cached_property, wraps
+from datetime import datetime, timezone
+from functools import wraps
 from typing import AsyncGenerator
 
 import ecs_logging
+from dateutil.tz import tzlocal
 
 from connectors import __version__
 
@@ -27,46 +29,40 @@ class ColorFormatter(logging.Formatter):
     RED = "\x1b[31;20m"
     BOLD_RED = "\x1b[31;1m"
     RESET = "\x1b[0m"
+    COLORS = {
+        logging.DEBUG: GREY,
+        logging.INFO: GREEN,
+        logging.WARNING: YELLOW,
+        logging.ERROR: RED,
+        logging.CRITICAL: BOLD_RED,
+    }
 
     DATE_FMT = "%H:%M:%S"
 
     def __init__(self, prefix):
         self.custom_format = "[" + prefix + "][%(asctime)s][%(levelname)s] %(message)s"
-        super().__init__()
-
-    @cached_property
-    def debug_formatter(self):
-        return logging.Formatter(
-            fmt=self.GREY + self.custom_format + self.RESET, datefmt=self.DATE_FMT
-        )
-
-    @cached_property
-    def info_formatter(self):
-        return logging.Formatter(
-            fmt=self.GREEN + self.custom_format + self.RESET, datefmt=self.DATE_FMT
-        )
-
-    @cached_property
-    def warning_formatter(self):
-        return logging.Formatter(
-            fmt=self.YELLOW + self.custom_format + self.RESET, datefmt=self.DATE_FMT
-        )
-
-    @cached_property
-    def error_formatter(self):
-        return logging.Formatter(
-            fmt=self.RED + self.custom_format + self.RESET, datefmt=self.DATE_FMT
-        )
-
-    @cached_property
-    def critical_formatter(self):
-        return logging.Formatter(
-            fmt=self.BOLD_RED + self.custom_format + self.RESET, datefmt=self.DATE_FMT
-        )
+        super().__init__(datefmt=self.DATE_FMT)
+        self.local_tz = tzlocal()
+
+    def converter(self, timestamp):
+        dt = datetime.fromtimestamp(timestamp, self.local_tz)
+        return dt.astimezone(timezone.utc)
+
+    # override logging.Formatter to use an aware datetime object
+    def formatTime(self, record, datefmt=None):
+        dt = self.converter(record.created)
+        if datefmt:
+            s = dt.strftime(datefmt)
+        else:
+            try:
+                s = dt.isoformat(timespec="milliseconds")
+            except TypeError:
+                s = dt.isoformat()
+        return s
 
     def format(self, record):  # noqa: A003
-        formatter = getattr(self, f"{record.levelname.lower()}_formatter")
-        return formatter.format(record)
+        self._style._fmt = self.COLORS[record.levelno] + self.custom_format + self.RESET
+        return super().format(record)
 
 
 class DocumentLogger:
diff --git a/tests/test_logger.py b/tests/test_logger.py
index 904b60b92..a507cc621 100644
--- a/tests/test_logger.py
+++ b/tests/test_logger.py
@@ -10,6 +10,7 @@
 from contextlib import contextmanager
 
 import pytest
+from freezegun import freeze_time
 
 import connectors.logger
 from connectors.logger import ColorFormatter, logger, set_logger, tracer
@@ -174,6 +175,24 @@ def _w(msg):
         assert logs[0].startswith(color)
 
 
+# first param is UTC time, second param is offset we run with
+@freeze_time("2024-07-10 12:00:00", tz_offset=-7)
+def test_timestamp_is_utc():
+    with unset_logger():
+        logger = set_logger(logging.DEBUG, filebeat=False)
+        logs = []
+
+        def _w(msg):
+            logs.append(msg)
+
+        logger.handlers[0].stream.write = _w
+        logger.debug("test")
+        assert len(logs) == 1
+        assert (
+            "[12:00:00]" in logs[0]
+        )  # if the local time was respected, this would be 05:00:00
+
+
 def test_colored_logging_with_filebeat():
     with unset_logger():
         logger = set_logger(logging.DEBUG, filebeat=True)