Merge branch 'upstreammaster'

[sfa.git] / sfa / methods / get_aggregates.py

diff --git a/sfa/methods/get_aggregates.py b/sfa/methods/get_aggregates.py
index c057fa0..23c8d60 100644 (file)
--- a/sfa/methods/get_aggregates.py
+++ b/sfa/methods/get_aggregates.py
@@ -1,11 +1,6 @@
-### $Id: get_slices.py 14387 2009-07-08 18:19:11Z faiyaza $
-### $URL: https://svn.planet-lab.org/svn/sfa/trunk/sfa/methods/get_aggregates.py $
-
-from sfa.util.faults import *
-from sfa.util.misc import *
+from sfa.util.xrn import urn_to_hrn
 from sfa.util.method import Method
 from sfa.util.parameter import Parameter, Mixed
-from sfa.trust.auth import Auth
 from sfa.server.aggregate import Aggregates
 
 class get_aggregates(Method):
@@ -13,7 +8,7 @@ class get_aggregates(Method):
     Get a list of connection information for all known aggregates.      
 
     @param cred credential string specifying the rights of the caller
-    @param a Human readable name (hrn), or list of hrns or None
+    @param a Human readable name (hrn or urn), or list of hrns or None
     @return list of dictionaries with aggregate information.  
     """
 
@@ -21,15 +16,16 @@ class get_aggregates(Method):
     
     accepts = [
         Parameter(str, "Credential string"),
-        Mixed([Parameter(str, "Human readable name (hrn)")],
-              Parameter(str, "Human readable name (hrn)"),
-              Parameter(None, "hrn not specified"))  
+        Mixed(Parameter(str, "Human readable name (hrn or urn)"),
+              Parameter(None, "hrn not specified"))
         ]
 
     returns = [Parameter(dict, "Aggregate interface information")]
     
-    def call(self, cred, hrn = None):
-       
+    def call(self, cred, xrn = None):
+        hrn, type = urn_to_hrn(xrn)
         self.api.auth.check(cred, 'list')
-        aggregates = Aggregates(self.api)
-        return aggregates.interfaces
+        aggregates = Aggregates(self.api).interfaces.values()
+        if hrn:
+            aggregates = [agg for agg in aggregates if agg['hrn'] == hrn]
+        return aggregates