fix: save cookies_txt into configuration

alandtse · Sep 29, 2020 · c74824e · c74824e
1 parent 56e020f
commit c74824e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/custom_components/alexa_media/__init__.py b/custom_components/alexa_media/__init__.py
@@ -44,6 +44,7 @@
 from .const import (
     ALEXA_COMPONENTS,
     CONF_ACCOUNTS,
+    CONF_COOKIES_TXT,
     CONF_DEBUG,
     CONF_EXCLUDE_DEVICES,
     CONF_INCLUDE_DEVICES,
@@ -1012,6 +1013,7 @@ async def test_login_status(hass, config_entry, login) -> bool:
             CONF_SCAN_INTERVAL: account[CONF_SCAN_INTERVAL].total_seconds()
             if isinstance(account[CONF_SCAN_INTERVAL], timedelta)
             else account[CONF_SCAN_INTERVAL],
+            CONF_COOKIES_TXT: account.get(CONF_COOKIES_TXT, ""),
         },
     )
     hass.components.persistent_notification.async_create(

diff --git a/custom_components/alexa_media/config_flow.py b/custom_components/alexa_media/config_flow.py
@@ -279,7 +279,6 @@ async def _test_login(self):
         _LOGGER.debug("Testing login status: %s", login.status)
         if login.status and login.status.get("login_successful"):
             existing_entry = await self.async_set_unique_id(f"{email} - {login.url}")
-            self.config.pop(CONF_COOKIES_TXT)
             if existing_entry:
                 self.hass.config_entries.async_update_entry(
                     existing_entry, data=self.config
@@ -541,6 +540,9 @@ def _update_schema_defaults(self) -> Any:
                 vol.Optional(
                     CONF_SCAN_INTERVAL, default=self.config.get(CONF_SCAN_INTERVAL, 60)
                 ): int,
+                vol.Optional(
+                    CONF_COOKIES_TXT, default=self.config.get(CONF_COOKIES_TXT, "")
+                ): str,
             },
         )
         return new_schema