- async_update_device

- add remove
AaronDavidSchneider · Apr 21, 2021 · 509d75e · 509d75e
1 parent 7c9fe78
commit 509d75e
Showing 1 changed file with 17 additions and 11 deletions.
diff --git a/custom_components/sonos_alarm/switch.py b/custom_components/sonos_alarm/switch.py
@@ -152,13 +152,29 @@ def _get_current_alarm_instance(self):
         else:
             return False
 
+    async def _async_remove(self):
+        """remove this entity as it is not available anymore"""
+        entity_registry = er.async_get(self.hass)
+        entity_registry.async_remove(self.entity_id)
+
+    async def _async_update_device(self):
+        """update the device, since this alarm moved to a different player"""
+        device_registry = dr.async_get(self.hass)
+        entity_registry = er.async_get(self.hass)
+        entry_id = entity_registry.async_get(self.entity_id).config_entry_id
+
+        new_device = device_registry.async_get_or_create(config_entry_id=entry_id, identifiers = {(SONOS_DOMAIN, self._unique_player_id)}, connections={(dr.CONNECTION_NETWORK_MAC, self._mac_address)})
+        if not entity_registry.async_get(self.entity_id).device_id == new_device.id:
+            entity_registry._async_update_entity(self.entity_id, device_id=new_device.id)
+
     async def async_update(self, now=None):
         """Retrieve latest state."""
         _LOGGER.debug("updating alarms")
         try:
             self._is_available = await self.hass.async_add_executor_job(self._get_current_alarm_instance)
 
             if not self._is_available:
+                self.hass.async_create_task(self._async_remove())
                 return
 
             self._is_on = self.alarm.enabled
@@ -170,7 +186,7 @@ async def async_update(self, now=None):
                 self._sw_version: str = speaker_info["software_version"]
                 self._mac_address: str = speaker_info["mac_address"]
                 self._unique_player_id: str = self.alarm.zone.uid
-                self._update_device()
+                await self._update_device()
 
 
             self._name = "Sonos Alarm {} {} {}".format(
@@ -201,16 +217,6 @@ async def async_update(self, now=None):
             )
             self._is_available = False
 
-    def _update_device(self):
-        device_registry = dr.async_get(self.hass)
-        entity_registry = er.async_get(self.hass)
-        entry_id = entity_registry.async_get(self.entity_id).config_entry_id
-
-        new_device = device_registry.async_get_or_create(config_entry_id=entry_id, identifiers = {(SONOS_DOMAIN, self._unique_player_id)}, connections={(dr.CONNECTION_NETWORK_MAC, self._mac_address)})
-        if not entity_registry.async_get(self.entity_id).device_id == new_device.id:
-            entity_registry._async_update_entity(self.entity_id, device_id=new_device.id)
-
-
     @property
     def _is_today(self):
         recurrance = self.alarm.recurrence