1 from PLC.Faults import *
2 from PLC.Method import Method
3 from PLC.Parameter import Parameter, Mixed
4 from PLC.NodeGroups import NodeGroup, NodeGroups
5 from PLC.Auth import Auth
7 related_fields = NodeGroup.related_fields.keys()
8 can_update = lambda (field, value): field in \
9 ['name', 'description'] + \
12 class UpdateNodeGroup(Method):
14 Updates a custom node group.
16 Returns 1 if successful, faults otherwise.
21 nodegroup_fields = dict(filter(can_update, NodeGroup.fields.items() + NodeGroup.related_fields.items()))
25 Mixed(NodeGroup.fields['nodegroup_id'],
26 NodeGroup.fields['name']),
30 returns = Parameter(int, '1 if successful')
32 def call(self, auth, nodegroup_id_or_name, nodegroup_fields):
33 nodegroup_fields = dict(filter(can_update, nodegroup_fields.items()))
35 # Get nodegroup information
36 nodegroups = NodeGroups(self.api, [nodegroup_id_or_name])
38 raise PLCInvalidArgument, "No such nodegroup"
39 nodegroup = nodegroups[0]
41 # Make requested associations
42 for field in related_fields:
43 if field in nodegroup_fields:
44 nodegroup.associate(auth, field, nodegroup_fields[field])
45 nodegroup_fields.pop(field)
47 nodegroup.update(nodegroup_fields)
51 self.event_objects = {'NodeGroup': [nodegroup['nodegroup_id']]}
52 self.message = 'Node group %d updated: %s' % \
53 (nodegroup['nodegroup_id'], ", ".join(nodegroup_fields.keys()))