mirror of
https://github.com/goharbor/harbor.git
synced 2024-11-27 12:46:03 +01:00
76 lines
3.8 KiB
Python
76 lines
3.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import time
|
||
|
|
import base
|
||
|
|
import v2_swagger_client
|
||
|
|
from v2_swagger_client.rest import ApiException
|
||
|
|
|
||
|
|
class Preheat(base.Base, object):
|
||
|
|
def __init__(self):
|
||
|
|
super(Preheat,self).__init__(api_type = "preheat")
|
||
|
|
|
||
|
|
def create_instance(self, name = None, description="It's a dragonfly instance", vendor="dragonfly",
|
||
|
|
endpoint_url="http://20.32.244.16", auth_mode="NONE", enabled=True, insecure=True,
|
||
|
|
expect_status_code = 201, expect_response_body = None, **kwargs):
|
||
|
|
if name is None:
|
||
|
|
name = base._random_name("instance")
|
||
|
|
client = self._get_client(**kwargs)
|
||
|
|
instance = v2_swagger_client.Instance(name=name, description=description,vendor=vendor,
|
||
|
|
endpoint=endpoint_url, auth_mode=auth_mode, enabled=enabled)
|
||
|
|
print("instance:",instance)
|
||
|
|
try:
|
||
|
|
_, status_code, header = client.create_instance_with_http_info(instance)
|
||
|
|
except ApiException as e:
|
||
|
|
base._assert_status_code(expect_status_code, e.status)
|
||
|
|
if expect_response_body is not None:
|
||
|
|
base._assert_status_body(expect_response_body, e.body)
|
||
|
|
return
|
||
|
|
base._assert_status_code(expect_status_code, status_code)
|
||
|
|
base._assert_status_code(201, status_code)
|
||
|
|
return base._get_id_from_header(header), name
|
||
|
|
|
||
|
|
def create_policy(self, project_name, project_id, provider_id, name = None, description="It's a dragonfly policy",
|
||
|
|
filters=r'[{"type":"repository","value":"re*"},{"type":"tag","value":"v1.0*"}]', trigger=r'{"type":"manual","trigger_setting":{"cron":""}}', enabled=True,
|
||
|
|
expect_status_code = 201, expect_response_body = None, **kwargs):
|
||
|
|
if name is None:
|
||
|
|
name = base._random_name("policy")
|
||
|
|
client = self._get_client(**kwargs)
|
||
|
|
policy = v2_swagger_client.PreheatPolicy(name=name, project_id=project_id, provider_id=provider_id,
|
||
|
|
description=description,filters=filters,
|
||
|
|
trigger=trigger, enabled=enabled)
|
||
|
|
print("policy:",policy)
|
||
|
|
try:
|
||
|
|
data, status_code, header = client.create_policy_with_http_info(project_name, policy)
|
||
|
|
except ApiException as e:
|
||
|
|
base._assert_status_code(expect_status_code, e.status)
|
||
|
|
if expect_response_body is not None:
|
||
|
|
base._assert_status_body(expect_response_body, e.body)
|
||
|
|
return
|
||
|
|
base._assert_status_code(expect_status_code, status_code)
|
||
|
|
base._assert_status_code(201, status_code)
|
||
|
|
return base._get_id_from_header(header), name
|
||
|
|
|
||
|
|
def get_instance(self, instance_name, **kwargs):
|
||
|
|
client = self._get_client(**kwargs)
|
||
|
|
return client.get_instance(instance_name)
|
||
|
|
|
||
|
|
def get_policy(self, project_name, preheat_policy_name, **kwargs):
|
||
|
|
client = self._get_client(**kwargs)
|
||
|
|
return client.get_policy(project_name, preheat_policy_name)
|
||
|
|
|
||
|
|
def update_policy(self, project_name, preheat_policy_name, policy, **kwargs):
|
||
|
|
client = self._get_client(**kwargs)
|
||
|
|
return client.update_policy(project_name, preheat_policy_name, policy)
|
||
|
|
|
||
|
|
def delete_instance(self, preheat_instance_name, expect_status_code = 200, expect_response_body = None, **kwargs):
|
||
|
|
client = self._get_client(**kwargs)
|
||
|
|
try:
|
||
|
|
_, status_code, header = _, status_code, _ = client.delete_instance_with_http_info(preheat_instance_name)
|
||
|
|
except ApiException as e:
|
||
|
|
base._assert_status_code(expect_status_code, e.status)
|
||
|
|
if expect_response_body is not None:
|
||
|
|
base._assert_status_body(expect_response_body, e.body)
|
||
|
|
else:
|
||
|
|
base._assert_status_code(expect_status_code, status_code)
|
||
|
|
base._assert_status_code(200, status_code)
|