set the 'name' attribute in the sliver_type element
[sfa.git] / sfa / rspecs / elements / versions / pgv2SliverType.py
1 from sfa.rspecs.elements.element import Element
2 from sfa.rspecs.elements.sliver import Sliver
3
4 class PGv2SliverType:
5
6     @staticmethod
7     def add_slivers(xml, slivers):
8         if not slivers:
9             return 
10         if not isinstance(slivers, list):
11             slivers = [slivers]
12         for sliver in slivers: 
13             sliver_elem = xml.add_element('sliver_type')
14             if sliver.get('type'):
15                 sliver_elem.set('name', sliver['type'])
16             if sliver.get('client_id'):
17                 sliver_elem.set('client_id', sliver['client_id'])  
18             PGv2SliverType.add_sliver_attributes(sliver_elem, sliver.get('pl_tags', []))
19     
20     @staticmethod
21     def add_sliver_attributes(xml, attributes):
22         for attribute in attributes:
23             if attribute['name'] == 'initscript':
24                 xml.add_element('{%s}initscript' % xml.namespaces['planetlab'], name=attribute['value'])
25             elif tag['tagname'] == 'flack_info':
26                 attrib_elem = xml.add_element('{%s}info' % self.namespaces['flack'])
27                 attrib_dict = eval(tag['value'])
28                 for (key, value) in attrib_dict.items():
29                     attrib_elem.set(key, value)                
30     @staticmethod
31     def get_slivers(xml, filter={}):
32         xpath = './default:sliver_type | ./sliver_type'
33         sliver_elems = xml.xpath(xpath)
34         slivers = []
35         for sliver_elem in sliver_elems:
36             sliver = Sliver(sliver_elem.attrib,sliver_elem)
37             if 'component_id' in xml.attrib:     
38                 sliver['component_id'] = xml.attrib['component_id']
39             if 'name' in sliver_elem.attrib:
40                 sliver['type'] = sliver_elem.attrib['name']
41             slivers.append(sliver)
42         return slivers
43
44     @staticmethod
45     def get_sliver_attributes(xml, filter={}):
46         return []