X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=state_tracker.py;h=e59231560e900305eaadab380746e5b671062242;hb=713a609bd19d491de03debf8a4a6ddf2540b13dc;hp=4836e3eb99915a6d25a06102ee0c93163cd69133;hpb=36fea7f15ed17150691b5b3ead75450e575229ef;p=python_utils.git diff --git a/state_tracker.py b/state_tracker.py index 4836e3e..e592315 100644 --- a/state_tracker.py +++ b/state_tracker.py @@ -1,10 +1,10 @@ #!/usr/bin/env python3 -from abc import ABC, abstractmethod import datetime import logging -import time import threading +import time +from abc import ABC, abstractmethod from typing import Dict, Optional import pytz @@ -73,21 +73,15 @@ class StateTracker(ABC): for update_id in sorted(self.last_reminder_ts.keys()): if force_all_updates_to_run: logger.debug('Forcing all updates to run') - self.update( - update_id, self.now, self.last_reminder_ts[update_id] - ) + self.update(update_id, self.now, self.last_reminder_ts[update_id]) self.last_reminder_ts[update_id] = self.now return refresh_secs = self.update_ids_to_update_secs[update_id] last_run = self.last_reminder_ts[update_id] if last_run is None: # Never run before - logger.debug( - f'id {update_id} has never been run; running it now' - ) - self.update( - update_id, self.now, self.last_reminder_ts[update_id] - ) + logger.debug(f'id {update_id} has never been run; running it now') + self.update(update_id, self.now, self.last_reminder_ts[update_id]) self.last_reminder_ts[update_id] = self.now else: delta = self.now - last_run @@ -148,9 +142,7 @@ class AutomaticStateTracker(StateTracker): This may block for as long as self.sleep_delay. """ - logger.debug( - 'Setting shutdown event and waiting for background thread.' - ) + logger.debug('Setting shutdown event and waiting for background thread.') self.should_terminate.set() self.updater_thread.join() logger.debug('Background thread terminated.') @@ -185,9 +177,7 @@ class WaitableAutomaticStateTracker(AutomaticStateTracker): 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()