Moving context extraction to the rspec manager
[sfa.git] / sfa / methods / get_resources.py
1 ### $Id$
2 ### $URL$
3
4 from sfa.util.faults import *
5 from sfa.util.method import Method
6 from sfa.util.parameter import Parameter, Mixed
7 from sfa.trust.auth import Auth
8 from sfa.util.config import Config
9 from sfa.plc.nodes import Nodes
10 # RSpecManager_pl is not used. This line is a check that ensures that everything is in place for the import to work.
11 import sfa.rspecs.aggregates.rspec_manager_pl
12 from sfa.trust.credential import Credential
13 from sfatables.runtime import SFATablesRules
14
15 class get_resources(Method):
16     """
17     Get an resource specification (rspec). The rspec may describe the resources
18     available at an authority or the resources being used by a slice.      
19
20     @param cred credential string specifying the rights of the caller
21     @param hrn human readable name of the slice we are interesed in or None 
22            for an authority.  
23     """
24
25     interfaces = ['aggregate', 'slicemgr']
26     
27     accepts = [
28         Parameter(str, "Credential string"),
29         Mixed(Parameter(str, "Human readable name (hrn)"),
30               Parameter(None, "hrn not specified"))
31         ]
32
33     returns = Parameter(str, "String representatin of an rspec")
34     
35     def call(self, cred, hrn=None, caller_cred=None):
36         sfa_aggregate_type = Config().get_aggregate_rspec_type()
37
38         self.api.auth.check(cred, 'listnodes')
39         if caller_cred==None:
40            caller_cred=cred
41
42         #log the call
43         self.api.logger.info("interface: %s\tcaller-hrn: %s\ttarget-hrn: %s\tmethod-name: %s"%(self.api.interface, Credential(string=caller_cred).get_gid_caller().get_hrn(), hrn, self.name))
44
45         # This code needs to be cleaned up so that 'pl' is treated as just another RSpec manager.
46         # The change ought to be straightforward as soon as we define PL's new RSpec.
47
48         rspec_manager = __import__("sfa.rspecs.aggregates.rspec_manager_"+sfa_aggregate_type, fromlist = ["sfa.rspecs.aggregates"])
49         if (sfa_aggregate_type == 'pl'):
50             nodes = Nodes(self.api, caller_cred=caller_cred)
51             if hrn:
52                 rspec = nodes.get_rspec(hrn)
53             else:
54                 nodes.refresh()
55                 rspec = nodes['rspec']
56         else:
57             rspec = rspec_manager.get_rspec(self.api, hrn)
58         
59         # Filter the outgoing rspec using sfatables
60         outgoing_rules = SFATablesRules('OUTGOING')
61         
62         request_context = rspec_manager.fetch_context(hrn, Credential(string=caller_cred).get_gid_caller().get_hrn(),outgoing_rules.contexts)
63         outgoing_rules.set_context(request_context) 
64         filtered_rspec = outgoing_rules.apply(rspec)
65
66         return filtered_rspec