refactored
authorTony Mack <tmack@paris.CS.Princeton.EDU>
Thu, 10 Nov 2011 22:19:17 +0000 (17:19 -0500)
committerTony Mack <tmack@paris.CS.Princeton.EDU>
Thu, 10 Nov 2011 22:19:17 +0000 (17:19 -0500)
sfa/rspecs/elements/versions/pgv2SliverType.py

index 0ea039d..70878ff 100644 (file)
@@ -4,8 +4,14 @@ from sfa.rspecs.elements.sliver import Sliver
 class PGv2SliverType:
 
     @staticmethod
-    def add_sliver(xml, sliver):
-        sliver_elem = Element.add(xml, 'sliver_type', sliver, ['name'])
+    def add_slivers(xml, sliver):
+        if not isinstance(slivers, list):
+            slivers = [slivers]
+        for sliver in slivers: 
+            sliver_elem = Element.add(xml, 'sliver_type', sliver, ['type', 'client_id'])
+            for tag in sliver.get('tags', []):
+                if tag['name'] == 'initscript':
+                    sliver_elem.add_element('{%s}initscript' % xml.namespaces['planetlab'], name=tag['value'])
                     
     @staticmethod
     def get_slivers(xml, filter={}):