diff options
Diffstat (limited to 'distbuild/mainloop.py')
-rw-r--r-- | distbuild/mainloop.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/distbuild/mainloop.py b/distbuild/mainloop.py index 97e439f3..f15a3ac1 100644 --- a/distbuild/mainloop.py +++ b/distbuild/mainloop.py @@ -57,8 +57,11 @@ class MainLoop(object): logging.debug('MainLoop.remove_state_machine: %s' % machine) self._machines.remove(machine) + def state_machines_of_type(self, machine_type): + return [m for m in self._machines if isinstance(m, machine_type)] + def n_state_machines_of_type(self, machine_type): - return len([m for m in self._machines if isinstance(m, machine_type)]) + return len(self.state_machines_of_type(machine_type)) def add_event_source(self, event_source): logging.debug('MainLoop.add_event_source: %s' % event_source) |