Feat: append, dispatch and __next__ (basic) scheduler
This commit is contained in:
29
bopytex/scheduler.py
Normal file
29
bopytex/scheduler.py
Normal file
@@ -0,0 +1,29 @@
|
||||
""" 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)
|
||||
return ans
|
||||
|
||||
def __next__(self):
|
||||
task = self._tasks.pop()
|
||||
self._done.append(task.output)
|
||||
return self.dispatch(task)
|
||||
Reference in New Issue
Block a user