mirror of
https://github.com/goharbor/harbor.git
synced 2024-12-24 09:38:09 +01:00
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
|
from __future__ import absolute_import
|
||
|
|
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import numpy
|
||
|
|
import threading
|
||
|
|
from datetime import *
|
||
|
|
from time import sleep, ctime
|
||
|
|
|
||
|
|
import library.repository
|
||
|
|
import library.docker_api
|
||
|
|
from library.base import _assert_status_code
|
||
|
|
from testutils import ADMIN_CLIENT
|
||
|
|
from testutils import harbor_server
|
||
|
|
|
||
|
|
from testutils import TEARDOWN
|
||
|
|
from library.project import Project
|
||
|
|
from library.repository import Repository
|
||
|
|
from library.artifact import Artifact
|
||
|
|
from library.repository import push_image_to_project
|
||
|
|
from library.repository import pull_harbor_image
|
||
|
|
from library.repository import push_special_image_to_project
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
def do_populate(name_index, repo_count):
|
||
|
|
project= Project()
|
||
|
|
artifact = Artifact()
|
||
|
|
repo = Repository()
|
||
|
|
url = ADMIN_CLIENT["endpoint"]
|
||
|
|
ADMIN_CLIENT["password"] = "qA5ZgV"
|
||
|
|
|
||
|
|
|
||
|
|
#2. Create a new project(PA) by user(UA);
|
||
|
|
project_name = "project"+str(name_index)
|
||
|
|
if project.check_project_name_exist(name=project_name, **ADMIN_CLIENT) is not True:
|
||
|
|
project.create_project(name=project_name, metadata = {"public": "false"}, **ADMIN_CLIENT)
|
||
|
|
print("Create Project:", project_name)
|
||
|
|
|
||
|
|
tag = 'latest'
|
||
|
|
for repo_index in range(int(repo_count)):
|
||
|
|
repo_name = "image"+ str(repo_index)
|
||
|
|
if artifact.check_reference_exist(project_name, repo_name, tag, ignore_not_found=True, **ADMIN_CLIENT) is not True:
|
||
|
|
push_special_image_to_project(project_name, harbor_server, ADMIN_CLIENT["username"], ADMIN_CLIENT["password"], repo_name, [tag], size=repo_index*30)
|
||
|
|
print("Push Image:", repo_name)
|
||
|
|
for tag_index in numpy.arange(1, 2, 0.1):
|
||
|
|
artifact.create_tag(project_name, repo_name, tag, str(tag_index), ignore_conflict = True, **ADMIN_CLIENT)
|
||
|
|
print("Add Tag:", str(tag_index))
|
||
|
|
|
||
|
|
|
||
|
|
def get_parser():
|
||
|
|
""" return a parser """
|
||
|
|
parser = argparse.ArgumentParser("populate")
|
||
|
|
|
||
|
|
parser.add_argument('--project-count','-p', dest='project_count', required=False, default=100)
|
||
|
|
parser.add_argument('--repo-count','-r', dest='repo_count', required=False, default=100)
|
||
|
|
args = parser.parse_args()
|
||
|
|
return (args.project_count, args.repo_count)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
""" main entrance """
|
||
|
|
project_count, repo_count = get_parser()
|
||
|
|
Threads = []
|
||
|
|
for i in range(int(project_count)):
|
||
|
|
t = threading.Thread(target=do_populate, args=(str(i), repo_count), name='T'+str(i))
|
||
|
|
t.setDaemon(True)
|
||
|
|
Threads.append(t)
|
||
|
|
sleep(3)
|
||
|
|
for t in Threads:
|
||
|
|
t.start()
|
||
|
|
for t in Threads:
|
||
|
|
t.join()
|
||
|
|
|
||
|
|
print('Job Finished:', ctime())
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|
||
|
|
|