X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Fmethods%2FListResources.py;h=2ae119a7c90fe510c4220de7cf4de4124d3d83de;hb=fd395e1944dcd49f10a4d5b27ce4983ad389fb96;hp=b017381222ff67a4d60b24361a77ce93ca1e8c25;hpb=952322d76247f8991f3c2688ed7e1f5a22ca4572;p=sfa.git diff --git a/sfa/methods/ListResources.py b/sfa/methods/ListResources.py index b0173812..2ae119a7 100644 --- a/sfa/methods/ListResources.py +++ b/sfa/methods/ListResources.py @@ -1,59 +1,64 @@ -from sfa.util.faults import * -from sfa.util.namespace import * +import zlib + +from sfa.util.xrn import urn_to_hrn from sfa.util.method import Method -from sfa.util.parameter import Parameter, Mixed +from sfa.util.sfatablesRuntime import run_sfatables +from sfa.util.sfalogging import logger + +from sfa.util.faults import SfaInvalidArgument from sfa.trust.credential import Credential -from sfatables.runtime import SFATablesRules -import sys -import zlib + +from sfa.storage.parameter import Parameter, Mixed + class ListResources(Method): """ - Returns information about available resources or resources allocated to this slice + Returns information about available resources @param credential list @param options dictionary @return string """ - interfaces = ['geni_am'] + interfaces = ['aggregate', 'slicemgr'] accepts = [ - Parameter(type([str]), "List of credentials"), + Mixed(Parameter(str, "Credential string"), + Parameter(type([str]), "List of credentials")), Parameter(dict, "Options") - ] + ] returns = Parameter(str, "List of resources") def call(self, creds, options): - self.api.logger.info("interface: %s\tmethod-name: %s" % (self.api.interface, self.name)) - + logger.info("interface: %s\tmethod-name: %s" % + (self.api.interface, self.name)) + + # client must specify a version + if not options.get('geni_rspec_version'): + if options.get('rspec_version'): + options['geni_rspec_version'] = options['rspec_version'] + else: + raise SfaInvalidArgument( + 'Must specify an rspec version option. geni_rspec_version cannot be null') + # Find the valid credentials - hrn = None - if options.has_key('geni_slice_urn'): - xrn = options['geni_slice_urn'] - hrn, _ = urn_to_hrn(xrn) - - ValidCreds = self.api.auth.checkCredentials(creds, '', hrn) - origin_hrn = Credential(string=ValidCreds[0]).get_gid_caller().get_hrn() - - - manager_base = 'sfa.managers' - - if self.api.interface in ['geni_am']: - mgr_type = self.api.config.SFA_GENI_AGGREGATE_TYPE - manager_module = manager_base + ".geni_am_%s" % mgr_type - manager = __import__(manager_module, fromlist=[manager_base]) - rspec = manager.ListResources(self.api, ValidCreds, options) - outgoing_rules = SFATablesRules('OUTGOING') - - - filtered_rspec = rspec - if outgoing_rules.sorted_rule_list: - context = {'sfa':{'user':{'hrn':origin_hrn}, 'slice':{'hrn':None}}} - outgoing_rules.set_context(context) - filtered_rspec = outgoing_rules.apply(rspec) - - if options.has_key('geni_compressed') and options['geni_compressed'] == True: - filtered_rspec = zlib.compress(rspec).encode('base64') - - - return filtered_rspec - - + valid_creds = self.api.auth.checkCredentialsSpeaksFor( + creds, 'listnodes', options=options) + + # get hrn of the original caller + origin_hrn = options.get('origin_hrn', None) + if not origin_hrn: + origin_hrn = Credential( + cred=valid_creds[0]).get_gid_caller().get_hrn() + rspec = self.api.manager.ListResources(self.api, creds, options) + + # filter rspec through sfatables + if self.api.interface in ['aggregate']: + chain_name = 'OUTGOING' + elif self.api.interface in ['slicemgr']: + chain_name = 'FORWARD-OUTGOING' + logger.debug( + "ListResources: sfatables on chain %s" % chain_name) + filtered_rspec = run_sfatables(chain_name, '', origin_hrn, rspec) + + if 'geni_compressed' in options and options['geni_compressed'] == True: + filtered_rspec = zlib.compress(filtered_rspec).encode('base64') + + return filtered_rspec