mirror of
https://github.com/goharbor/harbor.git
synced 2024-12-23 09:08:26 +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.
130 lines
3.9 KiB
Plaintext
Executable File
130 lines
3.9 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 export the existing projects of Harbor into a file.
|
|
It's only for VIC 1.2 migration.
|
|
"""
|
|
|
|
import json
|
|
import fileinput
|
|
from optparse import OptionParser
|
|
import os
|
|
import MySQLdb
|
|
import sys
|
|
|
|
class Parameters(object):
|
|
def __init__(self):
|
|
self.dbuser = ''
|
|
self.dbpwd = ''
|
|
self.exportpath = ''
|
|
self.init_from_input()
|
|
|
|
@staticmethod
|
|
def parse_input():
|
|
usage = "usage: %prog [options] <dbuser> <dbpwd> <exportpath>"
|
|
parser = OptionParser(usage)
|
|
parser.add_option("-u", "--dbuser", dest="dbuser", help="db user")
|
|
parser.add_option("-p", "--dbpwd", dest="dbpwd", help="db password")
|
|
parser.add_option("-o", "--exportpath", dest="exportpath", help="the path of exported json file")
|
|
|
|
(options, args) = parser.parse_args()
|
|
return (options.dbuser, options.dbpwd, options.exportpath)
|
|
|
|
def init_from_input(self):
|
|
(self.dbuser, self.dbpwd, self.exportpath) = Parameters.parse_input()
|
|
|
|
class Project:
|
|
def __init__(self, project_id, name, public):
|
|
self.project_id = project_id
|
|
self.project_name = name
|
|
if public == 0:
|
|
self.public = "false"
|
|
elif public == 1:
|
|
self.public = "true"
|
|
else:
|
|
self.public = "false"
|
|
|
|
class HarborUtil:
|
|
def __init__(self, dbuser, dbpwd):
|
|
self.serverName = 'localhost'
|
|
self.user = dbuser
|
|
self.password = dbpwd
|
|
self.port = '3306'
|
|
self.subDB = 'registry'
|
|
self.db = None
|
|
self.cursor = None
|
|
|
|
def connect(self):
|
|
try:
|
|
self.db = MySQLdb.connect(host=self.serverName, user=self.user,
|
|
passwd=self.password, db=self.subDB)
|
|
self.cursor = self.db.cursor()
|
|
except Exception, e:
|
|
raise Exception(e)
|
|
|
|
def close(self):
|
|
try:
|
|
self.cursor.close()
|
|
self.db.close()
|
|
except Exception, e:
|
|
print str(e)
|
|
|
|
def get_projects(self):
|
|
projects = []
|
|
try:
|
|
query = "SELECT project_id, name, public from registry.project where deleted=0"
|
|
self.cursor.execute(query)
|
|
self.cursor.fetchall()
|
|
for result in self.cursor:
|
|
projects.append(Project(int(result[0]), result[1], result[2]))
|
|
return projects
|
|
except Exception, e:
|
|
raise Exception(e)
|
|
|
|
def delfile(src):
|
|
if not os.path.exists(src):
|
|
return
|
|
try:
|
|
os.remove(src)
|
|
except Exception, e:
|
|
raise Exception("unable to delete file: %s, error: %s" % (src, str(e)))
|
|
|
|
def main():
|
|
commandline_input = Parameters()
|
|
harbor = HarborUtil(commandline_input.dbuser, commandline_input.dbpwd)
|
|
|
|
try:
|
|
harbor.connect()
|
|
projects = harbor.get_projects()
|
|
if len(projects) == 0:
|
|
return
|
|
|
|
harbor_projects_json = commandline_input.exportpath + '/harbor_projects.json'
|
|
delfile(harbor_projects_json)
|
|
|
|
with open(harbor_projects_json, 'w') as outfile:
|
|
json.dump({'projects': [project.__dict__ for project in projects]}, outfile, sort_keys=True, indent=4)
|
|
|
|
except Exception, e:
|
|
print e
|
|
sys.exit(1)
|
|
finally:
|
|
harbor.close()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|