PLC/Methods/GetSliceAttributes.py

   1 from PLC.Faults import *
   2 from PLC.Method import Method
   3 from PLC.Parameter import Parameter, Mixed
   4 from PLC.SliceAttributes import SliceAttribute, SliceAttributes
   5 from PLC.Sites import Site, Sites
   6 from PLC.Slices import Slice, Slices
   7 from PLC.Nodes import Node, Nodes
   8 from PLC.Auth import PasswordAuth
   9
  10 class GetSliceAttributes(Method):
  11     """
  12     Get an array of structs containing the values of slice and sliver
  13     attributes. An attribute is a sliver attribute if the node_id
  14     field is set. If slice_attribute_id_list is specified, only the
  15     specified attributes will be queried, if set.
  16
  17     Users may only query attributes of slices or slivers of which they
  18     are members. PIs may only query attributes of slices or slivers at
  19     their sites, or of which they are members. Admins may query
  20     attributes of any slice or sliver.
  21     """
  22
  23     roles = ['admin', 'pi', 'user']
  24
  25     accepts = [
  26         PasswordAuth(),
  27         [SliceAttribute.fields['slice_attribute_id']],
  28         ]
  29
  30     returns = [SliceAttribute.fields]
  31
  32     def call(self, auth, slice_attribute_ids = None):
  33         # If we are not admin, make sure to only return our own slice
  34         # and sliver attributes.
  35         if 'admin' not in self.caller['roles']:
  36             if not slice_attribute_ids:
  37                 slice_attribute_ids = []
  38                 slices = Slices(self.api, self.caller['slice_ids']).values()
  39                 if 'pi' in self.caller['roles']:
  40                     sites = Sites(self.api, self.caller['site_ids']).values()
  41                     slices += Slices(self.api, sites['slice_ids']).values()
  42                 for slice in slices:
  43                     slice_attribute_ids = set(slice_attribute_ids).union(slice['slice_attribute_ids'])
  44
  45         slice_attributes = SliceAttributes(self.api, slice_attribute_ids).values()
  46
  47         slice_attributes = [dict(slice_attribute) for slice_attribute in slice_attributes]
  48
  49         return slice_attributes