logging

logging.py

import logging
from datetime import datetime


class CustomFormatter(logging.Formatter):
    """Logging Formatter to add colors and count warning / errors"""

    grey = "\x1b[38;21m"
    yellow = "\x1b[33;21m"
    light_blue = "\x1b[1;36m"
    red = "\x1b[31;21m"
    bold_red = "\x1b[31;1m"
    reset = "\x1b[0m"
    format_debug = "[%(asctime)s][%(name)s:%(lineno)d][%(levelname)s] %(message)s"

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

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


def create_default_logger(name, log_level) -> logging.Logger:
    if log_level == "DEBUG":
        logging_level = logging.DEBUG
    elif log_level == "INFO":
        logging_level = logging.INFO
    elif log_level == "WARNING":
        logging_level = logging.WARNING
    elif log_level == "ERROR":
        logging_level = logging.ERROR
    elif log_level == "CRITICAL":
        logging_level = logging.CRITICAL

    try:
        name = "/".join(name.split("/")[name.split("/").index("src") :])
    except:
        pass

    logger = logging.getLogger(name)
    logger.setLevel(logging_level)

    # create console handler with a higher log level
    ch = logging.StreamHandler()
    ch.setLevel(logging_level)

    ch.setFormatter(CustomFormatter())

    logger.addHandler(ch)

    return logger


class StopWatch:
    def __init__(self, return_time_delta=False):
        self.__start = datetime.now()
        self.return_time_delta = return_time_delta

    def beep(self):
        """
        Return the time delta from the last beep or start.
        Returns formatted seconds and miliseconds as string.
        """

        td = datetime.now() - self.__start
        self.__restart_time()
        if self.return_time_delta:
            return td
        return f"{int(td.seconds)}.{str(int(td.microseconds) // 1000).zfill(3)} seconds"

    def __restart_time(self):
        self.__start = datetime.now()