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 from sfa.util.sfatime import utcparse, datetime_to_string, datetime_to_epoch
   5
   6 from sfa.rspecs.elements.element import Element
   7 from sfa.rspecs.elements.node import NodeElement
   8 from sfa.rspecs.elements.sliver import Sliver
   9 from sfa.rspecs.elements.location import Location
  10 from sfa.rspecs.elements.hardware_type import HardwareType
  11 from sfa.rspecs.elements.disk_image import DiskImage
  12 from sfa.rspecs.elements.interface import Interface
  13 from sfa.rspecs.elements.bwlimit import BWlimit
  14 from sfa.rspecs.elements.pltag import PLTag
  15 from sfa.rspecs.elements.versions.nitosv1Sliver import NITOSv1Sliver
  16 from sfa.rspecs.elements.versions.nitosv1PLTag import NITOSv1PLTag
  17 from sfa.rspecs.elements.versions.pgv2Services import PGv2Services
  18 from sfa.rspecs.elements.lease import Lease
  19 from sfa.rspecs.elements.channel import Channel
  20
  21
  22
  23 class NITOSv1Lease:
  24
  25     @staticmethod
  26     def add_leases(xml, leases, channels):
  27
  28         network_elems = xml.xpath('//network')
  29         if len(network_elems) > 0:
  30             network_elem = network_elems[0]
  31         elif len(leases) > 0:
  32             network_urn = Xrn(leases[0]['component_id']).get_authority_urn().split(':')[0]
  33             network_elem = xml.add_element('network', name = network_urn)
  34         else:
  35             network_elem = xml
  36
  37         # group the leases by slice and timeslots
  38         grouped_leases = []
  39
  40         while leases:
  41              slice_id = leases[0]['slice_id']
  42              start_time = leases[0]['start_time']
  43              duration = leases[0]['duration']
  44              group = []
  45
  46              for lease in leases:
  47                   if slice_id == lease['slice_id'] and start_time == lease['start_time'] and duration == lease['duration']:
  48                       group.append(lease)
  49
  50              grouped_leases.append(group)
  51
  52              for lease1 in group:
  53                   leases.remove(lease1)
  54
  55         lease_elems = []
  56         for lease in grouped_leases:
  57             #lease[0]['start_time'] = datetime_to_string(utcparse(lease[0]['start_time']))
  58
  59             lease_fields = ['slice_id', 'start_time', 'duration']
  60             lease_elem = network_elem.add_instance('lease', lease[0], lease_fields)
  61             lease_elems.append(lease_elem)
  62
  63             # add nodes of this lease
  64             for node in lease:
  65                  lease_elem.add_instance('node', node, ['component_id'])
  66
  67             # add reserved channels of this lease
  68             #channels = [{'channel_id': 1}, {'channel_id': 2}]
  69             for channel in channels:
  70                  #channel['start_time'] = datetime_to_string(utcparse(channel['start_time']))
  71                  if channel['slice_id'] == lease[0]['slice_id'] and channel['start_time'] == lease[0]['start_time'] and channel['duration'] == lease[0]['duration']:
  72                      lease_elem.add_instance('channel', channel, ['component_id'])
  73
  74
  75     @staticmethod
  76     def get_leases(xml, filter=None):
  77         if filter is None: filter={}
  78         xpath = '//lease%s | //default:lease%s' % (XpathFilter.xpath(filter), XpathFilter.xpath(filter))
  79         lease_elems = xml.xpath(xpath)
  80         return NITOSv1Lease.get_lease_objs(lease_elems)
  81
  82     @staticmethod
  83     def get_lease_objs(lease_elems):
  84         leases = []
  85         channels = []
  86         for lease_elem in lease_elems:
  87             #get nodes
  88             node_elems = lease_elem.xpath('./default:node | ./node')
  89             for node_elem in node_elems:
  90                  lease = Lease(lease_elem.attrib, lease_elem)
  91                  lease['slice_id'] = lease_elem.attrib['slice_id']
  92                  #lease['start_time'] = datetime_to_epoch(utcparse(lease_elem.attrib['start_time']))
  93                  lease['start_time'] = lease_elem.attrib['start_time']
  94                  lease['duration'] = lease_elem.attrib['duration']
  95                  lease['component_id'] = node_elem.attrib['component_id']
  96                  lease['type'] = 'node'
  97                  leases.append(lease)
  98             #get channels
  99             channel_elems = lease_elem.xpath('./default:channel | ./channel')
 100             for channel_elem in channel_elems:
 101                  channel = Channel(channel_elem.attrib, channel_elem)
 102                  channel['slice_id'] = lease_elem.attrib['slice_id']
 103                  #channel['start_time'] = datetime_to_epoch(utcparse(lease_elem.attrib['start_time']))
 104                  channel['start_time'] = lease_elem.attrib['start_time']
 105                  channel['duration'] = lease_elem.attrib['duration']
 106                  channel['component_id'] = channel_elem.attrib['component_id']
 107                  channel['type'] = 'channel'
 108                  channels.append(channel)
 109
 110         return leases + channels
 111