From f5fab862b2cb28dcc70a6ae4c906c6935ead216c Mon Sep 17 00:00:00 2001 From: Iustin Pop <iustin@google.com> Date: Fri, 5 Aug 2011 16:52:44 +0200 Subject: [PATCH] Further cleanup after multi-evacuate removal MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Commit f0edfcf6 removed the parsing of multi-evacuate result, but the code went from: if mode in (multi-evac, relocate): β¦ if mode == relocate: β¦ to: if mode == relocate: β¦ if mode == relocate β¦ This patch simply removes the nested if. Signed-off-by: Iustin Pop <iustin@google.com> Reviewed-by: Guido Trotter <ultrotter@google.com> --- lib/cmdlib.py | 23 ++++++++++------------- 1 file changed, 10 insertions(+), 13 deletions(-) diff --git a/lib/cmdlib.py b/lib/cmdlib.py index 70efe7721..277a2aa00 100644 --- a/lib/cmdlib.py +++ b/lib/cmdlib.py @@ -13179,26 +13179,23 @@ class IAllocator(object): errors.ECODE_INVAL) if self.mode == constants.IALLOCATOR_MODE_RELOC: + assert self.relocate_from is not None + assert self.required_nodes == 1 + node2group = dict((name, ndata["group"]) for (name, ndata) in self.in_data["nodes"].items()) fn = compat.partial(self._NodesToGroups, node2group, self.in_data["nodegroups"]) - if self.mode == constants.IALLOCATOR_MODE_RELOC: - assert self.relocate_from is not None - assert self.required_nodes == 1 - - request_groups = fn(self.relocate_from) - result_groups = fn(rdict["result"]) + request_groups = fn(self.relocate_from) + result_groups = fn(rdict["result"]) - if self.success and result_groups != request_groups: - raise errors.OpExecError("Groups of nodes returned by iallocator (%s)" - " differ from original groups (%s)" % - (utils.CommaJoin(result_groups), - utils.CommaJoin(request_groups))) - else: - raise errors.ProgrammerError("Unhandled mode '%s'" % self.mode) + if self.success and result_groups != request_groups: + raise errors.OpExecError("Groups of nodes returned by iallocator (%s)" + " differ from original groups (%s)" % + (utils.CommaJoin(result_groups), + utils.CommaJoin(request_groups))) elif self.mode == constants.IALLOCATOR_MODE_NODE_EVAC: assert self.evac_mode in constants.IALLOCATOR_NEVAC_MODES -- GitLab