sfa/rspecs/elements/versions/pgv2SliverType.py

   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
   5 class PGv2SliverType:
   6
   7     @staticmethod
   8     def add_slivers(xml, slivers):
   9         if not slivers:
  10             return
  11         if not isinstance(slivers, list):
  12             slivers = [slivers]
  13         for sliver in slivers:
  14             sliver_elem = xml.add_element('sliver_type')
  15             if sliver.get('type'):
  16                 sliver_elem.set('name', sliver['type'])
  17             if sliver.get('client_id'):
  18                 sliver_elem.set('client_id', sliver['client_id'])
  19             images = sliver.get('disk_images')
  20             if images and isinstance(images, list):
  21                 PGv2DiskImage.add_images(sliver_elem, images)
  22             PGv2SliverType.add_sliver_attributes(sliver_elem, sliver.get('tags', []))
  23
  24     @staticmethod
  25     def add_sliver_attributes(xml, attributes):
  26         if attributes:
  27             for attribute in attributes:
  28                 if attribute['name'] == 'initscript':
  29                     xml.add_element('{%s}initscript' % xml.namespaces['planetlab'], name=attribute['value'])
  30                 elif tag['tagname'] == 'flack_info':
  31                     attrib_elem = xml.add_element('{%s}info' % self.namespaces['flack'])
  32                     attrib_dict = eval(tag['value'])
  33                     for (key, value) in attrib_dict.items():
  34                         attrib_elem.set(key, value)
  35     @staticmethod
  36     def get_slivers(xml, filter={}):
  37         xpath = './default:sliver_type | ./sliver_type'
  38         sliver_elems = xml.xpath(xpath)
  39         slivers = []
  40         for sliver_elem in sliver_elems:
  41             sliver = Sliver(sliver_elem.attrib,sliver_elem)
  42             if 'component_id' in xml.attrib:
  43                 sliver['component_id'] = xml.attrib['component_id']
  44             if 'name' in sliver_elem.attrib:
  45                 sliver['type'] = sliver_elem.attrib['name']
  46             sliver['images'] = PGv2DiskImage.get_images(sliver_elem)
  47             slivers.append(sliver)
  48         return slivers
  49
  50     @staticmethod
  51     def get_sliver_attributes(xml, filter={}):
  52         return []