#50 remove unregistered devices

KiraPC · Jul 18, 2024 · 7ad3be8 · 7ad3be8
1 parent c7d21ac
commit 7ad3be8
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/custom_components/switchbotremote/__init__.py b/custom_components/switchbotremote/__init__.py
@@ -8,6 +8,9 @@
 from .client import SwitchBot
 
 from .const import DOMAIN
+from homeassistant.helpers import (
+    device_registry as dr,
+)
 
 PLATFORMS: list[Platform] = [
     Platform.CLIMATE,
@@ -37,6 +40,16 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
+    device_registry = dr.async_get(hass)
+    for device_entry in dr.async_entries_for_config_entry(
+        device_registry, entry.entry_id
+    ):
+        device_id = list(device_entry.identifiers)[0][1]
+        registered = next((remote for remote in remotes if remote.id == device_id), None)
+
+        if not registered:
+            device_registry.async_remove_device(device_entry.id)
+
     return True