mirror of
https://github.com/goharbor/harbor.git
synced 2024-12-27 19:17:47 +01:00
112 lines
4.2 KiB
Python
112 lines
4.2 KiB
Python
#!/usr/local/bin/python3
|
|
|
|
import subprocess
|
|
import signal
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import click
|
|
import requests
|
|
|
|
MIGRATE_CHART_SCRIPT = '/migrate_chart.sh'
|
|
HELM_CMD = '/linux-amd64/helm'
|
|
CA_UPDATE_CMD = 'update-ca-certificates'
|
|
CHART_URL_PATTERN = "https://{host}/api/v2.0/projects/{project}/repositories/{name}/artifacts/{version}"
|
|
CHART_SOURCE_DIR = Path('/chart_storage')
|
|
|
|
errs = []
|
|
|
|
def print_exist_errs():
|
|
if errs:
|
|
click.echo("There are {} errors exist".format(len(errs)), err=True)
|
|
for e in errs:
|
|
click.echo(e, err=True)
|
|
# Write the error to file
|
|
with open("/chart_storage/migration_errors.txt", "a") as f:
|
|
f.write(e + "\n")
|
|
|
|
def graceful_exit(signum, frame):
|
|
print_exist_errs()
|
|
sys.exit()
|
|
|
|
signal.signal(signal.SIGINT, graceful_exit)
|
|
signal.signal(signal.SIGTERM, graceful_exit)
|
|
|
|
class ChartV2:
|
|
|
|
def __init__(self, filepath:Path):
|
|
self.filepath = filepath
|
|
self.project = self.filepath.parts[-2]
|
|
parts = self.filepath.stem.split('-')
|
|
flag = False
|
|
try:
|
|
for i in range(len(parts)-1, -1, -1):
|
|
if parts[i][0].isnumeric() or ((parts[i][0]=='v' or parts[i][0]=='v') and parts[i][1].isnumeric()) :
|
|
self.name, self.version = '-'.join(parts[:i]), '-'.join(parts[i:])
|
|
flag = True
|
|
break
|
|
if not flag:
|
|
raise Exception('chart name: {} is illegal'.format('-'.join(parts)))
|
|
except Exception as e:
|
|
click.echo("Skipped chart: {} due to illegal chart name. Error: {}".format(filepath, e), err=True)
|
|
return
|
|
def __check_exist(self, hostname, username, password):
|
|
return requests.get(CHART_URL_PATTERN.format(
|
|
host=hostname,
|
|
project=self.project,
|
|
name=self.name,
|
|
version=self.version),
|
|
auth=requests.auth.HTTPBasicAuth(username, password))
|
|
|
|
def migrate(self, hostname, username, password):
|
|
res = self.__check_exist(hostname, username, password)
|
|
if res.status_code == 200:
|
|
raise Exception("Artifact already exist in harbor")
|
|
if res.status_code == 401:
|
|
raise Exception(res.reason)
|
|
|
|
oci_ref = "oci://{host}/{project}".format(
|
|
host=hostname,
|
|
project=self.project)
|
|
|
|
return subprocess.run([MIGRATE_CHART_SCRIPT, HELM_CMD, self.filepath, oci_ref],
|
|
text=True, stdout=subprocess.DEVNULL, stderr=subprocess.PIPE)
|
|
|
|
|
|
@click.command()
|
|
@click.option('--hostname', default='127.0.0.1', help='the password to login harbor')
|
|
@click.option('--username', default='admin', help='The username to login harbor')
|
|
@click.option('--password', default='Harbor12345', help='the password to login harbor')
|
|
def migrate(hostname, username, password):
|
|
"""
|
|
Migrate chart v2 to harbor oci registry
|
|
"""
|
|
if username != 'admin':
|
|
raise Exception('This operation only allowed for admin')
|
|
subprocess.run([CA_UPDATE_CMD])
|
|
subprocess.run([HELM_CMD, 'registry', 'login', hostname, '--username', username, '--password', password])
|
|
charts = [ChartV2(c) for p in CHART_SOURCE_DIR.iterdir() if p.is_dir() for c in p.iterdir() if c.is_file() and c.name.endswith(".tgz")]
|
|
with click.progressbar(charts, label="Migrating chart ...", length=len(charts),
|
|
item_show_func=lambda x: "{}/{}:{} total errors: {}".format(x.project, x.name, x.version, len(errs)) if x else '') as bar:
|
|
for chart in bar:
|
|
try:
|
|
result = chart.migrate(hostname, username, password)
|
|
if result.stderr:
|
|
errs.append("chart: {name}:{version} in {project} has err: {err}".format(
|
|
name=chart.name,
|
|
version=chart.version,
|
|
project=chart.project,
|
|
err=result.stderr
|
|
))
|
|
except Exception as e:
|
|
errs.append("chart: {name}:{version} in {project} has err: {err}".format(
|
|
name=chart.name,
|
|
version=chart.version,
|
|
project=chart.project,
|
|
err=e))
|
|
click.echo("Migration is Done.")
|
|
print_exist_errs()
|
|
|
|
if __name__ == '__main__':
|
|
migrate()
|