#!/usr/bin/python
-from lxml import etree
-from StringIO import StringIO
-from sfa.util.xrn import *
+from __future__ import print_function
+
+from sfa.util.xrn import hrn_to_urn
from sfa.rspecs.rspec import RSpec
from sfa.rspecs.version_manager import VersionManager
+
class SfaRSpecConverter:
@staticmethod
- def to_pg_rspec(rspec, content_type = None):
+ def to_pg_rspec(rspec, content_type=None):
if not isinstance(rspec, RSpec):
sfa_rspec = RSpec(rspec)
else:
sfa_rspec = rspec
-
+
if not content_type or content_type not in \
- ['ad', 'request', 'manifest']:
+ ['ad', 'request', 'manifest']:
content_type = sfa_rspec.version.content_type
-
-
+
version_manager = VersionManager()
pg_version = version_manager._get_version('protogeni', '2', 'request')
pg_rspec = RSpec(version=pg_version)
-
+
# get networks
networks = sfa_rspec.version.get_networks()
-
+
for network in networks:
# get nodes
- sfa_node_elements = sfa_rspec.version.get_node_elements(network=network)
+ sfa_node_elements = sfa_rspec.version.get_node_elements(
+ network=network)
for sfa_node_element in sfa_node_elements:
# create node element
node_attrs = {}
node_attrs['exclusive'] = 'false'
- node_attrs['component_manager_id'] = network
+ if 'component_manager_id' in sfa_node_element.attrib:
+ node_attrs['component_manager_id'] = sfa_node_element.attrib[
+ 'component_manager_id']
+ else:
+ node_attrs['component_manager_id'] = hrn_to_urn(
+ network, 'authority+cm')
+
+ if 'component_id' in sfa_node_element.attrib:
+ node_attrs['compoenent_id'] = sfa_node_element.attrib[
+ 'component_id']
+
if sfa_node_element.find('hostname') != None:
- node_attrs['component_name'] = sfa_node_element.find('hostname').text
- if sfa_node_element.find('urn') != None:
- node_attrs['component_id'] = sfa_node_element.find('urn').text
+ hostname = sfa_node_element.find('hostname').text
+ node_attrs['component_name'] = hostname
+ node_attrs['client_id'] = hostname
node_element = pg_rspec.xml.add_element('node', node_attrs)
- # create node_type element
- for hw_type in ['plab-pc', 'pc']:
- hdware_type_element = pg_rspec.xml.add_element('hardware_type', {'name': hw_type}, parent=node_element)
- # create available element
- pg_rspec.xml.add_element('available', {'now': 'true'}, parent=node_element)
- # create locaiton element
- # We don't actually associate nodes with a country.
- # Set country to "unknown" until we figure out how to make
- # sure this value is always accurate.
- location = sfa_node_element.find('location')
- if location != None:
- location_attrs = {}
- location_attrs['country'] = location.get('country', 'unknown')
- location_attrs['latitude'] = location.get('latitiue', 'None')
- location_attrs['longitude'] = location.get('longitude', 'None')
- pg_rspec.xml.add_element('location', location_attrs, parent=node_element)
-
- sliver_element = sfa_node_element.find('sliver')
- if sliver_element != None:
- if content_type == 'request':
- # remove all child elements
- for child in sfa_node_element.iterchildren():
- sfa_node_element.remove(child)
- # add the sliver
- pg_rspec.xml.add_element('sliver_type', {'name': 'planetlab-vnode'}, parent=node_element)
+ if content_type == 'request':
+ sliver_element = sfa_node_element.find('sliver')
+ sliver_type_elements = sfa_node_element.xpath(
+ './sliver_type', namespaces=sfa_rspec.namespaces)
+ available_sliver_types = [element.attrib[
+ 'name'] for element in sliver_type_elements]
+ valid_sliver_types = ['emulab-openvz', 'raw-pc']
+
+ # determine sliver type
+ requested_sliver_type = 'emulab-openvz'
+ for available_sliver_type in available_sliver_types:
+ if available_sliver_type in valid_sliver_types:
+ requested_sliver_type = available_sliver_type
+
+ if sliver_element != None:
+ pg_rspec.xml.add_element(
+ 'sliver_type', {'name': requested_sliver_type}, parent=node_element)
+ else:
+ # create node_type element
+ for hw_type in ['plab-pc', 'pc']:
+ hdware_type_element = pg_rspec.xml.add_element(
+ 'hardware_type', {'name': hw_type}, parent=node_element)
+ # create available element
+ pg_rspec.xml.add_element(
+ 'available', {'now': 'true'}, parent=node_element)
+ # create locaiton element
+ # We don't actually associate nodes with a country.
+ # Set country to "unknown" until we figure out how to make
+ # sure this value is always accurate.
+ location = sfa_node_element.find('location')
+ if location != None:
+ location_attrs = {}
+ location_attrs['country'] = location.get(
+ 'country', 'unknown')
+ location_attrs['latitude'] = location.get(
+ 'latitude', 'None')
+ location_attrs['longitude'] = location.get(
+ 'longitude', 'None')
+ pg_rspec.xml.add_element(
+ 'location', location_attrs, parent=node_element)
return pg_rspec.toxml()
if __name__ == '__main__':
import sys
- if len(sys.argv) > 1:
- print SfaRSpecConverter.to_pg_rspec(sys.argv[1])
+ if len(sys.argv) > 1:
+ print(SfaRSpecConverter.to_pg_rspec(sys.argv[1]))