Allow respecification of a node without requiring node_id
This commit is contained in:
parent
621fcbb9de
commit
e821770505
1 changed files with 16 additions and 19 deletions
|
@ -135,11 +135,21 @@ def _activate_virtualenv(path):
|
|||
execfile(activate_this, dict(__file__=activate_this))
|
||||
|
||||
|
||||
def _get_node(lb, node_id):
|
||||
"""Return a node with the given `node_id`"""
|
||||
for node in lb.nodes:
|
||||
if node.id == node_id:
|
||||
return node
|
||||
def _get_node(lb, node_id=None, address=None, port=None):
|
||||
"""Return a matching node"""
|
||||
searches = {
|
||||
'id': node_id,
|
||||
'address': address,
|
||||
'port': port
|
||||
}
|
||||
|
||||
for node in getattr(lb, 'nodes', []):
|
||||
try:
|
||||
if all(getattr(node, attr) == value
|
||||
for (attr, value) in searches.items() if value is not None):
|
||||
return node
|
||||
except AttributeError:
|
||||
continue
|
||||
return None
|
||||
|
||||
|
||||
|
@ -230,10 +240,7 @@ def main():
|
|||
except pyrax.exc.PyraxException, e:
|
||||
module.fail_json(msg='%s' % e.message)
|
||||
|
||||
if node_id:
|
||||
node = _get_node(lb, node_id)
|
||||
else:
|
||||
node = None
|
||||
node = _get_node(lb, node_id, address, port)
|
||||
|
||||
result = _node_to_dict(node)
|
||||
|
||||
|
@ -272,22 +279,12 @@ def main():
|
|||
except pyrax.exc.PyraxException, e:
|
||||
module.fail_json(msg='%s' % e.message)
|
||||
else: # Updating an existing node
|
||||
immutable = {
|
||||
'address': address,
|
||||
'port': port,
|
||||
}
|
||||
|
||||
mutable = {
|
||||
'condition': condition,
|
||||
'type': typ,
|
||||
'weight': weight,
|
||||
}
|
||||
|
||||
for name, value in immutable.items():
|
||||
if value:
|
||||
module.fail_json(
|
||||
msg='Attribute %s cannot be modified' % name)
|
||||
|
||||
for name, value in mutable.items():
|
||||
if value is None or value == getattr(node, name):
|
||||
mutable.pop(name)
|
||||
|
|
Loading…
Reference in a new issue