added dict get() instead of assuming keys exist; now handles config_flow setup failures without crashing

2022-10-09 19:54:31 -07:00
parent 7b37f07fd0
commit d2fa4dc2a0
2 changed files with 11 additions and 8 deletions
--- a/custom_components/localtuya/common.py
+++ b/custom_components/localtuya/common.py
@@ -124,8 +124,10 @@ def async_config_entry_by_device_id(hass, device_id):
    """Look up config entry by device id."""
    current_entries = hass.config_entries.async_entries(DOMAIN)
    for entry in current_entries:
-        if device_id in entry.data[CONF_DEVICES]:
+        if device_id in entry.data.get(CONF_DEVICES, []):
            return entry
+        else:
+            _LOGGER.warning(f"Missing device configuration for device_id {device_id}")
    return None