sfa/rspecs/elements/versions/nitosv1Lease.py

   1 from sfa.util.sfalogging import logger
   2 from sfa.util.xml import XpathFilter
   3 from sfa.util.xrn import Xrn
   4
   5 from sfa.rspecs.elements.element import Element
   6 from sfa.rspecs.elements.node import Node
   7 from sfa.rspecs.elements.sliver import Sliver
   8 from sfa.rspecs.elements.location import Location
   9 from sfa.rspecs.elements.hardware_type import HardwareType
  10 from sfa.rspecs.elements.disk_image import DiskImage
  11 from sfa.rspecs.elements.interface import Interface
  12 from sfa.rspecs.elements.bwlimit import BWlimit
  13 from sfa.rspecs.elements.pltag import PLTag
  14 from sfa.rspecs.elements.versions.nitosv1Sliver import NITOSv1Sliver
  15 from sfa.rspecs.elements.versions.nitosv1PLTag import NITOSv1PLTag
  16 from sfa.rspecs.elements.versions.pgv2Services import PGv2Services
  17 from sfa.rspecs.elements.lease import Lease
  18
  19 from sfa.nitos.nitosxrn import xrn_to_hostname
  20
  21 class NITOSv1Lease:
  22
  23     @staticmethod
  24     def add_leases(xml, leases, channels):
  25
  26         network_elems = xml.xpath('//network')
  27         if len(network_elems) > 0:
  28             network_elem = network_elems[0]
  29         elif len(leases) > 0:
  30             network_urn = Xrn(leases[0]['component_id']).get_authority_urn().split(':')[0]
  31             network_elem = xml.add_element('network', name = network_urn)
  32         else:
  33             network_elem = xml
  34
  35         # group the leases by slice and timeslots
  36         grouped_leases = []
  37
  38         while leases:
  39              slice_id = leases[0]['slice_id']
  40              start_time = leases[0]['start_time']
  41              duration = leases[0]['duration']
  42              group = []
  43
  44              for lease in leases:
  45                   if slice_id == lease['slice_id'] and start_time == lease['start_time'] and duration == lease['duration']:
  46                       group.append(lease)
  47
  48              grouped_leases.append(group)
  49
  50              for lease1 in group:
  51                   leases.remove(lease1)
  52
  53         lease_elems = []
  54         for lease in grouped_leases:
  55             #lease_fields = ['lease_id', 'component_id', 'slice_id', 'start_time', 'duration']
  56             lease_fields = ['slice_id', 'start_time', 'duration']
  57             lease_elem = network_elem.add_instance('lease', lease[0], lease_fields)
  58             lease_elems.append(lease_elem)
  59
  60             # add nodes of this lease
  61             for node in lease:
  62                  lease_elem.add_instance('node', node, ['component_id'])
  63
  64             # add reserved channels of this lease
  65             #channels = [{'channel_id': 1}, {'channel_id': 2}]
  66             for channel in channels:
  67                  if channel['slice_id'] == lease[0]['slice_id'] and channel['start_time'] == lease[0]['start_time'] and channel['duration'] == lease[0]['duration']:
  68                      lease_elem.add_instance('channel', channel, ['channel_num'])
  69
  70
  71     @staticmethod
  72     def get_leases(xml, filter={}):
  73         xpath = '//lease%s | //default:lease%s' % (XpathFilter.xpath(filter), XpathFilter.xpath(filter))
  74         lease_elems = xml.xpath(xpath)
  75         return NITOSv1Lease.get_lease_objs(lease_elems)
  76
  77     @staticmethod
  78     def get_lease_objs(lease_elems):
  79         leases = []
  80         for lease_elem in lease_elems:
  81             lease = Lease(lease_elem.attrib, lease_elem)
  82             if lease.get('lease_id'):
  83                lease['lease_id'] = lease_elem.attrib['lease_id']
  84             lease['component_id'] = lease_elem.attrib['component_id']
  85             lease['slice_id'] = lease_elem.attrib['slice_id']
  86             lease['start_time'] = lease_elem.attrib['start_time']
  87             lease['duration'] = lease_elem.attrib['duration']
  88
  89             leases.append(lease)
  90         return leases
  91