sfa/rspecs/elements/versions/pgv2Lease.py

   1 from sfa.util.xrn import Xrn
   2 from sfa.util.xml import XpathFilter
   3 from sfa.util.sfatime import utcparse, datetime_to_string, datetime_to_epoch
   4
   5 from sfa.rspecs.elements.node import NodeElement
   6 from sfa.rspecs.elements.sliver import Sliver
   7 from sfa.rspecs.elements.location import Location
   8 from sfa.rspecs.elements.hardware_type import HardwareType
   9 from sfa.rspecs.elements.disk_image import DiskImage
  10 from sfa.rspecs.elements.interface import Interface
  11 from sfa.rspecs.elements.bwlimit import BWlimit
  12 from sfa.rspecs.elements.pltag import PLTag
  13 from sfa.rspecs.elements.versions.pgv2Services import PGv2Services
  14 from sfa.rspecs.elements.versions.pgv2SliverType import PGv2SliverType
  15 from sfa.rspecs.elements.versions.pgv2Interface import PGv2Interface
  16 from sfa.rspecs.elements.lease import Lease
  17
  18 from sfa.planetlab.plxrn import xrn_to_hostname
  19
  20 class PGv2Lease:
  21     @staticmethod
  22     def add_leases(xml, leases):
  23         # group the leases by slice and timeslots
  24         grouped_leases = []
  25
  26         while leases:
  27              slice_id = leases[0]['slice_id']
  28              start_time = leases[0]['start_time']
  29              duration = leases[0]['duration']
  30              group = []
  31
  32              for lease in leases:
  33                   if slice_id == lease['slice_id'] and start_time == lease['start_time'] and duration == lease['duration']:
  34                       group.append(lease)
  35
  36              grouped_leases.append(group)
  37
  38              for lease1 in group:
  39                   leases.remove(lease1)
  40
  41         lease_elems = []
  42         for lease in grouped_leases:
  43             lease[0]['start_time'] = datetime_to_string(utcparse(lease[0]['start_time']))
  44
  45             lease_fields = ['slice_id', 'start_time', 'duration']
  46             lease_elem = xml.add_instance('lease', lease[0], lease_fields)
  47             lease_elems.append(lease_elem)
  48
  49             # add nodes of this lease
  50             for node in lease:
  51                  lease_elem.add_instance('node', node, ['component_id'])
  52
  53
  54     @staticmethod
  55     def get_leases(xml, filter={}):
  56         xpath = '//lease%s | //default:lease%s' % (XpathFilter.xpath(filter), XpathFilter.xpath(filter))
  57         lease_elems = xml.xpath(xpath)
  58         return PGv2Lease.get_lease_objs(lease_elems)
  59
  60
  61     @staticmethod
  62     def get_lease_objs(lease_elems):
  63         leases = []
  64         for lease_elem in lease_elems:
  65             #get nodes
  66             node_elems = lease_elem.xpath('./default:node | ./node')
  67             for node_elem in node_elems:
  68                  lease = Lease(lease_elem.attrib, lease_elem)
  69                  lease['slice_id'] = lease_elem.attrib['slice_id']
  70                  lease['start_time'] = datetime_to_epoch(utcparse(lease_elem.attrib['start_time']))
  71                  lease['duration'] = lease_elem.attrib['duration']
  72                  lease['component_id'] = node_elem.attrib['component_id']
  73                  leases.append(lease)
  74
  75         return leases