harbor/tests/apitests/python/library/registry.py

24 lines
930 B
Python

# -*- coding: utf-8 -*-
import sys
import base
import swagger_client
class Registry(base.Base):
def create_registry(self, endpoint, name=None, username="",
password="", insecure=True, **kwargs):
if name is None:
name = base._random_name("registry")
client = self._get_client(**kwargs)
registry = swagger_client.RepTargetPost(name=name, endpoint=endpoint,
username=username, password=password, insecure=insecure)
_, _, header = client.targets_post_with_http_info(registry)
return base._get_id_from_header(header), name
def get_registry_id_by_endpoint(self, endpoint, **kwargs):
client = self._get_client(**kwargs)
registries = client.targets_get()
for registry in registries or []:
if registry.endpoint == endpoint:
return registry.id
raise Exception("registry %s not found" % endpoint)