fix(helpers): fix key error during login exception

Closes #423
alandtse · Oct 11, 2019 · ebda699 · ebda699
1 parent cf44ab8
commit ebda699
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/custom_components/alexa_media/helpers.py b/custom_components/alexa_media/helpers.py
@@ -154,9 +154,11 @@ async def wrapper(*args, **kwargs) -> Any:
                 login = instance._login
                 email = login.email
                 hass = instance.hass if instance.hass else None
-                if (hass and not
-                    (hass.data[DATA_ALEXAMEDIA]['accounts'][email]
-                     ['configurator'])):
+                if (hass and (
+                    'configurator' not in (hass.data[DATA_ALEXAMEDIA]
+                                           ['accounts'][email])
+                    or not (hass.data[DATA_ALEXAMEDIA]['accounts'][email]
+                        ['configurator']))):
                     config_entry = (
                         hass.data[DATA_ALEXAMEDIA]
                         ['accounts']