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
9 def add_slivers(xml, slivers):
12 if not isinstance(slivers, list):
14 for sliver in slivers:
15 sliver_elem = xml.add_element('sliver_type')
16 if sliver.get('type'):
17 sliver_elem.set('name', sliver['type'])
18 if sliver.get('client_id'):
19 sliver_elem.set('client_id', sliver['client_id'])
20 images = sliver.get('disk_image')
21 if images and isinstance(images, list):
22 PGv2DiskImage.add_images(sliver_elem, images)
23 fw_rules = sliver.get('fw_rules')
24 if fw_rules and isinstance(fw_rules, list):
25 PLOSv1FWRule.add_rules(sliver_elem, fw_rules)
26 PGv2SliverType.add_sliver_attributes(sliver_elem, sliver.get('tags', []))
29 def add_sliver_attributes(xml, 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)
40 def get_slivers(xml, filter={}):
41 xpath = './default:sliver_type | ./sliver_type'
42 sliver_elems = xml.xpath(xpath)
44 for sliver_elem in sliver_elems:
45 sliver = Sliver(sliver_elem.attrib,sliver_elem)
46 if 'component_id' in xml.attrib:
47 sliver['component_id'] = xml.attrib['component_id']
48 if 'name' in sliver_elem.attrib:
49 sliver['type'] = sliver_elem.attrib['name']
50 sliver['disk_image'] = PGv2DiskImage.get_images(sliver_elem)
51 sliver['fw_rules'] = PLOSv1FWRule.get_rules(sliver_elem)
52 slivers.append(sliver)
56 def get_sliver_attributes(xml, filter={}):