1 from PLC.Faults import *
2 from PLC.Method import Method
3 from PLC.Parameter import Parameter, Mixed
4 from PLC.Sites import Site, Sites
5 from PLC.Auth import PasswordAuth
7 class AdmUpdateSite(Method):
9 Updates a site. Only the fields specified in update_fields are
10 updated, all other fields are left untouched.
12 To remove a value without setting a new one in its place (for
13 example, to remove an address from the node), specify -1 for int
14 and double fields and 'null' for string fields. hostname and
15 boot_state cannot be unset.
17 PIs can only update sites they are a member of. Only admins can
20 Returns 1 if successful, faults otherwise.
23 roles = ['admin', 'pi']
25 can_update = lambda (field, value): field in \
26 ['name', 'abbreviated_name',
27 'is_public', 'latitude', 'longitude', 'url',
29 update_fields = dict(filter(can_update, Site.fields.items()))
33 Mixed(Site.fields['site_id'],
34 Site.fields['login_base']),
38 returns = Parameter(int, '1 if successful')
40 def call(self, auth, site_id_or_login_base, update_fields):
41 # Check for invalid fields
42 if filter(lambda field: field not in self.update_fields, update_fields):
43 raise PLCInvalidArgument, "Invalid field specified"
45 # XML-RPC cannot marshal None, so we need special values to
47 for key, value in update_fields.iteritems():
48 if value == -1 or value == "null":
49 if key not in ['latitude', 'longitude', 'url',
50 'organization_id', 'ext_consortium_id']:
51 raise PLCInvalidArgument, "%s cannot be unset" % key
52 update_fields[key] = None
54 # Get site information
55 sites = Sites(self.api, [site_id_or_login_base])
57 raise PLCInvalidArgument, "No such site"
59 site = sites.values()[0]
61 # Authenticated function
62 assert self.caller is not None
64 # If we are not an admin, make sure that the caller is a
66 if 'admin' not in self.caller['roles']:
67 if site['site_id'] not in self.caller['site_ids']:
68 raise PLCPermissionDenied, "Not allowed to modify specified site"
70 if 'max_slices' in update_fields:
71 raise PLCInvalidArgument, "Only admins can update max_slices"
73 site.update(update_fields)