harbor/tools/migration/cfg/run.py

71 lines
2.8 KiB
Python
Raw Normal View History

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import print_function
import argparse
import os
import sys
import utils
import importlib
import glob
import shutil
import sys
def main():
target_version = '1.9.0'
parser = argparse.ArgumentParser(description='migrator of harbor.cfg')
parser.add_argument('--input', '-i', action="store", dest='input_path', required=True, help='The path to the old harbor.cfg that provides input value, this required value')
parser.add_argument('--output','-o', action="store", dest='output_path', required=False, help='The path of the migrated harbor.cfg, if not set the input file will be overwritten')
parser.add_argument('--target', action="store", dest='target_version', help='The target version that the harbor.cfg will be migrated to.')
args = parser.parse_args()
if args.target_version is not None:
target_version = args.target_version
input_version = utils.get_conf_version(args.input_path)
curr_dir = os.path.dirname(__file__)
if input_version == target_version:
print ("Version of input harbor.yml is identical to target %s, no need to upgrade" % input_version)
sys.exit(0)
if args.output_path is None:
if float(input_version[:input_version.rfind('.')]) < 1.8 and \
float(target_version[:target_version.rfind('.')]) >= 1.8:
print("Migrating config file from < 1.8.0 to >= 1.8.0, the output path cannot be empty")
sys.exit(1)
args.output_path = args.input_path
chain = []
if not search(curr_dir, input_version, target_version, chain):
print ("No migrator for version: %s" % input_version)
sys.exit(1)
else:
print ("input version: %s, migrator chain: %s" % (input_version, chain))
curr_input_path = args.input_path
for c in chain:
m = importlib.import_module(to_module_path(c))
curr_output_path = "harbor.cfg.%s.tmp" % c
print("migrating to version %s" % c)
m.migrate(curr_input_path, curr_output_path)
curr_input_path = curr_output_path
shutil.copy(curr_output_path, args.output_path)
print("Written new values to %s" % args.output_path)
for tmp_f in glob.glob("harbor.cfg.*.tmp"):
os.remove(tmp_f)
def to_module_path(ver):
return "migrator_%s" % ver.replace(".","_")
def search(basedir, input_ver, target_ver, l):
module = to_module_path(target_ver)
if os.path.isdir(os.path.join(basedir, module)):
m = importlib.import_module(module)
if input_ver in m.acceptable_versions:
l.append(target_ver)
return True
for v in m.acceptable_versions:
if search(basedir, input_ver, v, l):
l.append(target_ver)
return True
return False
if __name__ == "__main__":
main()