mirror of https://github.com/goharbor/harbor.git
76 lines
3.7 KiB
Python
76 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import sys
|
|
import time
|
|
import base
|
|
import swagger_client
|
|
|
|
class Replication(base.Base):
|
|
def create_replication_policy(self, dest_registry=None, src_registry=None, name=None, description="",
|
|
dest_namespace = "", filters=None, trigger=swagger_client.ReplicationTrigger(type="manual",trigger_settings=swagger_client.TriggerSettings(cron="")),
|
|
deletion=False, override=True, enabled=True, expect_status_code = 201, **kwargs):
|
|
if name is None:
|
|
name = base._random_name("rule")
|
|
if filters is None:
|
|
filters = []
|
|
for policy_filter in filters:
|
|
policy_filter["value"] = int(policy_filter["value"])
|
|
client = self._get_client(**kwargs)
|
|
policy = swagger_client.ReplicationPolicy(name=name, description=description,dest_namespace=dest_namespace,
|
|
dest_registry=dest_registry, src_registry=src_registry,filters=filters,
|
|
trigger=trigger, deletion=deletion, override=override, enabled=enabled)
|
|
_, status_code, header = client.replication_policies_post_with_http_info(policy)
|
|
base._assert_status_code(expect_status_code, status_code)
|
|
return base._get_id_from_header(header), name
|
|
|
|
def get_replication_rule(self, param = None, rule_id = None, expect_status_code = 200, **kwargs):
|
|
client = self._get_client(**kwargs)
|
|
if rule_id is None:
|
|
if param is None:
|
|
param = dict()
|
|
data, status_code, _ = client.replication_policies_id_get_with_http_info(param)
|
|
else:
|
|
data, status_code, _ = client.replication_policies_id_get_with_http_info(rule_id)
|
|
base._assert_status_code(expect_status_code, status_code)
|
|
return data
|
|
|
|
def check_replication_rule_should_exist(self, check_rule_id, expect_rule_name, expect_trigger = None, **kwargs):
|
|
rule_data = self.get_replication_rule(rule_id = check_rule_id, **kwargs)
|
|
if str(rule_data.name) != str(expect_rule_name):
|
|
raise Exception(r"Check replication rule failed, expect <{}> actual <{}>.".format(expect_rule_name, str(rule_data.name)))
|
|
else:
|
|
print r"Check Replication rule passed, rule name <{}>.".format(str(rule_data.name))
|
|
#get_trigger = str(rule_data.trigger.kind)
|
|
#if expect_trigger is not None and get_trigger == str(expect_trigger):
|
|
# print r"Check Replication rule trigger passed, trigger name <{}>.".format(get_trigger)
|
|
#else:
|
|
# raise Exception(r"Check replication rule trigger failed, expect <{}> actual <{}>.".format(expect_trigger, get_trigger))
|
|
|
|
|
|
def start_replication(self, rule_id, **kwargs):
|
|
client = self._get_client(**kwargs)
|
|
return client.replications_post(swagger_client.Replication(int(rule_id)))
|
|
|
|
def list_replication_jobs(self, rule_id, **kwargs):
|
|
client = self._get_client(**kwargs)
|
|
return client.jobs_replication_get(int(rule_id))
|
|
|
|
def wait_until_jobs_finish(self, rule_id, retry=10, interval=5, **kwargs):
|
|
finished = True
|
|
for i in range(retry):
|
|
finished = True
|
|
jobs = self.list_replication_jobs(rule_id, **kwargs)
|
|
for job in jobs:
|
|
if job.status != "finished":
|
|
finished = False
|
|
break
|
|
if not finished:
|
|
time.sleep(interval)
|
|
if not finished:
|
|
raise Exception("The jobs not finished")
|
|
|
|
def delete_replication_rule(self, rule_id, expect_status_code = 200, **kwargs):
|
|
client = self._get_client(**kwargs)
|
|
_, status_code, _ = client.replication_policies_id_delete_with_http_info(rule_id)
|
|
base._assert_status_code(expect_status_code, status_code)
|