X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Frspecs%2Frspec_converter.py;h=a24b2def22c7b68843ef8e220a276570b342ecf9;hb=47090c0a12d8ccffa6d67692461da61a7b7f67e7;hp=5301c2d85602e7e336140b2bb183214e56e6c413;hpb=0ae6f1e2beecf53d48b7d5e049b4f54e7953a675;p=sfa.git diff --git a/sfa/rspecs/rspec_converter.py b/sfa/rspecs/rspec_converter.py index 5301c2d8..a24b2def 100755 --- a/sfa/rspecs/rspec_converter.py +++ b/sfa/rspecs/rspec_converter.py @@ -1,39 +1,49 @@ #!/usr/bin/python +from __future__ import print_function + from sfa.rspecs.pg_rspec_converter import PGRSpecConverter from sfa.rspecs.sfa_rspec_converter import SfaRSpecConverter -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'].lower() == 'sfa': - return in_rspec - elif rspec.version['type'].lower() == 'protogeni': - 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'].lower() == 'protogeni': + @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'].lower() == 'sfa': - 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))