mirror of
https://github.com/goharbor/harbor.git
synced 2024-11-17 07:45:24 +01:00
c8265a8d53
Default target version is 1.5.0 This is mainly for VIC-appliance upgrade, and should be considered experimental for oss due to limited test. Tested with 1.2 and 1.3 harbor.cfg from VIC appliance.
128 lines
5.1 KiB
Plaintext
Executable File
128 lines
5.1 KiB
Plaintext
Executable File
// Copyright (c) 2017 VMware, Inc. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#!/usr/bin/python
|
|
|
|
"""
|
|
The script is to import the projects of source file into Admiral and save the results into mapprojectsfile.
|
|
It's only for VIC 1.2 migration.
|
|
"""
|
|
|
|
import json
|
|
from optparse import OptionParser
|
|
import os
|
|
import urllib2, ssl
|
|
import sys
|
|
import logging
|
|
import logging.config
|
|
|
|
logging.basicConfig(filename="import_project.log", level=logging.INFO)
|
|
logger = logging.getLogger()
|
|
|
|
class Parameters(object):
|
|
def __init__(self):
|
|
self.admiral_endpoint = ''
|
|
self.tokenfile = ''
|
|
self.projectsfile = ''
|
|
self.mapprojectsfile = ''
|
|
self.init_from_input()
|
|
|
|
@staticmethod
|
|
def parse_input():
|
|
usage = "usage: %prog [options] <admiralendpoint> <tokenfile> <projectsfile>"
|
|
parser = OptionParser(usage)
|
|
parser.add_option("-a", "--admiralendpoint", dest="admiral_endpoint", help="admiral endpoint")
|
|
parser.add_option("-t", "--tokenfile", dest="tokenfile", help="the path of token file")
|
|
parser.add_option("-f", "--projectsfile", dest="projectsfile", help="the path of exported json file")
|
|
parser.add_option("-m", "--mapprojectsfile", dest="mapprojectsfile", help="the path of output projects file for mapping project id")
|
|
|
|
(options, args) = parser.parse_args()
|
|
return (options.admiral_endpoint, options.tokenfile, options.projectsfile, options.mapprojectsfile)
|
|
|
|
def init_from_input(self):
|
|
(self.admiral_endpoint, self.tokenfile, self.projectsfile, self.mapprojectsfile) = Parameters.parse_input()
|
|
|
|
class Project:
|
|
def __init__(self, project_id, name, public):
|
|
self.project_id = project_id
|
|
self.project_name = name
|
|
self.public = public
|
|
self.index_id = ''
|
|
|
|
class Admiral:
|
|
def __init__(self, admiral_url, token):
|
|
self.admiral_url = admiral_url + '/projects'
|
|
self.token = token
|
|
|
|
def __import_project(self, project, retry=True):
|
|
project_data = json.dumps({ "name": project.project_name, "isPublic": project.public,
|
|
"customProperties": {"__enableContentTrust": False, "__preventVulnerableImagesFromRunning":False,
|
|
"__preventVulnerableImagesFromRunningSeverity":"high", "__automaticallyScanImagesOnPush":False }})
|
|
data_len = len(project_data)
|
|
request = urllib2.Request(self.admiral_url, project_data)
|
|
request.add_header('x-xenon-auth-token', self.token)
|
|
request.add_header('Content-Type', 'application/json')
|
|
request.add_header('Content-Length', data_len)
|
|
|
|
try:
|
|
response = urllib2.urlopen(request, context=ssl._create_unverified_context())
|
|
response_obj = response.read()
|
|
project.index_id = json.loads(response_obj)['customProperties']['__projectIndex']
|
|
except Exception, e:
|
|
if not retry:
|
|
logger.error("failed to import project: %s, admiral_endpoint: %s, error: %s " % (project.project_name, self.admiral_url, str(e)))
|
|
return
|
|
self.__import_project(project, False)
|
|
|
|
def import_project(self, projects):
|
|
for project in projects:
|
|
self.__import_project(project)
|
|
|
|
def main():
|
|
commandline_input = Parameters()
|
|
|
|
try:
|
|
if not os.path.exists(commandline_input.projectsfile):
|
|
raise Exception('Error: %s does not exist' % commandline_input.projectsfile)
|
|
|
|
if not os.path.exists(commandline_input.tokenfile):
|
|
raise Exception('Error: %s does not exist' % commandline_input.tokenfile)
|
|
|
|
with open(commandline_input.tokenfile, 'r') as f:
|
|
token = f.readlines()
|
|
|
|
if len(token) == 0:
|
|
raise Exception('No token found in the properties file %s' % commandline_input.tokenfile)
|
|
|
|
admiral = Admiral(commandline_input.admiral_endpoint, token[0])
|
|
|
|
with open(commandline_input.projectsfile, 'r') as project_data_file:
|
|
project_data = json.load(project_data_file)
|
|
|
|
projects_import_list = []
|
|
for item in project_data['projects']:
|
|
projects_import_list.append(Project(item['project_id'], item['project_name'], item['public']))
|
|
|
|
admiral.import_project(projects_import_list)
|
|
|
|
with open(commandline_input.mapprojectsfile, 'w') as outfile:
|
|
json.dump({'map_projects': [project.__dict__ for project in projects_import_list]}, outfile, sort_keys=True, indent=4)
|
|
|
|
except Exception, e:
|
|
logger.error("failed to import project, admiral_endpoint: %s, error: %s " % (commandline_input.admiral_endpoint, str(e)))
|
|
sys.exit(1)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|