X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=waitable_presence.py;h=e1da4310b76a77693aa505452e0f3748c4fcc62d;hb=90dc74b2ff8f36dd2495632a8150ce37dab61f1c;hp=9e0a9d0180663ad2337d7a6dc9ff42ccc677afb5;hpb=5f75cf834725ac26b289cc5f157af0cb71cd5f0e;p=python_utils.git diff --git a/waitable_presence.py b/waitable_presence.py index 9e0a9d0..e1da431 100644 --- a/waitable_presence.py +++ b/waitable_presence.py @@ -13,9 +13,9 @@ from typing import Optional, Tuple from overrides import overrides import base_presence -from type.locations import Location import site_config import state_tracker +from type.locations import Location logger = logging.getLogger(__name__) @@ -38,9 +38,9 @@ class WaitablePresenceDetectorWithMemory(state_tracker.WaitableAutomaticStateTra """ def __init__( - self, - override_update_interval_sec: float = 60.0, - override_location: Location = site_config.get_location(), + self, + override_update_interval_sec: float = 60.0, + override_location: Location = site_config.get_location(), ) -> None: self.last_someone_is_home: Optional[bool] = None self.someone_is_home: Optional[bool] = None @@ -96,6 +96,8 @@ class WaitablePresenceDetectorWithMemory(state_tracker.WaitableAutomaticStateTra if self.someone_is_home is None: raise Exception("Too Soon!") if self.someone_is_home: + assert self.someone_home_since is not None return (True, self.someone_home_since) else: + assert self.everyone_gone_since is not None return (False, self.everyone_gone_since)