🚧 More work on managers

src/store/store.py

@@ -1,18 +1,39 @@
+from typing import Iterable
+
+from gi.repository import GObject
+
 import src.shared as shared
 from src.game import Game
 from src.store.manager import Manager
 from src.utils.task import Task
 
 
-class Pipeline(set):
-    """Class representing a set of Managers for a game"""
+class Pipeline(GObject.Object):
+    """Class representing a set of managers for a game"""
+
+    game: Game
+
+    waiting: set[Manager]
+    running: set[Manager]
+    done: set[Manager]
+
+    def __init__(self, managers: Iterable[Manager]) -> None:
+        super().__init__()
+        self.waiting = set(managers)
+        self.running = set()
+        self.done = set()
 
     @property
-    def blocked_managers(self) -> set(Manager):
+    def not_done(self) -> set[Manager]:
+        """Get the managers that are not done yet"""
+        return self.waiting + self.running
+
+    @property
+    def blocked(self) -> set[Manager]:
         """Get the managers that cannot run because their dependencies aren't done"""
         blocked = set()
-        for manager_a in self:
-            for manager_b in self:
+        for manager_a in self.waiting:
+            for manager_b in self.not_done:
                 if manager_a == manager_b:
                     continue
                 if type(manager_b) in manager_a.run_after:
@@ -20,9 +41,43 @@ class Pipeline(set):
         return blocked
 
     @property
-    def startable_managers(self) -> set(Manager):
+    def ready(self) -> set[Manager]:
         """Get the managers that can be run"""
-        return self - self.blocked_managers
+        return self.waiting - self.blocked
+
+    def advance(self):
+        """Spawn tasks for managers that are able to run for a game"""
+        for manager in self.ready:
+            self.waiting.remove(manager)
+            self.running.add(manager)
+            data = (manager,)
+            task = Task.new(self, manager.cancellable, self.manager_task_callback, data)
+            task.set_task_data(data)
+            task.run_in_thread(self.manager_task_thread_func)
+
+    @GObject.Signal(name="manager-started")
+    def manager_started(self, manager: Manager) -> None:
+        """Signal emitted when a manager is started"""
+        pass
+
+    def manager_task_thread_func(self, _task, _source_object, data, cancellable):
+        """Thread function for manager tasks"""
+        manager, *_rest = data
+        self.emit("manager-started", manager)
+        manager.run(self.game, cancellable)
+
+    @GObject.Signal(name="manager-done")
+    def manager_done(self, manager: Manager) -> None:
+        """Signal emitted when a manager is done"""
+        pass
+
+    def manager_task_callback(self, _source_object, _result, data):
+        """Callback function for manager tasks"""
+        manager, *_rest = data
+        self.running.remove(manager)
+        self.done.add(manager)
+        self.emit("manager-done", manager)
+        self.advance()
 
 
 class Store:
@@ -42,35 +97,20 @@ class Store:
         """Add a manager class that will run when games are added"""
         self.managers.add(manager)
 
-    def add_game(self, game: Game, replace=False):
+    def add_game(self, game: Game, replace=False) -> Pipeline:
         """Add a game to the app if not already there
 
-        :param replace bool: Replace the game if it already exists"""
+        :param replace bool: Replace the game if it already exists
+        :return:
+        """
         if (
             game.game_id in self.games
             and not self.games[game.game_id].removed
             and not replace
         ):
             return
+        pipeline = Pipeline(self.managers)
         self.games[game.game_id] = game
-        self.pipelines[game.game_id] = Pipeline(self.managers)
-        self.advance_pipeline(game)
-
-    def advance_pipeline(self, game: Game):
-        """Spawn tasks for managers that are able to run for a game"""
-        for manager in self.pipelines[game.game_id].startable_managers:
-            data = (manager, game)
-            task = Task.new(None, None, self.manager_task_callback, data)
-            task.set_task_data(data)
-            task.run_in_thread(self.manager_task_thread_func)
-
-    def manager_task_thread_func(self, _task, _source_object, data, _cancellable):
-        """Thread function for manager tasks"""
-        manager, game, *_rest = data
-        manager.run(game)
-
-    def manager_task_callback(self, _source_object, _result, data):
-        """Callback function for manager tasks"""
-        manager, game, *_rest = data
-        self.pipelines[game.game_id].remove(manager)
-        self.advance_pipeline(game)
+        self.pipelines[game.game_id] = pipeline
+        pipeline.advance()
+        return pipeline