jaesivsm/MindYourNeighbors

import logging
from os import path
from configparser import ConfigParser
from logging.handlers import SysLogHandler

from mind_your_neighbors.const import KNOWN_MACHINES_SECTION

logger = logging.getLogger('MindYourNeighbors')


def set_logger(loglevel, logfile=None, write_to_term=False):
    # cleaning existing handlers
    for handler in logger.handlers:
        logger.removeHandler(handler)

    base_format = '%(levelname)s - %(message)s'
    if write_to_term:
        logger.addHandler(logging.StreamHandler())
    if logfile:
        logger.addHandler(logging.FileHandler(path.expanduser(logfile)))
        formatter = logging.Formatter('%(asctime)s ' + base_format)
    else:
        logger.addHandler(SysLogHandler(address='/dev/log'))
        formatter = logging.Formatter('%(name)s: ' + base_format)
    for handler in logger.handlers:
        handler.setFormatter(formatter)
        handler.setLevel(loglevel)
    logger.addHandler(handler)
    logger.setLevel(loglevel)
    return logger


def _get_default_config():
    return ConfigParser(defaults={
                'enabled': 'true',
                'loglevel': 'INFO',
                'error_on_stderr': 'true',
                'cache_file': '/run/shm/mind_your_neighbors.cache',
                'threshold': '3',
                'loop_time_sec': '120'})


def get_config(config=None, force_verbose=False, write_to_term=False):
    if not config:
        config = _get_default_config()

    config.read(['/etc/mind_your_neighbors.cfg',
                path.expanduser('~/.config/mind_your_neighbors.cfg')])
    if force_verbose:
        loglevel = logging.DEBUG
    else:
        loglevel = config.get(config.default_section, 'loglevel')
        loglevel = getattr(logging, loglevel.upper())

    logfile = config.get(config.default_section, 'logfile', fallback=None)

    set_logger(loglevel, logfile, write_to_term)
    return config


def get_known_machines(config):
    if KNOWN_MACHINES_SECTION not in config.sections():
        return {}
    return dict(config[KNOWN_MACHINES_SECTION])