initial checkin
authorTony Mack <tmack@paris.CS.Princeton.EDU>
Tue, 7 Feb 2012 16:45:12 +0000 (11:45 -0500)
committerTony Mack <tmack@paris.CS.Princeton.EDU>
Tue, 7 Feb 2012 16:47:31 +0000 (11:47 -0500)
sfa/rspecs/elements/versions/pgv2DiskImage.py [new file with mode: 0644]

diff --git a/sfa/rspecs/elements/versions/pgv2DiskImage.py b/sfa/rspecs/elements/versions/pgv2DiskImage.py
new file mode 100644 (file)
index 0000000..51363de
--- /dev/null
@@ -0,0 +1,24 @@
+from sfa.rspecs.elements.element import Element
+from sfa.rspecs.elements.disk_image import DiskImage
+
+class PGv2DiskImage:
+
+    @staticmethod
+    def add_images(xml, images):
+        if not images:
+            return 
+        if not isinstance(images, list):
+            images = [images]
+        for image in images: 
+            xml.add_instance('disk_image', image, DiskImage.fields)
+    
+    @staticmethod
+    def get_images(xml, filter={}):
+        xpath = './default:disk_image | ./disk_image'
+        image_elems = xml.xpath(xpath)
+        images = []
+        for image_elem in image_elems:
+            image = DiskImage(image_elem.attrib, image_elem)
+            images.append(image)
+        return images
+