PLC/Methods/GetSlices.py

   1 from PLC.Method import Method
   2 from PLC.Parameter import Parameter, Mixed
   3 from PLC.Filter import Filter
   4 from PLC.Auth import Auth
   5 from PLC.Persons import Person, Persons
   6 from PLC.Sites import Site, Sites
   7 from PLC.Slices import Slice, Slices
   8
   9 class GetSlices(Method):
  10     """
  11     Returns an array of structs containing details about slices. If
  12     slice_filter is specified and is an array of slice identifiers or
  13     slice names, or a struct of slice attributes, only slices matching
  14     the filter will be returned. If return_fields is specified, only the
  15     specified details will be returned.
  16
  17     Users may only query slices of which they are members. PIs may
  18     query any of the slices at their sites. Admins and nodes may query
  19     any slice. If a slice that cannot be queried is specified in
  20     slice_filter, details about that slice will not be returned.
  21     """
  22
  23     roles = ['admin', 'pi', 'user', 'node']
  24
  25     accepts = [
  26         Auth(),
  27         Mixed([Mixed(Slice.fields['slice_id'],
  28                      Slice.fields['name'])],
  29               Filter(Slice.fields)),
  30         Parameter([str], "List of fields to return", nullok = True)
  31         ]
  32
  33     returns = [Slice.fields]
  34
  35     def call(self, auth, slice_filter = None, return_fields = None):
  36         # If we are not admin, make sure to return only viewable
  37         # slices.
  38         if isinstance(self.caller, Person) and \
  39            'admin' not in self.caller['roles']:
  40             # Get slices that we are able to view
  41             valid_slice_ids = self.caller['slice_ids']
  42             if 'pi' in self.caller['roles'] and self.caller['site_ids']:
  43                 sites = Sites(self.api, self.caller['site_ids'])
  44                 for site in sites:
  45                     valid_slice_ids += site['slice_ids']
  46
  47             if not valid_slice_ids:
  48                 return []
  49
  50             if slice_filter is None:
  51                 slice_filter = valid_slice_ids
  52
  53         # Must query at least slice_id (see below)
  54         if return_fields is not None and 'slice_id' not in return_fields:
  55             return_fields.append('slice_id')
  56             added_fields = True
  57         else:
  58             added_fields = False
  59
  60         slices = Slices(self.api, slice_filter, return_fields)
  61
  62         # Filter out slices that are not viewable
  63         if isinstance(self.caller, Person) and \
  64            'admin' not in self.caller['roles']:
  65             slices = filter(lambda slice: slice['slice_id'] in valid_slice_ids, slices)
  66
  67         # Remove slice_id if not specified
  68         if added_fields:
  69             for slice in slices:
  70                 del slice['slice_id']
  71
  72         return slices