mia/__init__.py

import os, sys

sys.path.append(os.path.dirname(os.path.realpath(__file__)))
import logging
from dataclasses import dataclass

@dataclass
class Colors:
    grey: str = "\x1b[38;20m"
    blue: str = "\x1b[34;20m"
    bold_blue: str = "\x1b[34;1m"
    yellow: str = "\x1b[33;20m"
    red: str = "\x1b[31;20m"
    bold_red: str = "\x1b[31;1m"
    reset: str = "\x1b[0m"


class ColorFormatter(logging.Formatter):

    colors = Colors()
    format = "[%(asctime)s %(name)s %(levelname)s] %(message)s"
    datefmt="%Y-%m-%d %H:%M:%S"

    FORMATS = {
        logging.DEBUG: colors.grey + format + colors.reset,
        logging.INFO: colors.grey + format + colors.reset,
        logging.WARNING: colors.yellow + format + colors.reset,
        logging.ERROR: colors.red + format + colors.reset,
        logging.CRITICAL: colors.bold_red + format + colors.reset
    }

    def format(self, record):
        log_fmt = self.FORMATS.get(record.levelno)
        formatter = logging.Formatter(log_fmt, datefmt=self.datefmt)
        return formatter.format(record)

formatter = logging.Formatter(
    fmt="[%(asctime)s %(name)s %(levelname)s] %(message)s",
    datefmt="%Y-%m-%d %H:%M:%S",
)

handler = logging.StreamHandler()
handler.setFormatter(ColorFormatter())
handler.setLevel(logging.INFO)

logger = logging.getLogger("mia")
logger.setLevel(logging.INFO)
logger.addHandler(handler)
logger.propagate = False