from ..const import CREATE_URI

from ..common.getRequest import Request


class CronRequest(object):
    def __init__(self, token):
        self.stress_api = CREATE_URI
        self.headers = {'accesstoken': token, 'Accept': '*/*', 'content-type': 'application/json;charset=UTF-8'}

    def create(self, params):
        url = self.stress_api + '/back-end/stress/schedule/save'
        ret = Request.go('post', url, params, self.headers)
        if not ret:
            return
        return ret.get('id')

    def pause(self, jid):
        url = self.stress_api + '/back-end/stress/schedule/pause'
        params = [jid]
        Request.go('post', url, params, self.headers)

    def resume(self, jid):
        url = self.stress_api + '/back-end/stress/schedule/resume'
        params = [jid]
        Request.go('post', url, params, self.headers)

    def remove(self, jid):
        url = self.stress_api + '/back-end/stress/schedule/delete'
        params = [jid]
        Request.go('post', url, params, self.headers)

    def update(self, req_params):
        url = self.stress_api + '/back-end/stress/schedule/update'
        Request.go('post', url, req_params, self.headers)