Source code for blockade.config

#!/usr/bin/env python
"""Module to load and process local configurations."""
import json
import os
import sys
from datetime import datetime
from blockade.common.utils import get_logger, process_whitelists

CONFIG_PATH = os.path.expanduser('~/.config/blockade')
CONFIG_FILE = os.path.join(CONFIG_PATH, 'api_config.json')
CONFIG_DEFAULTS = {'api_server': 'api.blockade.io', 'api_key': '',
                   'email': '', 'whitelist_date': ''}



class Config(object):

    """Manage configuration to ease library use."""

    def __init__(self, **kwargs):
        """Initialize the class."""
        self.logger = get_logger('blockade-cfg')
        self.config = CONFIG_DEFAULTS
        try:
            self.load_config(**kwargs)
        except ValueError as e:
            sys.stderr.write('Error: {}\n'.format(e.message))
            sys.exit(1)

    def write_config(self):
        """Write the configuration to a local file.

        :return: Boolean if successful
        """
        json.dump(
            self.config,
            open(CONFIG_FILE, 'w'),
            indent=4,
            separators=(',', ': ')
        )
        return True

    def load_config(self, **kwargs):
        """Load the configuration for the user or seed it with defaults.

        :return: Boolean if successful
        """
        virgin_config = False
        if not os.path.exists(CONFIG_PATH):
            virgin_config = True
            os.makedirs(CONFIG_PATH)
        if not os.path.exists(CONFIG_FILE):
            virgin_config = True
        if not virgin_config:
            self.config = json.load(open(CONFIG_FILE))
        else:
            self.logger.info('[!] Processing whitelists, this may take a few minutes...')
            process_whitelists()
        if kwargs:
            self.config.update(kwargs)
        if virgin_config or kwargs:
            self.write_config()
        if 'api_key' not in self.config:
            sys.stderr.write('configuration missing API key\n')
        if 'email' not in self.config:
            sys.stderr.write('configuration missing email\n')
        if not ('api_key' in self.config and 'email' in self.config):
            sys.stderr.write('Errors have been reported. Run blockade-cfg '
                             'to fix these warnings.\n')
        try:
            last_update = datetime.strptime(self.config['whitelist_date'],
                                            "%Y-%m-%d")
            current = datetime.now()
            delta = (current - last_update).days
            if delta > 14:
                self.logger.info('[!] Refreshing whitelists, this may take a few minutes...')
                process_whitelists()
                self.config['whitelist_date'] = datetime.now().strftime("%Y-%m-%d")
                self.write_config()
        except Exception as e:
            self.logger.error(str(e))
            self.logger.info('[!] Processing whitelists, this may take a few minutes...')
            process_whitelists()
            self.config['whitelist_date'] = datetime.now().strftime("%Y-%m-%d")
            self.write_config()

        return True

    @property
    def options(self):
        """Return configuration option data.

        :return: Dict of configuration keys
        """
        return self.config.keys()

    def get(self, item, default=None):
        """Get details from the configuration.

        :param str item: Key used for search
        :param default: Default value if search misses
        :return: Configuration value
        """
        return self.config.get(item, default)