OpServ-Monitoring/opserv-backend

import logging
import sys

from application_settings.logging_settings import LoggingSettings

DEFAULT_FILE_LOGLEVEL = logging.DEBUG


def setup_argparse_logger():
    # TODO Transform to  logging  package values
    console_log_enabled = LoggingSettings.get_setting(LoggingSettings.KEY_CONSOLE_LOG) is not None
    file_log_enabled = LoggingSettings.get_setting(LoggingSettings.KEY_FILE_LOG) is not None
    file_log_path = LoggingSettings.get_setting(LoggingSettings.KEY_FILE_LOG)
    console_loglevel = LoggingSettings.get_setting(LoggingSettings.KEY_CONSOLE_LOG)

    setup_logger(
        console_log_enabled,
        file_log_enabled,
        console_loglevel,
        file_log_path,
        True,
        True
    )


def setup_logger(log_to_console, log_to_file, console_loglevel, file_log_path, log_server, log_gathering):
    # SETUP LOGGER
    rootLogger = logging.getLogger()
    rootLogger.setLevel(logging.NOTSET)
    rootLogger.propagate = True

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

    # create file handler, set the formatting to it and add it as an handler
    if log_to_file:
        # TODO Use runtime arg - maybe as a param to prevent dependencies

        fh = logging.FileHandler(file_log_path)
        fh.setLevel(DEFAULT_FILE_LOGLEVEL)
        fh.setFormatter(formatter)
        rootLogger.addHandler(fh)

    # Create Console logging handler, if activated
    ch = logging.StreamHandler(sys.stdout)

    if log_to_console:
        ch.setLevel(console_loglevel)
    else:
        ch.setLevel(logging.WARNING)

    # create formatter and add it to the handlers
    ch.setFormatter(formatter)

    # Add the desired filters to the console handler
    if not log_gathering:
        ch.addFilter(Blacklist("opserv.gathering"))
    if not log_server:
        ch.addFilter(Blacklist("opserv.server"))
    # Add the handler to the logger
    rootLogger.addHandler(ch)


class Whitelist(logging.Filter):
    """ src: http://stackoverflow.com/questions/17275334/what-is-a-correct-way-to-filter-different-loggers-using-python-logging """

    def __init__(self, *whitelist):
        self.whitelist = [logging.Filter(name) for name in whitelist]

    def filter(self, record):
        return any(f.filter(record) for f in self.whitelist)


class Blacklist(Whitelist):
    """ src: http://stackoverflow.com/questions/17275334/what-is-a-correct-way-to-filter-different-loggers-using-python-logging """

    def filter(self, record):
        return not Whitelist.filter(self, record)