From 51f6f7174be4884af72817f2af3c0264735afe33 Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Sat, 9 Jul 2022 23:08:43 +0200
Subject: [PATCH] Remove miliseconds from logs

---
 great_ai/utilities/logger/custom_formatter.py | 22 ++++++++++---------
 great_ai/utilities/logger/get_logger.py       |  5 ++---
 2 files changed, 14 insertions(+), 13 deletions(-)

diff --git a/great_ai/utilities/logger/custom_formatter.py b/great_ai/utilities/logger/custom_formatter.py
index 81cb958..6d3fcf1 100644
--- a/great_ai/utilities/logger/custom_formatter.py
+++ b/great_ai/utilities/logger/custom_formatter.py
@@ -1,21 +1,23 @@
 import logging
+from typing import Any
 
 from .colors import BLUE, BOLD_RED, GREY, RED, RESET, YELLOW
 
 
 class CustomFormatter(logging.Formatter):
-    def __init__(self, fmt: str):
-        super().__init__()
-        self._fmt = fmt
+    def __init__(self, fmt: str, *args: Any, **kwargs: Any):
+        self._log_format = fmt
+        self._date_format = "%Y-%m-%d %H:%M:%S"
+
         self._color_mapping = {
-            logging.DEBUG: GREY + self._fmt + RESET,
-            logging.INFO: BLUE + self._fmt + RESET,
-            logging.WARNING: YELLOW + self._fmt + RESET,
-            logging.ERROR: RED + self._fmt + RESET,
-            logging.CRITICAL: BOLD_RED + self._fmt + RESET,
+            logging.DEBUG: GREY + fmt + RESET,
+            logging.INFO: BLUE + fmt + RESET,
+            logging.WARNING: YELLOW + fmt + RESET,
+            logging.ERROR: RED + fmt + RESET,
+            logging.CRITICAL: BOLD_RED + fmt + RESET,
         }
 
     def format(self, record: logging.LogRecord) -> str:
-        log_fmt = self._color_mapping.get(record.levelno)
-        formatter = logging.Formatter(log_fmt)
+        log_format = self._color_mapping.get(record.levelno)
+        formatter = logging.Formatter(log_format, self._date_format)
         return formatter.format(record)
diff --git a/great_ai/utilities/logger/get_logger.py b/great_ai/utilities/logger/get_logger.py
index fdae0d6..3e6fb90 100644
--- a/great_ai/utilities/logger/get_logger.py
+++ b/great_ai/utilities/logger/get_logger.py
@@ -13,12 +13,11 @@ def get_logger(
         logger = logging.getLogger(name)
         logger.setLevel(level)
 
-        fmt = "%(asctime)s | %(levelname)8s | %(message)s"
-
+        log_format = "%(asctime)s | %(levelname)8s | %(message)s"
         stdout_handler = logging.StreamHandler()
         stdout_handler.setLevel(level)
         if not disable_colors:
-            stdout_handler.setFormatter(CustomFormatter(fmt))
+            stdout_handler.setFormatter(CustomFormatter(log_format))
 
         logger.addHandler(stdout_handler)
         loggers[name] = logger