chore: template usage

cool_domain/logs.py

@@ -0,0 +1,84 @@
+import enum
+import logging
+import typing
+
+if typing.TYPE_CHECKING:
+    from .params import Parameters
+
+
+class TermColor(enum.StrEnum):
+    RESET = "0"
+    BOLD = "1"
+    FEINT = "2"
+    UNDERLINE = "4"
+
+    BLACK = "30"
+    RED = "31"
+    GREEN = "32"
+    YELLOW = "33"
+    BLUE = "34"
+    MAGENTA = "35"
+    CYAN = "36"
+    WHITE = "37"
+    GREY = "38"
+
+    def __str__(self) -> str:
+        return f"\033[{self.value}m"
+
+    @typing.override
+    def __add__(self, value: typing.Any, /) -> str:  # ty:ignore[invalid-method-override]
+        return str(self) + str(value)
+
+    def __radd__(self, value: typing.Any, /) -> str:
+        return str(value) + str(self)
+
+
+class ColoredLoggingFormatter(logging.Formatter):
+    pre_format = "%(asctime)s | "
+    level_format = "%(levelname)-8s"
+    post_format = " | [%(name)s] %(message)s"
+    trace_format = (
+        TermColor.FEINT
+        + TermColor.GREY
+        + " (%(filename)s:%(lineno)d)"
+        + TermColor.RESET
+    )
+
+    FORMAT_COLORS: typing.ClassVar[dict[int, TermColor | str]] = {
+        logging.DEBUG: TermColor.FEINT + TermColor.GREY,
+        logging.INFO: TermColor.GREEN,
+        logging.WARNING: TermColor.YELLOW,
+        logging.ERROR: TermColor.RED,
+        logging.CRITICAL: TermColor.RED,
+    }
+
+    @typing.override
+    def __init__(self, trace: bool) -> None:
+        self.trace: bool = trace
+        super().__init__()
+
+    @typing.override
+    def format(self, record: logging.LogRecord) -> str:
+        log_color: TermColor | str = self.FORMAT_COLORS.get(
+            record.levelno,
+            TermColor.MAGENTA,
+        )
+        formatter = logging.Formatter(
+            self.pre_format
+            + log_color
+            + TermColor.BOLD
+            + self.level_format
+            + TermColor.RESET
+            + self.post_format
+            + (self.trace_format if self.trace else ""),
+        )
+        return formatter.format(record)
+
+
+def setup_logs(params: Parameters) -> None:
+    stream_handler = logging.StreamHandler()
+    stream_handler.setFormatter(ColoredLoggingFormatter(trace=params.debug))
+    log_level = logging.INFO
+    if params.debug:
+        log_level = logging.DEBUG
+    logging.basicConfig(level=log_level, handlers=[stream_handler])