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