PLC/Methods/UpdateSite.py

   1 # $Id$
   2 from PLC.Faults import *
   3 from PLC.Method import Method
   4 from PLC.Parameter import Parameter, Mixed
   5 from PLC.Sites import Site, Sites
   6 from PLC.Auth import Auth
   7
   8 related_fields = Site.related_fields.keys()
   9 can_update = lambda (field, value): field in \
  10              ['name', 'abbreviated_name', 'login_base',
  11               'is_public', 'latitude', 'longitude', 'url',
  12               'max_slices', 'max_slivers', 'enabled', 'ext_consortium_id'] + \
  13               related_fields
  14
  15 class UpdateSite(Method):
  16     """
  17     Updates a site. Only the fields specified in update_fields are
  18     updated, all other fields are left untouched.
  19
  20     PIs can only update sites they are a member of. Only admins can
  21     update max_slices, max_slivers, and login_base.
  22
  23     Returns 1 if successful, faults otherwise.
  24     """
  25
  26     roles = ['admin', 'pi']
  27
  28     site_fields = dict(filter(can_update, Site.fields.items() + Site.related_fields.items()))
  29
  30     accepts = [
  31         Auth(),
  32         Mixed(Site.fields['site_id'],
  33               Site.fields['login_base']),
  34         site_fields
  35         ]
  36
  37     returns = Parameter(int, '1 if successful')
  38
  39     def call(self, auth, site_id_or_login_base, site_fields):
  40         site_fields = dict(filter(can_update, site_fields.items()))
  41
  42         # Get site information
  43         sites = Sites(self.api, [site_id_or_login_base])
  44         if not sites:
  45             raise PLCInvalidArgument, "No such site"
  46         site = sites[0]
  47
  48         if site['peer_id'] is not None:
  49             raise PLCInvalidArgument, "Not a local site"
  50
  51         # Authenticated function
  52         assert self.caller is not None
  53
  54         # If we are not an admin, make sure that the caller is a
  55         # member of the site.
  56         if 'admin' not in self.caller['roles']:
  57             if site['site_id'] not in self.caller['site_ids']:
  58                 raise PLCPermissionDenied, "Not allowed to modify specified site"
  59
  60             # Remove admin only fields
  61             for key in 'max_slices', 'max_slivers', 'login_base':
  62                 if key in site_fields:
  63                     del site_fields[key]
  64
  65         # Make requested associations
  66         for field in related_fields:
  67             if field in site_fields:
  68                 site.associate(auth, field, site_fields[field])
  69                 site_fields.pop(field)
  70
  71         site.update(site_fields)
  72         site.update_last_updated(False)
  73         site.sync()
  74
  75         # Logging variables
  76         self.event_objects = {'Site': [site['site_id']]}
  77         self.message = 'Site %d updated: %s' % \
  78                 (site['site_id'], ", ".join(site_fields.keys()))
  79
  80         return 1