Exemplo n.º 1
0
    def __init__(self, config=None, config_dir=None):
        """
        must either pass config as a dict or vaping.config.Config
        or config_dir as a path to where the config dir is located
        """
        if config:
            if isinstance(config, dict):
                self.config = vaping.Config(data=config)
            else:
                if not config.meta:
                    raise ValueError("config was not specified or empty")
                self.config = config
        elif config_dir:
            self.config = vaping.Config(read=config_dir)
        else:
            raise ValueError("config was not specified or empty")

        self.joins = []
        self._logger = None
        self.plugin_context = PluginContext(self.config)

        vcfg = self.config.get('vaping', None)
        if not vcfg:
            vcfg = dict()

        # get either home_dir from config, or use config_dir
        self.home_dir = vcfg.get('home_dir', None)
        if not self.home_dir:
            self.home_dir = self.config.meta['config_dir']

        if not os.path.exists(self.home_dir):
            raise ValueError("home directory '{}' does not exist".format(
                self.home_dir))

        if not os.access(self.home_dir, os.W_OK):
            raise ValueError("home directory '{}' is not writable".format(
                self.home_dir))

        # change to home for working dir
        os.chdir(self.home_dir)

        # instantiate all defined plugins
        # TODO remove and let them lazy init?
        plugins = self.config.get('plugins', None)
        if not plugins:
            raise ValueError('no plugins specified')

        plugin.instantiate(self.config['plugins'], self.plugin_context)

        # check that probes don't name clash with plugins
        for probe in self.config.get('probes', []):
            if plugin.exists(probe["name"]):
                raise ValueError(
                    "probes may not share names with plugins ({})".format(
                        probe["name"]))

        # TODO move to daemon
        pidname = vcfg.get('pidfile', 'vaping.pid')
        self.pidfile = pidfile.PidFile(pidname=pidname, piddir=self.home_dir)
Exemplo n.º 2
0
    def __init__(self, config=None, config_dir=None):
        if config:
            if not config.meta:
                raise ValueError(
                    "no config specified, please use specify a home directory")
            self.config = config
        elif config_dir:
            self.config = vaping.Config(read=config_dir)
        else:
            raise ValueError(
                "no config specified, please use specify a home directory")

        self.joins = []
        self._logger = None
        self.plugin_context = PluginContext(self.config)

        vcfg = self.config.get('vaping', {})

        # change to home for working dir
        self.home_dir = os.path.abspath(self.config.meta['config_dir'])
        os.chdir(self.home_dir)

        # instantiate all defined plugins
        # TODO remove and let them lazy init?
        plugins = self.config.get('plugins', None)
        if not plugins:
            raise ValueError('no plugins specified')

        plugin.instantiate(self.config['plugins'], self.plugin_context)

        # TODO move to daemon
        pidname = vcfg.get('pidfile', 'vaping.pid')
        self.pidfile = pidfile.PidFile(pidname=pidname, piddir=self.home_dir)
Exemplo n.º 3
0
def test_plugin_context():
    data = {"1": "two"}
    cfg = vaping.Config(data=data)
    ctx = vaping.daemon.PluginContext(cfg)
    assert data == ctx.config.data

    # test immutable
    ctx.config.data["1"] = "three"
    assert data != ctx.config.data
Exemplo n.º 4
0
def test_plugin_context():
    data = {'1': 'two'}
    cfg = vaping.Config(data=data)
    ctx = vaping.daemon.PluginContext(cfg)
    assert data == ctx.config.data

    # test immutable
    ctx.config.data['1'] = 'three'
    assert data != ctx.config.data
Exemplo n.º 5
0
def test_start_stop(this_dir):
    config_dir = os.path.join(this_dir, "data", "config", "fping")
    vaping.daemon.Vaping(config=vaping.Config(read=config_dir))
Exemplo n.º 6
0
def test_empty_config_object():
    with pytest.raises(ValueError) as excinfo:
        vaping.daemon.Vaping(config=vaping.Config())
    assert "config was not specified" in str(excinfo.value)
Exemplo n.º 7
0
    def __init__(self, config=None, config_dir=None):
        """
        must either pass config as a dict or vaping.config.Config
        or config_dir as a path to where the config dir is located
        """
        if config:
            if isinstance(config, dict):
                self.config = vaping.Config(data=config)
            else:
                if not config.meta:
                    raise ValueError("config was not specified or empty")
                self.config = config
        elif config_dir:
            self.config = vaping.Config(read=config_dir)
        else:
            raise ValueError("config was not specified or empty")

        self.joins = []
        self._logger = None

        # configure vaping logging
        if "logging" in self.config:
            logging.config.dictConfig(self.config.get("logging"))

        self.plugin_context = PluginContext(self.config)

        vcfg = self.config.get("vaping", None)
        if not vcfg:
            vcfg = dict()

        # get either home_dir from config, or use config_dir
        self.home_dir = vcfg.get("home_dir", None)

        if not self.home_dir:
            self.home_dir = self.config.meta["config_dir"]

        self.home_dir = os.path.abspath(self.home_dir)

        if not os.path.exists(self.home_dir):
            raise ValueError(
                f"home directory '{self.home_dir}' does not exist")

        if not os.access(self.home_dir, os.W_OK):
            raise ValueError(
                f"home directory '{self.home_dir}' is not writable")

        # change to home for working dir
        os.chdir(self.home_dir)

        # instantiate all defined plugins
        # TODO remove and let them lazy init?
        plugins = self.config.get("plugins", None)
        if not plugins:
            raise ValueError("no plugins specified")

        plugin.instantiate(self.config["plugins"], self.plugin_context)

        # check that probes don't name clash with plugins
        for probe in self.config.get("probes", []):
            if plugin.exists(probe["name"]):
                raise ValueError(
                    "probes may not share names with plugins ({})".format(
                        probe["name"]))

        self.pidname = vcfg.get("pidfile", "vaping.pid")