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, NodeTag
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_auth = self.api.config.PLC_HRN_ROOT
93 # sub auth is the login base of this node's site
94 sites = Sites(self.api, node['site_id'], ['login_base'])
96 login_base = site['login_base']
97 tags['hrn'] = hostname_to_hrn(root_auth, login_base, node['hostname'])
99 for (tagname,value) in tags.iteritems():
100 # the tagtype instance is assumed to exist, just check that
101 tag_types = TagTypes(self.api,{'tagname':tagname})
103 raise PLCInvalidArgument,"No such TagType %s"%tagname
104 tag_type = tag_types[0]
105 node_tags=NodeTags(self.api,{'tagname':tagname,'node_id':node['node_id']})
107 node_tag = NodeTag(self.api)
108 node_tag['node_id'] = node['node_id']
109 node_tag['tag_type_id'] = tag_type['tag_type_id']
110 node_tag['tagname'] = tagname
111 node_tag['value'] = value
114 node_tag = node_tags[0]
115 node_tag['value'] = value
118 self.event_objects = {'Node': [node['node_id']]}
119 if 'hostname' in node:
120 self.message = 'Node %s updated'%node['hostname']
122 self.message = 'Node %d updated'%node['node_id']
123 self.message += " [%s]." % (", ".join(node_fields.keys()),)
124 if 'boot_state' in node_fields.keys():
125 self.message += ' boot_state updated to %s' % node_fields['boot_state']