sfa/rspecs/pg_rspec_converter.py

   1 #!/usr/bin/python
   2 from lxml import etree
   3 from StringIO import StringIO
   4 from sfa.util.xrn import Xrn, urn_to_hrn
   5 from sfa.rspecs.rspec import RSpec
   6 from sfa.rspecs.version_manager import VersionManager
   7
   8 xslt='''<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
   9 <xsl:output method="xml" indent="no"/>
  10
  11 <xsl:template match="/|comment()|processing-instruction()">
  12     <xsl:copy>
  13       <xsl:apply-templates/>
  14     </xsl:copy>
  15 </xsl:template>
  16
  17 <xsl:template match="*">
  18     <xsl:element name="{local-name()}">
  19       <xsl:apply-templates select="@*|node()"/>
  20     </xsl:element>
  21 </xsl:template>
  22
  23 <xsl:template match="@*">
  24     <xsl:attribute name="{local-name()}">
  25       <xsl:value-of select="."/>
  26     </xsl:attribute>
  27 </xsl:template>
  28 </xsl:stylesheet>
  29 '''
  30
  31 xslt_doc=etree.parse(StringIO(xslt))
  32 transform=etree.XSLT(xslt_doc)
  33
  34 class PGRSpecConverter:
  35
  36     @staticmethod
  37     def to_sfa_rspec(rspec, content_type = None):
  38         if not isinstance(rspec, RSpec):
  39             pg_rspec = RSpec(rspec)
  40         else:
  41             pg_rspec = rspec
  42
  43         version_manager = VersionManager()
  44         sfa_version = version_manager._get_version('sfa', '1')
  45         sfa_rspec = RSpec(version=sfa_version)
  46
  47         #nodes = pg_rspec.version.get_nodes()
  48         #sfa_rspec.version.add_nodes(nodes())
  49         #sfa_rspec.version.add_links(pg_rspec.version.get_links())
  50         #return sfa_rspec.toxml()
  51
  52         # get network
  53         networks = pg_rspec.version.get_networks()
  54         network_hrn = networks[0]
  55         network_element = sfa_rspec.xml.add_element('network', name=network_hrn, id=network_hrn)
  56
  57         # get nodes
  58         pg_nodes_elements = pg_rspec.version.get_node_elements()
  59         nodes_with_slivers = pg_rspec.version.get_nodes_with_slivers()
  60         i = 1
  61         for pg_node_element in pg_nodes_elements:
  62             attribs = dict(pg_node_element.attrib.items())
  63             attribs['id'] = 'n'+str(i)
  64
  65             node_element = network_element.add_element('node')
  66             for attrib in attribs:
  67                 node_element.set(attrib, attribs[attrib])
  68             urn = pg_node_element.xpath('@component_id', namespaces=pg_rspec.namespaces)
  69             if urn:
  70                 urn = urn[0]
  71                 hostname = Xrn.urn_split(urn)[-1]
  72                 hostname_element = node_element.add_element('hostname')
  73                 hostname_element.set_text(hostname)
  74                 if hostname in nodes_with_slivers:
  75                     node_element.add_element('sliver')
  76
  77             # just copy over remaining child elements
  78             for child in pg_node_element.getchildren():
  79                 node_element.append(transform(child).getroot())
  80             i = i+1
  81
  82         return sfa_rspec.toxml()
  83
  84 if __name__ == '__main__':
  85     import sys
  86     if len(sys.argv) > 1:
  87         print PGRSpecConverter.to_sfa_rspec(sys.argv[1])