Move files

great_ai/utilities/logger/__init__.py

@@ -0,0 +1,2 @@
+from .custom_formatter import CustomFormatter
+from .get_logger import get_logger

great_ai/utilities/logger/colors.py

@@ -0,0 +1,6 @@
+GREY = "\x1b[38;21m"
+BLUE = "\x1b[38;5;39m"
+YELLOW = "\x1b[38;5;226m"
+RED = "\x1b[38;5;196m"
+BOLD_RED = "\x1b[31;1m"
+RESET = "\x1b[0m"

great_ai/utilities/logger/custom_formatter.py

@@ -0,0 +1,21 @@
+import logging
+
+from .colors import BLUE, BOLD_RED, GREY, RED, RESET, YELLOW
+
+
+class CustomFormatter(logging.Formatter):
+    def __init__(self, fmt: str):
+        super().__init__()
+        self._fmt = fmt
+        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,
+        }
+
+    def format(self, record: logging.LogRecord) -> str:
+        log_fmt = self._color_mapping.get(record.levelno)
+        formatter = logging.Formatter(log_fmt)
+        return formatter.format(record)

great_ai/utilities/logger/get_logger.py

@@ -0,0 +1,26 @@
+import logging
+from typing import Dict
+
+from .custom_formatter import CustomFormatter
+
+loggers: Dict[str, logging.Logger] = {}
+
+
+def get_logger(
+    name: str, level: int = logging.INFO, disable_colors: bool = False
+) -> logging.Logger:
+    if name not in loggers:
+        logger = logging.getLogger(name)
+        logger.setLevel(level)
+
+        fmt = "%(asctime)s | %(levelname)8s | %(message)s"
+
+        stdout_handler = logging.StreamHandler()
+        stdout_handler.setLevel(level)
+        if not disable_colors:
+            stdout_handler.setFormatter(CustomFormatter(fmt))
+
+        logger.addHandler(stdout_handler)
+        loggers[name] = logger
+
+    return loggers[name]