sfa/methods/ListResources.py

   1 import zlib
   2
   3 from sfa.util.xrn import urn_to_hrn
   4 from sfa.util.method import Method
   5 from sfa.util.sfatablesRuntime import run_sfatables
   6 from sfa.util.faults import SfaInvalidArgument
   7 from sfa.trust.credential import Credential
   8
   9 from sfa.storage.parameter import Parameter, Mixed
  10
  11 class ListResources(Method):
  12     """
  13     Returns information about available resources or resources allocated to this slice
  14     @param credential list
  15     @param options dictionary
  16     @return string
  17     """
  18     interfaces = ['aggregate', 'slicemgr']
  19     accepts = [
  20         Mixed(Parameter(str, "Credential string"),
  21               Parameter(type([str]), "List of credentials")),
  22         Parameter(dict, "Options")
  23         ]
  24     returns = Parameter(str, "List of resources")
  25
  26     def call(self, creds, options):
  27         self.api.logger.info("interface: %s\tmethod-name: %s" % (self.api.interface, self.name))
  28
  29         # client must specify a version
  30         if not options.get('geni_rspec_version'):
  31             if options.get('rspec_version'):
  32                 options['geni_rspec_version'] = options['rspec_version']
  33             else:
  34                 raise SfaInvalidArgument('Must specify an rspec version option. geni_rspec_version cannot be null')
  35
  36         # get slice's hrn from options
  37         xrn = options.get('geni_slice_urn', '')
  38         (hrn, _) = urn_to_hrn(xrn)
  39
  40         # Find the valid credentials
  41         valid_creds = self.api.auth.checkCredentials(creds, 'listnodes', hrn)
  42
  43         # get hrn of the original caller
  44         origin_hrn = options.get('origin_hrn', None)
  45         if not origin_hrn:
  46             origin_hrn = Credential(string=valid_creds[0]).get_gid_caller().get_hrn()
  47         rspec = self.api.manager.ListResources(self.api, creds, options)
  48
  49         # filter rspec through sfatables
  50         if self.api.interface in ['aggregate']:
  51             chain_name = 'OUTGOING'
  52         elif self.api.interface in ['slicemgr']:
  53             chain_name = 'FORWARD-OUTGOING'
  54         self.api.logger.debug("ListResources: sfatables on chain %s"%chain_name)
  55         filtered_rspec = run_sfatables(chain_name, hrn, origin_hrn, rspec)
  56
  57         if options.has_key('geni_compressed') and options['geni_compressed'] == True:
  58             filtered_rspec = zlib.compress(filtered_rspec).encode('base64')
  59
  60         return filtered_rspec
  61
  62