X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Frspecs%2Fsfa_rspec_converter.py;h=297d371b60ddad6bdcda6fb790e16490b66307c6;hb=31cc4fa991896f000abb106916662923f37bc3a6;hp=972835fbdd4e83adcc9c69791278c8de34defc95;hpb=1ea72ca96f0f43bd80baa9fb41ec930ea02de271;p=sfa.git diff --git a/sfa/rspecs/sfa_rspec_converter.py b/sfa/rspecs/sfa_rspec_converter.py index 972835fb..297d371b 100755 --- a/sfa/rspecs/sfa_rspec_converter.py +++ b/sfa/rspecs/sfa_rspec_converter.py @@ -1,62 +1,101 @@ -#!/usr/bin/python +#!/usr/bin/env python2 + +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 -from lxml import etree -from StringIO import StringIO -from sfa.util.xrn import * -from sfa.rspecs.sfa_rspec import SfaRSpec -from sfa.rspecs.pg_rspec import PGRSpec class SfaRSpecConverter: @staticmethod - def to_pg_rspec(rspec): - if isinstance(rspec, SfaRSpec): - sfa_rspec = rspec + def to_pg_rspec(rspec, content_type=None): + if not isinstance(rspec, RSpec): + sfa_rspec = RSpec(rspec) else: - sfa_rspec = SfaRSpec(rspec=rspec) - pg_rspec = PGRSpec() - + sfa_rspec = rspec + + if not content_type or content_type not in \ + ['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.get_networks() - + networks = sfa_rspec.version.get_networks() + for network in networks: # get nodes - sfa_node_elements = sfa_rspec.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 - node_element = pg_rspec.add_element('node', node_attrs) - - # create node_type element - for hw_type in ['plab-pc', 'pc']: - hdware_type_element = pg_rspec.add_element('hardware_type', {'name': hw_type}, parent=node_element) - # create available element - pg_rspec.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.add_element('location', location_attrs, parent=node_element) - - sliver_element = sfa_node_element.find('sliver') - if sliver_element != None: - pg_rspec.add_element('sliver_type', {'name': 'planetlab-vnode'}, parent=node_element) + 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) + + 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]))