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

39 lines
1.6 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
import base
from v2_swagger_client.rest import ApiException
class Schedule(base.Base):
def __init__(self):
super(Schedule, self).__init__(api_type="schedule")
def get_schedule_paused(self, job_type, expect_status_code=200, expect_response_body=None, **kwargs):
try:
return_data, status_code, _ = self._get_client(**kwargs).get_schedule_paused_with_http_info(job_type)
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)
return return_data
def list_schedules(self, page_size=50, page=1, expect_status_code=200, expect_response_body=None, **kwargs):
try:
schedules, status_code, _ = self._get_client(**kwargs).list_schedules_with_http_info(page_size=50, page=1)
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)
schedule_dict = {}
for schedule in schedules:
if schedule.vendor_id in [ None, -1]:
schedule_dict[schedule.vendor_type] = schedule
else:
schedule_dict["%s-%d" % (schedule.vendor_type, schedule.vendor_id)] = schedule
return schedule_dict