Bopytex/bopytex/scheduler.py

""" Scheduler for action to make """


class Scheduler:
    def __init__(self, actions: list):
        self.actions = actions
        self._tasks = []
        self._done = []

    @property
    def tasks(self):
        return self._tasks

    @property
    def done(self):
        return self._done

    def append(self, tasks):
        self._tasks += tasks

    def dispatch(self, task):
        """ Do a task """
        ans = self.actions[task.action](task.deps, task.args)
        return ans

    def __next__(self):
        task = self._tasks.pop(0)
        ans = self.dispatch(task)
        self._done.append(ans)
        return ans