fix: add additional catches for login errors

Closes #461
alandtse · Dec 20, 2019 · 89a057b · 89a057b
1 parent a8bdbf4
commit 89a057b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/custom_components/alexa_media/media_player.py b/custom_components/alexa_media/media_player.py
@@ -529,6 +529,7 @@ def state(self):
             return STATE_IDLE
         return STATE_STANDBY
 
+    @_catch_login_errors
     async def async_update(self):
         """Get the latest details on a media player.
 

diff --git a/custom_components/alexa_media/switch.py b/custom_components/alexa_media/switch.py
@@ -224,6 +224,7 @@ def should_poll(self):
         return not (self.hass.data[DATA_ALEXAMEDIA]
                     ['accounts'][self._account]['websocket'])
 
+    @_catch_login_errors
     async def async_update(self):
         """Update state."""
         try: