Synnefo » snf-ganeti » ganeti-local

from ganeti import mcpu

def RunLocalHooks(hook_opcode, hooks_path, env_builder_fn):

  """Decorator that runs hooks before and after the decorated function.

  @type hook_opcode: string

  @param hook_opcode: opcode of the hook

  @type hooks_path: string

  @param hooks_path: path of the hooks

  @type env_builder_fn: function

  @param env_builder_fn: function that returns a dictionary containing the

    environment variables for the hooks.

  @raise RPCFail: in case of pre-hook failure

  """

  def decorator(fn):

    def wrapper(*args, **kwargs):

      _, myself = ssconf.GetMasterAndMyself()

      nodes = ([myself], [myself])  # these hooks run locally

      cfg = _GetConfig()

      hr = HooksRunner()

      hm = mcpu.HooksMaster(hook_opcode, hooks_path, nodes, hr.RunLocalHooks,

                            None, env_builder_fn, logging.warning,

                            cfg.GetClusterName(), cfg.GetMasterNode())

      hm.RunPhase(constants.HOOKS_PHASE_PRE)

      result = fn(*args, **kwargs)

      hm.RunPhase(constants.HOOKS_PHASE_POST)

      return result

    return wrapper

  return decorator

  def RunLocalHooks(self, node_list, hpath, phase, env):

    """Check that the hooks will be run only locally and then run them.

    """

    assert len(node_list) == 1

    node = node_list[0]

    _, myself = ssconf.GetMasterAndMyself()

    assert node == myself

    results = self.RunHooks(hpath, phase, env)

    # Return values in the form expected by HooksMaster

    return {node: (None, False, results)}