add flag [-conf] for specify harbor.cfg position in prepare

This commit is contained in:
yhua 2016-08-24 14:24:13 +08:00
parent bd2aa49fe0
commit 3f8a3889d5
1 changed files with 6 additions and 1 deletions

View File

@ -6,6 +6,7 @@ import random
import string
import os
import sys
import argparse
from io import open
if sys.version_info[:3][0] == 2:
@ -16,10 +17,14 @@ if sys.version_info[:3][0] == 3:
import configparser as ConfigParser
import io as StringIO
parser = argparse.ArgumentParser()
parser.add_argument('-conf', dest='cfgfile', default = 'harbor.cfg',type=str,help="the path of Harbor configuration file")
args = parser.parse_args()
#Read configurations
conf = StringIO.StringIO()
conf.write("[configuration]\n")
conf.write(open("harbor.cfg").read())
conf.write(open(args.cfgfile).read())
conf.seek(0, os.SEEK_SET)
rcp = ConfigParser.RawConfigParser()
rcp.readfp(conf)