diff options
Diffstat (limited to 'state_tracker.py')
| -rw-r--r-- | state_tracker.py | 25 |
1 files changed, 13 insertions, 12 deletions
diff --git a/state_tracker.py b/state_tracker.py index 9ce61e3..4836e3e 100644 --- a/state_tracker.py +++ b/state_tracker.py @@ -22,6 +22,7 @@ class StateTracker(ABC): provided to the c'tor. """ + def __init__(self, update_ids_to_update_secs: Dict[str, float]) -> None: """The update_ids_to_update_secs dict parameter describes one or more update types (unique update_ids) and the periodicity(ies), in @@ -106,6 +107,7 @@ class AutomaticStateTracker(StateTracker): terminate the updates. """ + @background_thread def pace_maker(self, should_terminate) -> None: """Entry point for a background thread to own calling heartbeat() @@ -128,6 +130,7 @@ class AutomaticStateTracker(StateTracker): override_sleep_delay: Optional[float] = None, ) -> None: import math_utils + super().__init__(update_ids_to_update_secs) if override_sleep_delay is not None: logger.debug(f'Overriding sleep delay to {override_sleep_delay}') @@ -174,15 +177,17 @@ class WaitableAutomaticStateTracker(AutomaticStateTracker): # else before looping up into wait again. """ + def __init__( - self, - update_ids_to_update_secs: Dict[str, float], - *, - override_sleep_delay: Optional[float] = None, + self, + update_ids_to_update_secs: Dict[str, float], + *, + override_sleep_delay: Optional[float] = None, ) -> None: self._something_changed = threading.Event() - super().__init__(update_ids_to_update_secs, - override_sleep_delay=override_sleep_delay) + super().__init__( + update_ids_to_update_secs, override_sleep_delay=override_sleep_delay + ) def something_changed(self): self._something_changed.set() @@ -193,9 +198,5 @@ class WaitableAutomaticStateTracker(AutomaticStateTracker): def reset(self): self._something_changed.clear() - def wait(self, - *, - timeout=None): - return self._something_changed.wait( - timeout=timeout - ) + def wait(self, *, timeout=None): + return self._something_changed.wait(timeout=timeout) |