sdk_plan.py

'''Utilities relating to interaction with service plans'''

import dcos
import sdk_spin
import shakedown


def get_deployment_plan(service_name):
    return get_plan(service_name, "deploy")


def get_sidecar_plan(service_name):
    return get_plan(service_name, "sidecar")


def start_sidecar_plan(service_name, parameters=None):
    start_plan(service_name, "sidecar", parameters)


def start_plan(service_name, plan, parameters=None):
    return dcos.http.post("{}/v1/plans/{}/start".format(shakedown.dcos_service_url(service_name), plan),
                          json=parameters if parameters is not None else {})


def get_plan(service_name, plan):
    def fn():
        response = dcos.http.get("{}/v1/plans/{}".format(shakedown.dcos_service_url(service_name), plan))
        response.raise_for_status()
        return response

    return sdk_spin.time_wait_return(lambda: fn())