269de566413c7203084d5ed153169e02f57d19b0
[sfa.git] / sfa / rspecs / elements / versions / iotlabv1Sliver.py
1 from __future__ import print_function
2
3 from sfa.rspecs.elements.element import Element
4 from sfa.rspecs.elements.sliver import Sliver
5
6 #from sfa.rspecs.elements.versions.pgv2DiskImage import PGv2DiskImage
7 import sys
8 class Iotlabv1Sliver:
9
10     @staticmethod
11     def add_slivers(xml, slivers):
12         if not slivers:
13             return
14         if not isinstance(slivers, list):
15             slivers = [slivers]
16         for sliver in slivers:
17             #sliver_elem = xml.add_element('sliver_type')
18             sliver_elem = xml.add_element('sliver')
19             if sliver.get('type'):
20                 sliver_elem.set('name', sliver['type'])
21             if sliver.get('client_id'):
22                 sliver_elem.set('client_id', sliver['client_id'])
23             #images = sliver.get('disk_images')
24             #if images and isinstance(images, list):
25                 #Iotlabv1DiskImage.add_images(sliver_elem, images)
26             Iotlabv1Sliver.add_sliver_attributes(sliver_elem, sliver.get('tags', []))
27
28     @staticmethod
29     def add_sliver_attributes(xml, attributes):
30         if attributes:
31             for attribute in attributes:
32                 if attribute['name'] == 'initscript':
33                     xml.add_element('{%s}initscript' % xml.namespaces['planetlab'], name=attribute['value'])
34                 elif tag['tagname'] == 'flack_info':
35                     attrib_elem = xml.add_element('{%s}info' % self.namespaces['flack'])
36                     attrib_dict = eval(tag['value'])
37                     for (key, value) in attrib_dict.items():
38                         attrib_elem.set(key, value)
39     @staticmethod
40     def get_slivers(xml, filter=None):
41         if filter is None: filter={}
42         xpath = './default:sliver | ./sliver'
43
44         sliver_elems = xml.xpath(xpath)
45         slivers = []
46         for sliver_elem in sliver_elems:
47             sliver = Sliver(sliver_elem.attrib,sliver_elem)
48
49             if 'component_id' in xml.attrib:
50                 sliver['component_id'] = xml.attrib['component_id']
51             if 'name' in sliver_elem.attrib:
52                 sliver['type'] = sliver_elem.attrib['name']
53             #sliver['images'] = Iotlabv1DiskImage.get_images(sliver_elem)
54
55             print("\r\n \r\n SLABV1SLIVER.PY  \t\t\t  get_slivers sliver %s " %( sliver), file=sys.stderr)
56             slivers.append(sliver)
57         return slivers
58
59     @staticmethod
60     def get_sliver_attributes(xml, filter=None):
61         if filter is None: filter={}
62         return []