Feat: scheduler handle dependencies (basic)

bopytex/scheduler.py

@@ -19,12 +19,19 @@ class Scheduler:
         self._tasks += tasks
 
     def dispatch(self, task):
-        """ Do a task """
+        """Do a task"""
         ans = self.actions[task.action](task.deps, task.args)
         return ans
 
     def __next__(self):
+        undoable = []
+
         task = self._tasks.pop(0)
+        while not all([d in self.done for d in task.deps]):
+            undoable.append(task)
+            task = self._tasks.pop(0)
+
+        self.append(undoable)
         ans = self.dispatch(task)
         self._done.append(ans)
         return ans

test/test_scheduler.py

@@ -59,3 +59,28 @@ def test_schedule_multiple_tasks():
     assert result == "three done"
     assert scheduler.tasks == []
     assert scheduler.done == ["one done", "two done", "three done"]
+
+
+def test_schedule_multiple_tasks_with_dependencies():
+    actions = {"DO": lambda deps, args: f"{args['task']} done"}
+    scheduler = Scheduler(actions)
+    t1 = Task(action="DO", args={"task": "one"}, deps=["three done"])
+    t2 = Task(action="DO", args={"task": "two"}, deps=["one done"])
+    t3 = Task(action="DO", args={"task": "three"}, deps=[])
+    scheduler.append([t1, t2, t3])
+
+    result = scheduler.__next__()
+    assert result == "three done"
+    assert scheduler.tasks == [t1, t2]
+    assert scheduler.done == ["three done"]
+
+    result = scheduler.__next__()
+    assert result == "one done"
+    assert scheduler.tasks == [t2]
+    assert scheduler.done == ["three done", "one done"]
+
+    result = scheduler.__next__()
+    assert result == "two done"
+    assert scheduler.tasks == []
+    assert scheduler.done == ["three done", "one done", "two done"]
+