feat: colored logging formatter

Dockerfile

@@ -5,7 +5,7 @@
 ARG PYTHON_VERSION=3.11
 FROM python:${PYTHON_VERSION}-slim-bookworm AS builder
 
-ARG POETRY_VERSION=1.6
+ARG POETRY_VERSION=1.7
 ARG INSTALL_EXTRAS
 
 ENV POETRY_NO_INTERACTION=1 \
@@ -19,6 +19,7 @@ ENV POETRY_NO_INTERACTION=1 \
 RUN apt-get update \
  && apt-get install --no-install-recommends -y build-essential \
  && apt-get clean && rm -rf /var/lib/apt/lists/* \
+ && pip install --upgrade pip \
  && pip install --no-cache-dir poetry==$POETRY_VERSION
 
 WORKDIR /app

log.conf

@@ -23,5 +23,6 @@ level=DEBUG
 formatter=consoleFormatter
 
 [formatter_consoleFormatter]
-format=%(asctime)s - %(thread)d - %(levelname)s - %(message)s
-class=logging.Formatter
+format=%(asctime)s | %(thread)d | %(levelname)s | %(message)s
+datefmt=%Y-%m-%d %H:%M:%S
+class=numalogic.udfs.ColoredFormatter

numalogic/udfs/__init__.py

@@ -14,6 +14,31 @@
 from numalogic.udfs.trainer import TrainerUDF, PromTrainerUDF, DruidTrainerUDF
 
 
+class ColoredFormatter(logging.Formatter):
+ """Custom formatter for colored console logs."""
+
+ WHITE = "\x1b[1m"
+ BLUE = "\x1b[1;36m"
+ YELLOW = "\x1b[33;20m"
+ RED = "\x1b[31;20m"
+ BOLD_RED = "\x1b[31;1m"
+ RESET = "\x1b[0m"
+
+ def __init__(self, *args, **kwargs):
+ super().__init__(*args, **kwargs)
+ self._formats = {
+ logging.DEBUG: f"{self.BLUE}{self._fmt}{self.RESET}",
+ logging.INFO: f"{self.WHITE}{self._fmt}{self.RESET}",
+ logging.WARNING: f"{self.YELLOW}{self._fmt}{self.RESET}",
+ logging.ERROR: f"{self.RED}{self._fmt}{self.RESET}",
+ logging.CRITICAL: f"{self.BOLD_RED}{self._fmt}{self.RESET}",
+ }
+
+ def format(self, record):
+ self._style._fmt = self._formats.get(record.levelno)
+ return super().format(record)
+
+
 def set_logger() -> None:
  """Sets the logger for the UDFs."""
  logconf.fileConfig(