autopep8
[sfa.git] / sfa / rspecs / elements / versions / pgv2DiskImage.py
1 from sfa.rspecs.elements.element import Element
2 from sfa.rspecs.elements.disk_image import DiskImage
3
4
5 class PGv2DiskImage:
6
7     @staticmethod
8     def add_images(xml, images):
9         if not images:
10             return
11         if not isinstance(images, list):
12             images = [images]
13         for image in images:
14             xml.add_instance('disk_image', image, DiskImage.fields)
15
16     @staticmethod
17     def get_images(xml, filter=None):
18         if filter is None:
19             filter = {}
20         xpath = './default:disk_image | ./disk_image'
21         image_elems = xml.xpath(xpath)
22         images = []
23         for image_elem in image_elems:
24             image = DiskImage(image_elem.attrib, image_elem)
25             images.append(image)
26         return images