class SchedulerTask(object):
    def __init__(self, slave_id, slave_ip):
        self.task = mesos_pb2.TaskInfo()
        self.slave_id = slave_id

    def add_resources_configurations(self, id):
        self.task = Task(id, "dscheduler", self.slave_id)
        self.task.configure_command_protobuf("dscheduler")

        self.task.configure_memory_resources()
        self.task.configure_container_protobuf(expose_ports=True)
        self.task.configure_port_resources()
        self.task.configure_cpu_resources()
class WorkerTask(object):
    def __init__(self, slave_id, slave_ip):
        self.task = mesos_pb2.TaskInfo()
        self.slave_id = slave_id
        self.slave_ip = slave_ip

    def add_resources_configurations(self, id):
        self.task = Task(id, "dwoker", self.slave_id)
        # TODO: add config for test environment vs production evs for worker uri
        self.task.configure_command_protobuf("dworker 192.168.99.100:33001".format(self.slave_ip))
        self.task.configure_container_protobuf()
        self.task.configure_cpu_resources()
        self.task.configure_memory_resources()