Esempio n. 1
0
    def test_load__multi_reference(self):
        self.sample = """
        multi_ref: "9 + $(sample2:opt) + $(sample3:opt) + $(auto:home) + 12"
        """
        self.sample2 = """opt: 10"""
        self.sample3 = """opt: 11"""

        self._write_samples()

        processed = self.loader.load("sample")

        self.assertTrue(isinstance(processed, utils.OrderedDict))
        self.assertEqual(len(processed), 1)
        self.assertEqual(processed["multi_ref"], "9 + 10 + 11 + " + shell.gethomedir() + " + 12")
Esempio n. 2
0
    def test_load__auto_reference(self):
        self.sample = """
        ip: "$(auto:ip)"
        host: "$(auto:hostname)"
        home: "$(auto:home)"
        """
        self._write_samples()

        processed = self.loader.load('sample')

        self.assertTrue(isinstance(processed, utils.OrderedDict))
        self.assertEqual(len(processed), 3)
        self.assertEqual(processed['ip'], utils.get_host_ip())
        self.assertEqual(processed['host'], shell.hostname())
        self.assertEqual(processed['home'], shell.gethomedir())
Esempio n. 3
0
    def test_load__multi_reference(self):
        self.sample = """
        multi_ref: "9 + $(sample2:opt) + $(sample3:opt) + $(auto:home) + 12"
        """
        self.sample2 = """opt: 10"""
        self.sample3 = """opt: 11"""

        self._write_samples()

        processed = self.loader.load('sample')

        self.assertTrue(isinstance(processed, utils.OrderedDict))
        self.assertEqual(len(processed), 1)
        self.assertEqual(processed['multi_ref'],
                         "9 + 10 + 11 + " + shell.gethomedir() + " + 12")
Esempio n. 4
0
    def test_load__auto_reference(self):
        self.sample = """
        ip: "$(auto:ip)"
        host: "$(auto:hostname)"
        home: "$(auto:home)"
        """
        self._write_samples()

        processed = self.loader.load('sample')

        self.assertTrue(isinstance(processed, utils.OrderedDict))
        self.assertEqual(len(processed), 3)
        self.assertEqual(processed['ip'], utils.get_host_ip())
        self.assertEqual(processed['host'], shell.hostname())
        self.assertEqual(processed['home'], shell.gethomedir())
Esempio n. 5
0
def run(args):
    """
    Starts the execution after args have been parsed and logging has been setup.

    Arguments: N/A
    Returns: True for success to run, False for failure to start
    """
    LOG.debug("CLI arguments are:")
    utils.log_object(args, logger=LOG, level=logging.DEBUG, item_max_len=128)

    # Keep the old args around so we have the full set to write out
    saved_args = dict(args)
    action = args.pop("action", '').strip().lower()
    if action not in actions.names():
        raise excp.OptionException("Invalid action name %r specified!" %
                                   (action))

    # Determine + setup the root directory...
    # If not provided attempt to locate it via the environment control files
    args_root_dir = args.pop("dir")
    root_dir = env.get_key('INSTALL_ROOT')
    if not root_dir:
        root_dir = args_root_dir
    if not root_dir:
        root_dir = sh.joinpths(sh.gethomedir(), 'openstack')
    root_dir = sh.abspth(root_dir)
    sh.mkdir(root_dir)

    persona_fn = args.pop('persona_fn')
    if not persona_fn:
        raise excp.OptionException("No persona file name specified!")
    if not sh.isfile(persona_fn):
        raise excp.OptionException("Invalid persona file %r specified!" %
                                   (persona_fn))

    # !!
    # Here on out we should be using the logger (and not print)!!
    # !!

    # Stash the dryrun value (if any)
    if 'dryrun' in args:
        env.set("ANVIL_DRYRUN", str(args['dryrun']))

    # Ensure the anvil etc dir is there if others are about to use it
    ensure_anvil_dir()

    # Load the distro
    dist = distro.load(settings.DISTRO_DIR)

    # Load + verify the person
    try:
        persona_obj = persona.load(persona_fn)
        persona_obj.verify(dist)
    except Exception as e:
        raise excp.OptionException("Error loading persona file: %s due to %s" %
                                   (persona_fn, e))

    # Get the object we will be running with...
    runner_cls = actions.class_for(action)
    runner = runner_cls(distro=dist,
                        root_dir=root_dir,
                        name=action,
                        cli_opts=args)

    (repeat_string, line_max_len) = utils.welcome()
    print(center_text("Action Runner", repeat_string, line_max_len))

    # Now that the settings are known to work, store them for next run
    store_current_settings(saved_args)

    LOG.info("Starting action %s on %s for distro: %s",
             colorizer.quote(action), colorizer.quote(utils.iso8601()),
             colorizer.quote(dist.name))
    LOG.info("Using persona: %s", colorizer.quote(persona_fn))
    LOG.info("In root directory: %s", colorizer.quote(root_dir))

    start_time = time.time()
    runner.run(persona_obj)
    end_time = time.time()

    pretty_time = utils.format_time(end_time - start_time)
    LOG.info("It took %s seconds or %s minutes to complete action %s.",
             colorizer.quote(pretty_time['seconds']),
             colorizer.quote(pretty_time['minutes']), colorizer.quote(action))
Esempio n. 6
0
def _get_default_dir():
    root_dir = env.get_key('INSTALL_ROOT')
    if root_dir:
        return root_dir
    return sh.joinpths(sh.gethomedir(), 'openstack')
Esempio n. 7
0
def run(args):
    """
    Starts the execution after args have been parsed and logging has been setup.
    """

    LOG.debug("CLI arguments are:")
    utils.log_object(args, logger=LOG, level=logging.DEBUG, item_max_len=128)

    # Keep the old args around so we have the full set to write out
    saved_args = dict(args)
    action = args.pop("action", '').strip().lower()
    try:
        runner_cls = actions.class_for(action)
    except Exception as ex:
        raise excp.OptionException(str(ex))

    if runner_cls.needs_sudo:
        ensure_perms()

    persona_fn = args.pop('persona_fn')
    if not persona_fn:
        raise excp.OptionException("No persona file name specified!")
    if not sh.isfile(persona_fn):
        raise excp.OptionException("Invalid persona file %r specified!" % (persona_fn))

    # Determine + setup the root directory...
    # If not provided attempt to locate it via the environment control files
    args_root_dir = args.pop("dir")
    root_dir = env.get_key('INSTALL_ROOT')
    if not root_dir:
        root_dir = args_root_dir
    if not root_dir:
        root_dir = sh.joinpths(sh.gethomedir(), 'openstack')
    root_dir = sh.abspth(root_dir)

    (repeat_string, line_max_len) = utils.welcome()
    print(center_text("Action Runner", repeat_string, line_max_len))

    # !!
    # Here on out we should be using the logger (and not print)!!
    # !!

    # Stash the dryrun value (if any)
    if 'dryrun' in args:
        sh.set_dry_run(args['dryrun'])

    # Ensure the anvil dirs are there if others are about to use it...
    ensure_anvil_dirs(root_dir)

    # Load the distro
    dist = distro.load(settings.DISTRO_DIR)

    # Load + verify the person
    try:
        persona_obj = persona.load(persona_fn)
        persona_obj.verify(dist)
    except Exception as e:
        raise excp.OptionException("Error loading persona file: %s due to %s" % (persona_fn, e))

    # Get the object we will be running with...
    runner = runner_cls(distro=dist,
                        root_dir=root_dir,
                        name=action,
                        cli_opts=args)

    # Now that the settings are known to work, store them for next run
    store_current_settings(saved_args)

    LOG.info("Starting action %s on %s for distro: %s",
             colorizer.quote(action), colorizer.quote(utils.iso8601()),
             colorizer.quote(dist.name))
    LOG.info("Using persona: %s", colorizer.quote(persona_fn))
    LOG.info("In root directory: %s", colorizer.quote(root_dir))

    start_time = time.time()
    runner.run(persona_obj)
    end_time = time.time()

    pretty_time = utils.format_time(end_time - start_time)
    LOG.info("It took %s seconds or %s minutes to complete action %s.",
             colorizer.quote(pretty_time['seconds']), colorizer.quote(pretty_time['minutes']), colorizer.quote(action))
Esempio n. 8
0
def _get_default_dir():
    root_dir = env.get_key('INSTALL_ROOT')
    if root_dir:
        return root_dir
    return sh.joinpths(sh.gethomedir(), 'openstack')
Esempio n. 9
0
def _get_default_dir():
    root_dir = env.get_key("INSTALL_ROOT")
    if root_dir:
        return root_dir
    return sh.joinpths(sh.gethomedir(), "openstack")