sfa/rspecs/elements/versions/iotlabv1Lease.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
   7 #from sfa.rspecs.elements.versions.sfav1PLTag import SFAv1PLTag
   8 #from sfa.rspecs.elements.versions.pgv2Services import PGv2Services
   9 from sfa.rspecs.elements.lease import Lease
  10
  11
  12
  13 class Iotlabv1Lease:
  14
  15     @staticmethod
  16     def add_leases(xml, leases):
  17
  18         network_elems = xml.xpath('//network')
  19         if len(network_elems) > 0:
  20             network_elem = network_elems[0]
  21         elif len(leases) > 0:
  22             network_urn = Xrn(leases[0]['component_id']).get_authority_urn().split(':')[0]
  23             network_elem = xml.add_element('network', name = network_urn)
  24         else:
  25             network_elem = xml
  26
  27         lease_elems = []
  28         for lease in leases:
  29             lease['start_time'] = datetime_to_string(utcparse(lease['start_time']))
  30
  31             lease_fields = ['lease_id', 'component_id', 'slice_id', 'start_time', 'duration']
  32             lease_elem = network_elem.add_instance('lease', lease, lease_fields)
  33             lease_elems.append(lease_elem)
  34
  35
  36     @staticmethod
  37     def get_leases(xml, filter={}):
  38         xpath = '//lease%s | //default:lease%s' % (XpathFilter.xpath(filter), XpathFilter.xpath(filter))
  39         lease_elems = xml.xpath(xpath)
  40         return Iotlabv1Lease.get_lease_objs(lease_elems)
  41
  42     @staticmethod
  43     def get_lease_objs(lease_elems):
  44         leases = []
  45         for lease_elem in lease_elems:
  46             #get nodes
  47             node_elems = lease_elem.xpath('./default:node | ./node')
  48             for node_elem in node_elems:
  49                  lease = Lease(lease_elem.attrib, lease_elem)
  50                  lease['slice_id'] = lease_elem.attrib['slice_id']
  51                  lease['start_time'] = datetime_to_epoch(utcparse(lease_elem.attrib['start_time']))
  52                  lease['duration'] = lease_elem.attrib['duration']
  53                  lease['component_id'] = node_elem.attrib['component_id']
  54                  leases.append(lease)
  55
  56         return leases