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
8 from PLC.Namespace import hostname_to_hrn
9 from PLC.Peers import Peers
10 from PLC.Sites import Sites
11 from PLC.Nodes import Node, Nodes
12 from PLC.TagTypes import TagTypes
13 from PLC.NodeTags import NodeTags
14 from PLC.Methods.AddNodeTag import AddNodeTag
15 from PLC.Methods.UpdateNodeTag import UpdateNodeTag
17 admin_only = [ 'key', 'session', 'boot_nonce', 'site_id']
18 can_update = ['hostname', 'node_type', 'boot_state', 'model', 'version'] + admin_only
20 class UpdateNode(Method):
22 Updates a node. Only the fields specified in node_fields are
23 updated, all other fields are left untouched.
25 PIs and techs can update only the nodes at their sites. Only
26 admins can update the key, session, and boot_nonce fields.
28 Returns 1 if successful, faults otherwise.
31 roles = ['admin', 'pi', 'tech']
33 accepted_fields = Row.accepted_fields(can_update,Node.fields)
34 # xxx check the related_fields feature
35 accepted_fields.update(Node.related_fields)
36 accepted_fields.update(Node.tags)
40 Mixed(Node.fields['node_id'],
41 Node.fields['hostname']),
45 returns = Parameter(int, '1 if successful')
47 def call(self, auth, node_id_or_hostname, node_fields):
49 # split provided fields
50 [native,related,tags,rejected] = Row.split_fields(node_fields,[Node.fields,Node.related_fields,Node.tags])
53 native = Row.check_fields (native, self.accepted_fields)
55 raise PLCInvalidArgument, "Cannot update Node column(s) %r"%rejected
57 # Authenticated function
58 assert self.caller is not None
60 # Remove admin only fields
61 if 'admin' not in self.caller['roles']:
62 for key in admin_only:
63 if native.has_key(key):
66 # Get account information
67 nodes = Nodes(self.api, [node_id_or_hostname])
69 raise PLCInvalidArgument, "No such node %r"%node_id_or_hostname
72 if node['peer_id'] is not None:
73 raise PLCInvalidArgument, "Not a local node %r"%node_id_or_hostname
75 # If we are not an admin, make sure that the caller is a
76 # member of the site at which the node is located.
77 if 'admin' not in self.caller['roles']:
78 if node['site_id'] not in self.caller['site_ids']:
79 raise PLCPermissionDenied, "Not allowed to delete nodes from specified site"
81 # Make requested associations
82 for (k,v) in related.iteritems():
83 node.associate(auth, k,v)
86 node.update_last_updated(commit=False)
87 node.sync(commit=True)
89 # if hostname was modifed make sure to update the hrn
91 if 'hostname' in native:
92 # root authority should be PLC_HRN_ROOT for local
93 # objects or peer['hrn_root'] for peer objects
94 root_auth = self.api.config.PLC_HRN_ROOT
96 peers = Peers(self.api, node['peer_id'], ['hrn_root'])
98 root_auth = peers[0]['hrn_root']
100 # sub auth is the login base of this node's site
101 sites = Sites(self.api, node['site_id'], ['login_base'])
103 login_base = site['login_base']
104 tags['hrn'] = hostname_to_hrn(root_auth, login_base, node['hostname'])
106 for (tagname,value) in tags.iteritems():
107 # the tagtype instance is assumed to exist, just check that
108 if not TagTypes(self.api,{'tagname':tagname}):
109 raise PLCInvalidArgument,"No such TagType %s"%tagname
110 node_tags=NodeTags(self.api,{'tagname':tagname,'node_id':node['node_id']})
112 AddNodeTag(self.api).__call__(auth,node['node_id'],tagname,value)
114 UpdateNodeTag(self.api).__call__(auth,node_tags[0]['node_tag_id'],value)
117 self.event_objects = {'Node': [node['node_id']]}
118 if 'hostname' in node:
119 self.message = 'Node %s updated'%node['hostname']
121 self.message = 'Node %d updated'%node['node_id']
122 self.message += " [%s]." % (", ".join(node_fields.keys()),)
123 if 'boot_state' in node_fields.keys():
124 self.message += ' boot_state updated to %s' % node_fields['boot_state']