fix: avoid pruning devices in secondary accounts

closes #1300
alandtse · May 18, 2021 · a62a371 · a62a371
1 parent 10c9fb5
commit a62a371
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/custom_components/alexa_media/__init__.py b/custom_components/alexa_media/__init__.py
@@ -621,16 +621,20 @@ async def async_update_data() -> Optional[AlexaEntityData]:
             device_registry, config_entry.entry_id
         ):
             for (_, identifier) in device_entry.identifiers:
-                if (
-                    identifier
-                    in hass.data[DATA_ALEXAMEDIA]["accounts"][email]["devices"][
+                if identifier in hass.data[DATA_ALEXAMEDIA]["accounts"][email][
+                    "devices"
+                ]["media_player"].keys() or identifier in map(
+                    lambda x: slugify(f"{x}_{email}"),
+                    hass.data[DATA_ALEXAMEDIA]["accounts"][email]["devices"][
                         "media_player"
-                    ].keys()
+                    ].keys(),
                 ):
                     break
             else:
                 device_registry.async_remove_device(device_entry.id)
-                _LOGGER.debug("Removing stale device %s", device_entry.name)
+                _LOGGER.debug(
+                    "%s: Removing stale device %s", hide_email(email), device_entry.name
+                )
 
         await login_obj.save_cookiefile()
         if login_obj.access_token: