from PLC.Method import Method
from PLC.Parameter import Parameter, Mixed
from PLC.Nodes import Node, Nodes
-from PLC.Auth import PasswordAuth
+from PLC.Auth import Auth
can_update = lambda (field, value): field in \
- ['hostname', 'boot_state', 'model', 'version']
+ ['hostname', 'boot_state', 'model', 'version',
+ 'key', 'session', 'boot_nonce']
class UpdateNode(Method):
"""
Updates a node. Only the fields specified in node_fields are
updated, all other fields are left untouched.
-
- To remove a value without setting a new one in its place (for
- example, to remove an address from the node), specify -1 for int
- and double fields and 'null' for string fields. hostname and
- boot_state cannot be unset.
- PIs and techs can only update the nodes at their sites.
+ PIs and techs can update only the nodes at their sites. Only
+ admins can update the key, session, and boot_nonce fields.
Returns 1 if successful, faults otherwise.
"""
roles = ['admin', 'pi', 'tech']
- update_fields = dict(filter(can_update, Node.fields.items()))
+ node_fields = dict(filter(can_update, Node.fields.items()))
accepts = [
- PasswordAuth(),
+ Auth(),
Mixed(Node.fields['node_id'],
Node.fields['hostname']),
- update_fields
+ node_fields
]
returns = Parameter(int, '1 if successful')
def call(self, auth, node_id_or_hostname, node_fields):
node_fields = dict(filter(can_update, node_fields.items()))
+ # Remove admin only fields
+ if 'admin' not in self.caller['roles']:
+ for key in 'key', 'session', 'boot_nonce':
+ del node_fields[key]
+
# Get account information
nodes = Nodes(self.api, [node_id_or_hostname])
if not nodes:
raise PLCInvalidArgument, "No such node"
- node = nodes.values()[0]
+ node = nodes[0]
# Authenticated function
assert self.caller is not None