3 from PLC.Faults import *
4 from PLC.Method import Method
5 from PLC.Parameter import Parameter, Mixed
6 from PLC.Table import Row
7 from PLC.Auth import Auth
9 from PLC.Nodes import Node, Nodes
10 from PLC.TagTypes import TagTypes
11 from PLC.NodeTags import NodeTags
12 from PLC.Methods.AddNodeTag import AddNodeTag
13 from PLC.Methods.UpdateNodeTag import UpdateNodeTag
15 admin_only = [ 'key', 'session', 'boot_nonce', 'site_id']
16 can_update = ['hostname', 'node_type', 'boot_state', 'model', 'version'] + admin_only
18 class UpdateNode(Method):
20 Updates a node. Only the fields specified in node_fields are
21 updated, all other fields are left untouched.
23 PIs and techs can update only the nodes at their sites. Only
24 admins can update the key, session, and boot_nonce fields.
26 Returns 1 if successful, faults otherwise.
29 roles = ['admin', 'pi', 'tech']
31 accepted_fields = Row.accepted_fields(can_update,Node.fields)
32 # xxx check the related_fields feature
33 accepted_fields.update(Node.related_fields)
34 accepted_fields.update(Node.tags)
38 Mixed(Node.fields['node_id'],
39 Node.fields['hostname']),
43 returns = Parameter(int, '1 if successful')
45 def call(self, auth, node_id_or_hostname, node_fields):
47 # split provided fields
48 [native,related,tags,rejected] = Row.split_fields(node_fields,[Node.fields,Node.related_fields,Node.tags])
51 native = Row.check_fields (native, self.accepted_fields)
53 raise PLCInvalidArgument, "Cannot update Node column(s) %r"%rejected
55 # Authenticated function
56 assert self.caller is not None
58 # Remove admin only fields
59 if 'admin' not in self.caller['roles']:
60 for key in admin_only:
61 if native.has_key(key):
64 # Get account information
65 nodes = Nodes(self.api, [node_id_or_hostname])
67 raise PLCInvalidArgument, "No such node %r"%node_id_or_hostname
70 if node['peer_id'] is not None:
71 raise PLCInvalidArgument, "Not a local node %r"%node_id_or_hostname
73 # If we are not an admin, make sure that the caller is a
74 # member of the site at which the node is located.
75 if 'admin' not in self.caller['roles']:
76 if node['site_id'] not in self.caller['site_ids']:
77 raise PLCPermissionDenied, "Not allowed to delete nodes from specified site"
79 # Make requested associations
80 for (k,v) in related.iteritems():
81 node.associate(auth, k,v)
84 node.update_last_updated(commit=False)
85 node.sync(commit=True)
87 for (tagname,value) in tags.iteritems():
88 # the tagtype instance is assumed to exist, just check that
89 if not TagTypes(self.api,{'tagname':tagname}):
90 raise PLCInvalidArgument,"No such TagType %s"%tagname
91 node_tags=NodeTags(self.api,{'tagname':tagname,'node_id':node['node_id']})
93 AddNodeTag(self.api).__call__(auth,node['node_id'],tagname,value)
95 UpdateNodeTag(self.api).__call__(auth,node_tags[0]['node_tag_id'],value)
98 self.event_objects = {'Node': [node['node_id']]}
99 if 'hostname' in node:
100 self.message = 'Node %s updated'%node['hostname']
102 self.message = 'Node %d updated'%node['node_id']
103 self.message += " [%s]." % (", ".join(node_fields.keys()),)
104 if 'boot_state' in node_fields.keys():
105 self.message += ' boot_state updated to %s' % node_fields['boot_state']