sfa/rspecs/elements/versions/nitosv1Channel.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 from sfa.rspecs.elements.spectrum import Spectrum
  19 from sfa.rspecs.elements.channel import Channel
  20
  21 from sfa.planetlab.plxrn import xrn_to_hostname
  22
  23 class NITOSv1Channel:
  24
  25     @staticmethod
  26     def add_channels(xml, channels):
  27
  28         network_elems = xml.xpath('//network')
  29         if len(network_elems) > 0:
  30             network_elem = network_elems[0]
  31         elif len(channels) > 0:
  32             #network_urn = Xrn(leases[0]['component_id']).get_authority_urn().split(':')[0]
  33             network_urn = "pla"
  34             network_elem = xml.add_element('network', name = network_urn)
  35         else:
  36             network_elem = xml
  37
  38 #        spectrum_elems = xml.xpath('//spectrum')
  39 #        spectrum_elem = xml.add_element('spectrum')
  40
  41 #        if len(spectrum_elems) > 0:
  42 #            spectrum_elem = spectrum_elems[0]
  43 #        elif len(channels) > 0:
  44 #            spectrum_elem = xml.add_element('spectrum')
  45 #        else:
  46 #            spectrum_elem = xml
  47
  48         spectrum_elem = network_elem.add_instance('spectrum', [])
  49
  50         channel_elems = []
  51         for channel in channels:
  52             channel_fields = ['channel_num', 'frequency', 'standard']
  53             channel_elem = spectrum_elem.add_instance('channel', channel, channel_fields)
  54             channel_elems.append(channel_elem)
  55
  56
  57     @staticmethod
  58     def get_channels(xml, filter={}):
  59         xpath = '//channel%s | //default:channel%s' % (XpathFilter.xpath(filter), XpathFilter.xpath(filter))
  60         channel_elems = xml.xpath(xpath)
  61         return NITOSv1Channel.get_channel_objs(channel_elems)
  62
  63     @staticmethod
  64     def get_channel_objs(channel_elems):
  65         channels = []
  66         for channel_elem in channel_elems:
  67             channel = Channel(channel_elem.attrib, channel_elem)
  68             channel['channel_num'] = channel_elem.attrib['channel_num']
  69             channel['frequency'] = channel_elem.attrib['frequency']
  70             channel['standard'] = channel_elem.attrib['standard']
  71
  72             channels.append(channel)
  73         return channels
  74