X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Frspecs%2Frspec_converter.py;h=49555c832335a7706a7cffb16f27cf1286c71da2;hb=31cc4fa991896f000abb106916662923f37bc3a6;hp=8a661089943f2e92db769800935abdc1fba3264f;hpb=658da4bc128e776d210b28141194cb63e46660a6;p=sfa.git

diff --git a/sfa/rspecs/rspec_converter.py b/sfa/rspecs/rspec_converter.py
index 8a661089..49555c83 100755
--- a/sfa/rspecs/rspec_converter.py
+++ b/sfa/rspecs/rspec_converter.py
@@ -1,41 +1,49 @@
-#!/usr/bin/python
+#!/usr/bin/env python2
+
+from __future__ import print_function
 
 from sfa.rspecs.pg_rspec_converter import PGRSpecConverter
 from sfa.rspecs.sfa_rspec_converter import SfaRSpecConverter
-from sfa.rspecs.sfa_rspec import sfa_rspec_version
-from sfa.rspecs.pg_rspec import pg_rspec_version
-from sfa.rspecs.rspec_parser import parse_rspec
+from sfa.rspecs.rspec import RSpec
+from sfa.rspecs.version_manager import VersionManager
 
 
 class RSpecConverter:
 
     @staticmethod
-    def to_sfa_rspec(in_rspec):
-        rspec = parse_rspec(in_rspec)
-        if rspec.version['type'] == sfa_rspec_version['type']: 
-          return in_rspec
-        elif rspec.version['type'] == pg_rspec_version['type']:
-            return PGRSpecConverter.to_sfa_rspec(in_rspec)
+    def to_sfa_rspec(in_rspec, content_type=None):
+        rspec = RSpec(in_rspec)
+        version_manager = VersionManager()
+        sfa_version = version_manager._get_version('sfa', '1')
+        pg_version = version_manager._get_version('protogeni', '2')
+        if rspec.version.type.lower() == sfa_version.type.lower():
+            return in_rspec
+        elif rspec.version.type.lower() == pg_version.type.lower():
+            return PGRSpecConverter.to_sfa_rspec(in_rspec, content_type)
         else:
-             return in_rspec 
+            return in_rspec
 
-    @staticmethod 
-    def to_pg_rspec(in_rspec):
-        rspec = parse_rspec(in_rspec)
-        if rspec.version['type'] == pg_rspec_version['type']:
+    @staticmethod
+    def to_pg_rspec(in_rspec, content_type=None):
+        rspec = RSpec(in_rspec)
+        version_manager = VersionManager()
+        sfa_version = version_manager._get_version('sfa', '1')
+        pg_version = version_manager._get_version('protogeni', '2')
+
+        if rspec.version.type.lower() == pg_version.type.lower():
             return in_rspec
-        elif rspec.version['type'] == sfa_rspec_version['type']:
-            return SfaRSpecConverter.to_pg_rspec(in_rspec)
+        elif rspec.version.type.lower() == sfa_version.type.lower():
+            return SfaRSpecConverter.to_pg_rspec(in_rspec, content_type)
         else:
-            return in_rspec 
+            return in_rspec
 
 
 if __name__ == '__main__':
     pg_rspec = 'test/protogeni.rspec'
-    sfa_rspec = 'test/nodes.rspec'  
+    sfa_rspec = 'test/nodes.rspec'
+
+    print("converting pg rspec to sfa rspec")
+    print(RSpecConverter.to_sfa_rspec(pg_rspec))
 
-    print "converting pg rspec to sfa rspec"
-    print RSpecConverter.to_sfa_rspec(pg_rspec)
-    
-    print "converting sfa rspec to pg rspec"
-    print RSpecConverter.to_pg_rspec(sfa_rspec)                   
+    print("converting sfa rspec to pg rspec")
+    print(RSpecConverter.to_pg_rspec(sfa_rspec))