Synnefo » snf-ganeti

class LUSetNodeParams(LogicalUnit):

  """Modifies the parameters of a node.

  """

  HPATH = "node-modify"

  HTYPE = constants.HTYPE_NODE

  _OP_REQP = ["node_name"]

  REQ_BGL = False

  def CheckArguments(self):

    node_name = self.cfg.ExpandNodeName(self.op.node_name)

    if node_name is None:

      raise errors.OpPrereqError("Invalid node name '%s'" % self.op.node_name)

    self.op.node_name = node_name

    if not hasattr(self.op, 'master_candidate'):

      raise errors.OpPrereqError("Please pass at least one modification")

    self.op.master_candidate = bool(self.op.master_candidate)

  def ExpandNames(self):

    self.needed_locks = {locking.LEVEL_NODE: self.op.node_name}

  def BuildHooksEnv(self):

    """Build hooks env.

    This runs on the master node.

    """

    env = {

      "OP_TARGET": self.op.node_name,

      "MASTER_CANDIDATE": str(self.op.master_candidate),

      }

    nl = [self.cfg.GetMasterNode(),

          self.op.node_name]

    return env, nl, nl

  def CheckPrereq(self):

    """Check prerequisites.

    This only checks the instance list against the existing names.

    """

    force = self.force = self.op.force

    return

  def Exec(self, feedback_fn):

    """Modifies a node.

    """

    node = self.cfg.GetNodeInfo(self.op.node_name)

    result = []

    if self.op.master_candidate is not None:

      node.master_candidate = self.op.master_candidate

      result.append(("master_candidate", str(self.op.master_candidate)))

    # this will trigger configuration file update, if needed

    self.cfg.Update(node)

    # this will trigger job queue propagation or cleanup

    self.context.ReaddNode(node)

    return result

    opcodes.OpSetNodeParams: cmdlib.LUSetNodeParams,

class OpSetNodeParams(OpCode):

  """Change the parameters of a node."""

  OP_ID = "OP_NODE_SET_PARAMS"

  OP_DSC_FIELD = "node_name"

  __slots__ = [

    "node_name",

    "force",

    "master_candidate",

    ]

def SetNodeParams(opts, args):

  """Modifies a node.

  @param opts: the command line options selected by the user

  @type args: list

  @param args: should contain only one element, the node name

  @rtype: int

  @return: the desired exit code

  """

  if opts.master_candidate is None:

    ToStderr("Please give at least one of the parameters.")

    return 1

  candidate = opts.master_candidate == 'yes'

  op = opcodes.OpSetNodeParams(node_name=args[0],

                              master_candidate=candidate,

                              force=opts.force)

  # even if here we process the result, we allow submit only

  result = SubmitOrSend(op, opts)

  if result:

    ToStdout("Modified node %s", args[0])

    for param, data in result:

      ToStdout(" - %-5s -> %s", param, data)

  return 0

  'modify': (SetNodeParams, ARGS_ONE,

             [DEBUG_OPT, FORCE_OPT,

              SUBMIT_OPT,

              make_option("-C", "--master-candidate", dest="master_candidate",

                          choices=('yes', 'no'), default=None,

                          help="Set the master_candidate flag on the node"),

              ],

             "<instance>", "Alters the parameters of an instance"),