Przeglądaj źródła

Revert "Revert "Refactor async_setup_entry.""

This reverts commit 8a086fd2fbf90aa99023ede9a5e25539644a3461.
Jason Rumney 3 lat temu
rodzic
commit
55ca20f98a
1 zmienionych plików z 4 dodań i 5 usunięć
  1. 4 5
      custom_components/tuya_local/__init__.py

+ 4 - 5
custom_components/tuya_local/__init__.py

@@ -151,14 +151,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
         _LOGGER.error(f"Configuration file for {config[CONF_TYPE]} not found.")
         return False
 
-    entities = {}
+    entities = set()
     e = device_conf.primary_entity
-    entities[e.entity] = True
+    entities.add(e.entity)
     for e in device_conf.secondary_entities():
-        entities[e.entity] = True
+        entities.add(e.entity)
 
-    for e in entities:
-        hass.async_create_task(hass.config_entries.async_forward_entry_setup(entry, e))
+    await hass.config_entries.async_forward_entry_setups(entry, entities)
 
     entry.add_update_listener(async_update_entry)