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 from sfa.rspecs.elements.versions.plosv1FWRule import PLOSv1FWRule
   5
   6 from sfa.util.sfalogging import logger
   7
   8
   9 class PGv2SliverType:
  10
  11     @staticmethod
  12     def add_slivers(xml, slivers):
  13         if not slivers:
  14             return
  15         if not isinstance(slivers, list):
  16             slivers = [slivers]
  17         for sliver in slivers:
  18             sliver_elem = xml.add_element('sliver_type')
  19             if sliver.get('type'):
  20                 sliver_elem.set('name', sliver['type'])
  21             attrs = ['client_id', 'cpus', 'memory', 'storage']
  22             for attr in attrs:
  23                 if sliver.get(attr):
  24                     sliver_elem.set(attr, sliver[attr])
  25
  26             images = sliver.get('disk_image')
  27             if images and isinstance(images, list):
  28                 PGv2DiskImage.add_images(sliver_elem, images)
  29             fw_rules = sliver.get('fw_rules')
  30             if fw_rules and isinstance(fw_rules, list):
  31                 PLOSv1FWRule.add_rules(sliver_elem, fw_rules)
  32             PGv2SliverType.add_sliver_attributes(
  33                 sliver_elem, sliver.get('tags', []))
  34
  35     @staticmethod
  36     def add_sliver_attributes(xml, tags):
  37         if tags is None:
  38             return
  39         for tag in tags:
  40             tagname = tag['tagname'] if 'tagname' in tag else tag['name']
  41             if tagname == 'flack_info':
  42                 attrib_elem = xml.add_element(
  43                     '{%s}info' % self.namespaces['flack'])
  44                 try:
  45                     attrib_dict = eval(tag['value'])
  46                     for (key, value) in list(attrib_dict.items()):
  47                         attrib_elem.set(key, value)
  48                 except Exception as e:
  49                     logger.warning(
  50                         "Could not parse dictionary in flack tag -- {}".format(e))
  51             elif tagname == 'initscript':
  52                 xml.add_element('{%s}initscript' % xml.namespaces['planetlab'],
  53                                 name=tag['value'])
  54             else:
  55                 xml.add_element('{%s}attribute' % (xml.namespaces['planetlab']),
  56                                 name=tagname,
  57                                 value=tag['value'],
  58                                 scope=tag.get('scope', 'unknown'),
  59                                 )
  60
  61     @staticmethod
  62     def get_slivers(xml, filter=None):
  63         if filter is None:
  64             filter = {}
  65         xpath = './default:sliver_type | ./sliver_type'
  66         sliver_elems = xml.xpath(xpath)
  67         slivers = []
  68         for sliver_elem in sliver_elems:
  69             sliver = Sliver(sliver_elem.attrib, sliver_elem)
  70             if 'component_id' in xml.attrib:
  71                 sliver['component_id'] = xml.attrib['component_id']
  72             if 'name' in sliver_elem.attrib:
  73                 sliver['type'] = sliver_elem.attrib['name']
  74             sliver['disk_image'] = PGv2DiskImage.get_images(sliver_elem)
  75             sliver['fw_rules'] = PLOSv1FWRule.get_rules(sliver_elem)
  76             slivers.append(sliver)
  77         return slivers
  78
  79     @staticmethod
  80     def get_sliver_attributes(xml, filter=None):
  81         if filter is None:
  82             filter = {}
  83         return []