python-logging.py

import logging
import colorlog

# Init logger variable.
LOGGER = colorlog.getLogger()

def set_logger(verbose):
    """
        Setup logging module
        :param verbose: Set debug log level.
        :type verbose: int
    """
    # Enable verbose if true
    if verbose:
        LOGGER.setLevel(logging.DEBUG)
    else:
        LOGGER.setLevel(logging.INFO)

    # Set formatter
    date_formatter = "%Y-%m-%d %H:%M:%S"
    message_formatter = '%(log_color)s%(bold)s%(levelname)s%(reset)s: %(message)s'
    log_formatter = colorlog.ColoredFormatter(message_formatter, date_formatter)

    # Create Stream Handler
    stream_handler = colorlog.StreamHandler(sys.stdout)
    stream_handler.setFormatter(log_formatter)

    LOGGER.addHandler(stream_handler)