fix: fix key error if entry exists

alandtse · Nov 23, 2021 · 4f2a2b2 · 4f2a2b2
1 parent f6fb70b
commit 4f2a2b2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/custom_components/alexa_media/config_flow.py b/custom_components/alexa_media/config_flow.py
@@ -677,6 +677,12 @@ async def _test_login(self):
                 self.hass.components.persistent_notification.async_dismiss(
                     f"alexa_media_{slugify(email)}{slugify(login.url[7:])}"
                 )
+                if not self.hass.data[DATA_ALEXAMEDIA]["accounts"].get(
+                    self.config[CONF_EMAIL]
+                ):
+                    self.hass.data[DATA_ALEXAMEDIA]["accounts"][
+                        self.config[CONF_EMAIL]
+                    ] = {}
                 self.hass.data[DATA_ALEXAMEDIA]["accounts"][self.config[CONF_EMAIL]][
                     "login_obj"
                 ] = self.login