from monitorT.task import Task
import time
import os


class TaskCustom(Task):
    CONTROL = True

    def set_up(self):
        self.FLAG_PAUSED = False

    def cancel(self):
        pass

    def pause(self):
        self.FLAG_PAUSED = True

    def start(self):
        self.FLAG_PAUSED = False


def post(task, user=None, input_layer=None, output_layer=None, step=10, **kwargs):
    """

    Parameters
    ----------
    user: str
        User id
    graph: dict
        Dict containing graph architecture

    Returns
    -------


    """

    # os.system('ogrinfo /vsicurl/ftp://{}:{}@{}:{}{}'.format(layer['user'], layer['password'], layer['ip'], layer['port'], layer['source']))

    for i in range(step):
        while task.FLAG_PAUSED:
            continue
        task.set_progress(100*i/step)
        time.sleep(1)
    return 'ueeeee'