Source code for whisk.log

# -*- coding: utf-8 -*-

"""Module for setting up logging."""

import logging
import sys

LOG_LEVELS = {
    'DEBUG': logging.DEBUG,
    'INFO': logging.INFO,
    'WARNING': logging.WARNING,
    'ERROR': logging.ERROR,
    'CRITICAL': logging.CRITICAL,
}

LOG_FORMATS = {
    'DEBUG': u'%(levelname)s %(name)s: %(message)s',
    'INFO': u'%(levelname)s: %(message)s',
}


def configure_logger(stream_level='DEBUG', log_file=None):
    """Configure logging for whisk.
    Set up logging to stdout with given level. If ``log_file`` is given set
    up logging to file with DEBUG level.
    """
    # Set up 'whisk' logger
    logger = logging.getLogger('whisk')
    logger.setLevel(logging.DEBUG)

    # Remove all attached handlers, in case there was
    # a logger using the name 'whisk'
    del logger.handlers[:]

    # Create a file handler if a log file is provided
    if log_file is not None:
        debug_formatter = logging.Formatter(LOG_FORMATS['DEBUG'])
        file_handler = logging.FileHandler(log_file)
        file_handler.setLevel(LOG_LEVELS['DEBUG'])
        file_handler.setFormatter(debug_formatter)
        logger.addHandler(file_handler)

    # Get settings based on the given stream_level
    log_formatter = logging.Formatter(LOG_FORMATS[stream_level])
    log_level = LOG_LEVELS[stream_level]

    # Create a stream handler
    stream_handler = logging.StreamHandler(stream=sys.stdout)
    stream_handler.setLevel(log_level)
    # Doesn't call the line below so that when running via tty there's less info for an end-user to parse thru.
    # This is esc. applicable for create/setup.
    # stream_handler.setFormatter(log_formatter)
    logger.addHandler(stream_handler)

    return logger