Remove miliseconds from logs

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):
+    def __init__(self, fmt: str, *args: Any, **kwargs: Any):
-        super().__init__()
+        self._log_format = fmt
-        self._fmt = fmt
+        self._date_format = "%Y-%m-%d %H:%M:%S"
+
         self._color_mapping = {
-            logging.DEBUG: GREY + self._fmt + RESET,
+            logging.DEBUG: GREY + fmt + RESET,
-            logging.INFO: BLUE + self._fmt + RESET,
+            logging.INFO: BLUE + fmt + RESET,
-            logging.WARNING: YELLOW + self._fmt + RESET,
+            logging.WARNING: YELLOW + fmt + RESET,
-            logging.ERROR: RED + self._fmt + RESET,
+            logging.ERROR: RED + fmt + RESET,
-            logging.CRITICAL: BOLD_RED + self._fmt + RESET,
+            logging.CRITICAL: BOLD_RED + fmt + RESET,
         }
 
     def format(self, record: logging.LogRecord) -> str:
-        log_fmt = self._color_mapping.get(record.levelno)
+        log_format = self._color_mapping.get(record.levelno)
-        formatter = logging.Formatter(log_fmt)
+        formatter = logging.Formatter(log_format, self._date_format)
         return formatter.format(record)

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