Просмотр исходного кода

Device: don't reuse cached connections for test

When testing a connection after a config flow, ensure that new connections
are used, and do not cache the test connection.

Issue #3058
Jason Rumney 11 месяцев назад
Родитель
Сommit
daf220aca1
1 измененных файлов с 6 добавлено и 4 удалено
  1. 6 4
      custom_components/tuya_local/device.py

+ 6 - 4
custom_components/tuya_local/device.py

@@ -76,22 +76,24 @@ class TuyaLocalDevice(object):
         self.dev_cid = dev_cid
         try:
             if dev_cid:
-                if hass.data[DOMAIN].get(dev_id):
+                if hass.data[DOMAIN].get(dev_id) and name != "Test":
                     parent = hass.data[DOMAIN][dev_id]["tuyadevice"]
                 else:
                     parent = tinytuya.Device(dev_id, address, local_key)
-                    hass.data[DOMAIN][dev_id] = {"tuyadevice": parent}
+                    if name != "Test":
+                        hass.data[DOMAIN][dev_id] = {"tuyadevice": parent}
                 self._api = tinytuya.Device(
                     dev_cid,
                     cid=dev_cid,
                     parent=parent,
                 )
             else:
-                if hass.data[DOMAIN].get(dev_id):
+                if hass.data[DOMAIN].get(dev_id) and name != "Test":
                     self._api = hass.data[DOMAIN][dev_id]["tuyadevice"]
                 else:
                     self._api = tinytuya.Device(dev_id, address, local_key)
-                    hass.data[DOMAIN][dev_id] = {"tuyadevice": self._api}
+                    if name != "Test":
+                        hass.data[DOMAIN][dev_id] = {"tuyadevice": self._api}
         except Exception as e:
             _LOGGER.error(
                 "%s: %s while initialising device %s",