Source code for toolbox.contrib.config.config

from toolbox.plugin import ToolboxPlugin
from toolbox.mixins import ConfigMixin
from toolbox.defaults import *


class ConfigPlugin(ConfigMixin, ToolboxPlugin):
    """
    The Config plugin manages the global configuration of the toolbox.
    It can be used from the commandline aswell as progamatically by loading it from the registry within an other plugin
    """
    name = 'config'
    description = 'Config API holds and controls the global config'

    def prepare_parser(self, parser):
        parser.add_argument('method', choices=['set', 'get'], help="method")
        parser.add_argument('attribute', type=str, help="config key")
        parser.add_argument('value', nargs='?')

    def execute(self, args):
        if args.method == 'set' and not args.value is None:
            self.set(args.attribute, args.value)
        elif args.method == 'get':
            print("Attribute {} is set to {}".format(args.attribute, self.get(
                args.attribute)))

    def set(self, key, value):
        self.get_config()[key] = value

    def get(self, key):
        return self.get_config()[key]

    def set_defaults(self):
        """
        Sets the default config variables, used when toolbox is used for the first time
        :return:
        """
        self.set('toolbox_dir', TOOLBOX_DIR)
        self.set('config_dir', CONF_DIR)
        self.set('local_plugin_dir', LOCAL_PLUGIN_DIR)
        self.set('toolbox_prefix', TOOLBOX_PREFIX)
        self.set('external_plugins', [])