X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Frspecs%2Fpg_rspec_converter.py;h=7de30efd988da896863b4adb18f3d5cc274f230e;hb=fd395e1944dcd49f10a4d5b27ce4983ad389fb96;hp=6f47549e9f6e80112f23b5aa7598b6baca37e464;hpb=cc799d820c7107c55137ee31dc0208c3c579634d;p=sfa.git

diff --git a/sfa/rspecs/pg_rspec_converter.py b/sfa/rspecs/pg_rspec_converter.py
index 6f47549e..7de30efd 100755
--- a/sfa/rspecs/pg_rspec_converter.py
+++ b/sfa/rspecs/pg_rspec_converter.py
@@ -1,11 +1,14 @@
-#!/usr/bin/python 
+#!/usr/bin/python
+from __future__ import print_function
+
 from lxml import etree
-from StringIO import StringIO
-from sfa.util.xrn import *
-from sfa.rspecs.pg_rspec import PGRSpec 
-from sfa.rspecs.sfa_rspec import SfaRSpec
+from sfa.util.xrn import Xrn, urn_to_hrn
+from sfa.rspecs.rspec import RSpec
+from sfa.rspecs.version_manager import VersionManager
+
+from sfa.util.py23 import StringIO
 
-xslt='''<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
+xslt = '''<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
 <xsl:output method="xml" indent="no"/>
 
 <xsl:template match="/|comment()|processing-instruction()">
@@ -28,38 +31,70 @@ xslt='''<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Tran
 </xsl:stylesheet>
 '''
 
-xslt_doc=etree.parse(StringIO(xslt))
-transform=etree.XSLT(xslt_doc)
+xslt_doc = etree.parse(StringIO(xslt))
+transform = etree.XSLT(xslt_doc)
+
 
 class PGRSpecConverter:
 
     @staticmethod
-    def to_sfa_rspec(rspec):
-        pg_rspec = PGRSpec(rspec=rspec)
-        sfa_rspec = SfaRSpec()
+    def to_sfa_rspec(rspec, content_type=None):
+        if not isinstance(rspec, RSpec):
+            pg_rspec = RSpec(rspec)
+        else:
+            pg_rspec = rspec
+
+        version_manager = VersionManager()
+        sfa_version = version_manager._get_version('sfa', '1')
+        sfa_rspec = RSpec(version=sfa_version)
+
+        #nodes = pg_rspec.version.get_nodes()
+        # sfa_rspec.version.add_nodes(nodes())
+        # sfa_rspec.version.add_links(pg_rspec.version.get_links())
+        # return sfa_rspec.toxml()
 
         # get network
-        network_urn = pg_rspec.get_network()
-        network,  _ = urn_to_hrn(network_urn)
-        network_element = sfa_rspec.add_element('network', {'name': network, 'id': network})
-        
+        networks = pg_rspec.version.get_networks()
+        network_hrn = networks[0]["name"]
+        network_element = sfa_rspec.xml.add_element(
+            'network', name=network_hrn, id=network_hrn)
+
         # get nodes
-        pg_nodes_elements = pg_rspec.get_node_elements()
+        pg_nodes_elements = pg_rspec.version.get_nodes()
+        nodes_with_slivers = pg_rspec.version.get_nodes_with_slivers()
         i = 1
-        for pg_node_element in pg_nodes_elements:
-            urn = pg_node_element.get('component_uuid')
-            hostname = Xrn.urn_split(urn)[-1]
-            node_element = sfa_rspec.add_element('node', {'id': 'n'+str(i)}, parent=network_element)
-            hostname_element = sfa_rspec.add_element('hostname', parent=node_element, text=hostname) 
-            urn_element = sfa_rspec.add_element('urn', parent=node_element, text=urn)
-            for child in pg_node_element.getchildren():
-                node_element.append(transform(child).getroot())
-            i = i+1
-            break 
- 
+        for pg_node in pg_nodes_elements:
+            attribs = dict(pg_node.items())
+            attribs['id'] = 'n' + str(i)
+
+            node_element = network_element.add_element('node')
+            for attrib in attribs:
+                if type(attribs[attrib]) == str:
+                    node_element.set(attrib, attribs[attrib])
+            urn = pg_node["component_id"]
+            if urn:
+                if type(urn) == list:
+                    # legacy code, not sure if urn is ever a list...
+                    urn = urn[0]
+                hostname = Xrn.urn_split(urn)[-1]
+                hostname_element = node_element.add_element('hostname')
+                hostname_element.set_text(hostname)
+                if hostname in nodes_with_slivers:
+                    node_element.add_element('sliver')
+
+            for hardware_type in pg_node["hardware_types"]:
+                if "name" in hardware_type:
+                    node_element.add_element(
+                        "hardware_type", name=hardware_type["name"])
+
+            # just copy over remaining child elements
+            # for child in pg_node_element.getchildren():
+            #    node_element.append(transform(child).getroot())
+            i = i + 1
+
         return sfa_rspec.toxml()
-         
+
 if __name__ == '__main__':
     import sys
-    if len(sys.argv) > 1:        
-        print PGRSpecConverter.to_sfa_rspec(sys.argv[1])  
+    if len(sys.argv) > 1:
+        print(PGRSpecConverter.to_sfa_rspec(sys.argv[1]))