1 from PLC.Faults import *
2 from PLC.Method import Method
3 from PLC.Parameter import Parameter, Mixed
4 from PLC.Nodes import Node, Nodes
5 from PLC.NodeGroups import NodeGroup, NodeGroups
6 from PLC.Sites import Site, Sites
7 from PLC.Auth import PasswordAuth
9 can_update = lambda (field, value): field in \
10 ['boot_state', 'model', 'version']
12 class AddNode(Method):
14 Adds a new node. Any values specified in node_fields are used,
15 otherwise defaults are used.
17 PIs and techs may only add nodes to their own sites. Admins may
18 add nodes to any site.
20 Returns the new node_id (> 0) if successful, faults otherwise.
23 roles = ['admin', 'pi', 'tech']
25 update_fields = dict(filter(can_update, Node.fields.items()))
29 Mixed(Site.fields['site_id'],
30 Site.fields['login_base']),
31 Node.fields['hostname'],
35 returns = Parameter(int, 'New node_id (> 0) if successful')
37 def call(self, auth, site_id_or_login_base, hostname, node_fields = {}):
38 node_fields = dict(filter(can_update, node_fields.items()))
40 # Get site information
41 sites = Sites(self.api, [site_id_or_login_base])
43 raise PLCInvalidArgument, "No such site"
45 site = sites.values()[0]
47 # Authenticated function
48 assert self.caller is not None
50 # If we are not an admin, make sure that the caller is a
52 if 'admin' not in self.caller['roles']:
53 if site['site_id'] not in self.caller['site_ids']:
54 assert self.caller['person_id'] not in site['person_ids']
55 raise PLCPermissionDenied, "Not allowed to add nodes to specified site"
57 assert self.caller['person_id'] in site['person_ids']
59 node = Node(self.api, node_fields)
60 node['hostname'] = hostname
61 node['site_id'] = site['site_id']
64 return node['node_id']