logger.py

import logging
import sys

APP_LOGGER_NAME = "H-FISTA"


def setup_logger(logger_name=APP_LOGGER_NAME, is_debug=True, file_name=None):
    logger = logging.getLogger(logger_name)
    logger.setLevel(logging.DEBUG if is_debug else logging.INFO)

    formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")

    sh = logging.StreamHandler(sys.stdout)
    sh.setFormatter(formatter)
    logger.handlers.clear()
    logger.addHandler(sh)

    if file_name:
        fh = logging.FileHandler(file_name)
        fh.setFormatter(formatter)
        logger.addHandler(fh)

    return logger


def get_logger(module_name):
    return logging.getLogger(APP_LOGGER_NAME).getChild(module_name)


# Add a custom logger level, verbose debug
DEBUGV = 5

logging.addLevelName(DEBUGV, "DEBUGV")


def debugv(self, message, *args, **kws):
    if self.isEnabledFor(DEBUGV):
        # Yes, logger takes its '*args' as 'args'.
        self._log(DEBUGV, message, args, **kws)


logging.Logger.debugv = debugv