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