Simplify device retrieval

RogerSelwyn · Dec 24, 2022 · bd23260 · bd23260
1 parent 625306f
commit bd23260
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/custom_components/mqtt_discoverystream/__init__.py b/custom_components/mqtt_discoverystream/__init__.py
@@ -271,12 +271,10 @@ async def _state_publisher(
                 publish_config = True
 
             if publish_config:
-                for entry in ent_reg.entities.values():
-                    if entry.entity_id != entity_id:
-                        continue
-                    for device in dev_reg.devices.values():
-                        if device.id != entry.device_id:
-                            continue
+                entry = ent_reg.async_get(entity_id)
+                if entry and entry.device_id:
+                    device = dev_reg.async_get(entry.device_id)
+                    if device:
                         config["dev"] = {}
                         if device.manufacturer:
                             config["dev"]["mf"] = device.manufacturer