1 from sfa.rspecs.elements.element import Element
2 from sfa.rspecs.elements.sliver import Sliver
3 from sfa.rspecs.elements.versions.pgv2DiskImage import PGv2DiskImage
4 from sfa.rspecs.elements.versions.plosv1FWRule import PLOSv1FWRule
6 from sfa.util.sfalogging import logger
11 def add_slivers(xml, slivers):
14 if not isinstance(slivers, list):
16 for sliver in slivers:
17 sliver_elem = xml.add_element('sliver_type')
18 if sliver.get('type'):
19 sliver_elem.set('name', sliver['type'])
20 attrs = ['client_id', 'cpus', 'memory', 'storage']
23 sliver_elem.set(attr, sliver[attr])
25 images = sliver.get('disk_image')
26 if images and isinstance(images, list):
27 PGv2DiskImage.add_images(sliver_elem, images)
28 fw_rules = sliver.get('fw_rules')
29 if fw_rules and isinstance(fw_rules, list):
30 PLOSv1FWRule.add_rules(sliver_elem, fw_rules)
31 PGv2SliverType.add_sliver_attributes(sliver_elem, sliver.get('tags', []))
34 def add_sliver_attributes(xml, tags):
38 tagname = tag['tagname'] if 'tagname' in tag else tag['name']
39 if tagname == 'flack_info':
40 attrib_elem = xml.add_element('{%s}info' % self.namespaces['flack'])
42 attrib_dict = eval(tag['value'])
43 for (key, value) in attrib_dict.items():
44 attrib_elem.set(key, value)
45 except Exception as e:
46 logger.warning("Could not parse dictionary in flack tag -- {}".format(e))
47 elif tagname == 'initscript':
48 xml.add_element('{%s}initscript' % xml.namespaces['planetlab'],
51 xml.add_element('{%s}attribute' % (xml.namespaces['planetlab']),
54 scope = tag.get('scope', 'unknown'),
58 def get_slivers(xml, filter=None):
59 if filter is None: filter={}
60 xpath = './default:sliver_type | ./sliver_type'
61 sliver_elems = xml.xpath(xpath)
63 for sliver_elem in sliver_elems:
64 sliver = Sliver(sliver_elem.attrib,sliver_elem)
65 if 'component_id' in xml.attrib:
66 sliver['component_id'] = xml.attrib['component_id']
67 if 'name' in sliver_elem.attrib:
68 sliver['type'] = sliver_elem.attrib['name']
69 sliver['disk_image'] = PGv2DiskImage.get_images(sliver_elem)
70 sliver['fw_rules'] = PLOSv1FWRule.get_rules(sliver_elem)
71 slivers.append(sliver)
75 def get_sliver_attributes(xml, filter=None):
76 if filter is None: filter={}