fix: improve reauth

Author: Lash-L

python_snoo/pubnub_async.py

@@ -42,6 +42,10 @@ def __init__(self, pubnub: PubNubAsyncio, device_id: str) -> None:
 
     def update_token(self, token: str):
         self.pubnub.config.auth_key = token
+        if self.task:
+            self.task.cancel()  # cancel the task if it exists.
+            self.task = None
+        asyncio.create_task(self.run())  # restart the task
 
     def presence(self, pubnub: PubNubAsyncio, presence):
         _LOGGER.debug("Received new presence: %s", presence)

python_snoo/snoo.py

@@ -227,6 +227,7 @@ async def authorize(self) -> AuthorizationInfo:
         return self.tokens
 
     async def schedule_reauthorization(self, snoo_expiry: int):
+        _LOGGER.info("Snoo token has expired - reauthorizing...")
         try:
             await asyncio.sleep(snoo_expiry)
             await self.authorize()
@@ -235,7 +236,7 @@ async def schedule_reauthorization(self, snoo_expiry: int):
             self.pubnub.config.auth_token = self.tokens.snoo
 
         except Exception as ex:
-            print(f"Error during reauthorization: {ex}")
+            _LOGGER.exception(f"Error during reauthorization: {ex}")
 
     async def get_devices(self) -> list[SnooDevice]:
         hdrs = self.generate_snoo_auth_headers(self.tokens.aws_id)