X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=waitable_presence.py;h=e1da4310b76a77693aa505452e0f3748c4fcc62d;hb=20dc43226d16e404514c3cdfba00e04469dbc17d;hp=d54511ff362bc45ceaf5c95e15174523f0327be9;hpb=e6f32fdd9b373dfcd100c7accb41f57d83c2f0a1;p=python_utils.git diff --git a/waitable_presence.py b/waitable_presence.py index d54511f..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__) @@ -47,9 +47,7 @@ class WaitablePresenceDetectorWithMemory(state_tracker.WaitableAutomaticStateTra self.everyone_gone_since: Optional[datetime.datetime] = None self.someone_home_since: Optional[datetime.datetime] = None self.location = override_location - self.detector: base_presence.PresenceDetection = ( - base_presence.PresenceDetection() - ) + self.detector: base_presence.PresenceDetection = base_presence.PresenceDetection() super().__init__( { 'poll_presence': override_update_interval_sec, @@ -98,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)