Merge pull request #22 from xZetsubou/fix_devices_setup_connection

Fix devices setup connect [Sub Devices takes 1mins to connect]
xZetsubou · Sep 1, 2023 · e8ad6d4 · e8ad6d4
2 parents 4c514f7 + 6557fb7
commit e8ad6d4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/custom_components/localtuya/__init__.py b/custom_components/localtuya/__init__.py
@@ -143,10 +143,10 @@ def _device_discovered(device: TuyaDevice):
             # device = hass.data[DOMAIN][TUYA_DEVICES][device_id]
             # if not device.connected:
             #     hass.create_task(device.async_connect())
-        elif device_id in hass.data[DOMAIN][TUYA_DEVICES]:
-            device = hass.data[DOMAIN][TUYA_DEVICES][device_id]
-            if not device.connected:
-                hass.create_task(device.async_connect())
+        # elif device_id in hass.data[DOMAIN][TUYA_DEVICES]:
+        #     device = hass.data[DOMAIN][TUYA_DEVICES][device_id]
+        #     if not device.connected:
+        #         hass.create_task(device.async_connect())
 
     def _shutdown(event):
         """Clean up resources when shutting down."""
@@ -285,6 +285,8 @@ async def setup_entities(device_ids):
 
         await async_remove_orphan_entities(hass, entry)
         await hass.config_entries.async_forward_entry_setups(entry, platforms)
+        for dev_id in device_ids:
+            hass.create_task(hass.data[DOMAIN][TUYA_DEVICES][dev_id].async_connect())
 
     await setup_entities(entry.data[CONF_DEVICES].keys())
     unsub_listener = entry.add_update_listener(update_listener)