Don't crash if we failed to retrive config

ronengr · Apr 21, 2024 · d451802 · d451802
1 parent c0a21cc
commit d451802
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/custom_components/hass_nuki_bt/entity.py b/custom_components/hass_nuki_bt/entity.py
@@ -39,10 +39,10 @@ def __init__(self, coordinator: NukiDataUpdateCoordinator) -> None:
             model=coordinator.device.device_type,
             name=coordinator.device_name,
             hw_version=".".join(
-                str(x) for x in coordinator.device.config["hardware_revision"]
+                str(x) for x in coordinator.device.config.get("hardware_revision",[])
             ),
             sw_version=".".join(
-                str(x) for x in coordinator.device.config["firmware_version"]
+                str(x) for x in coordinator.device.config.get("firmware_version",[])
             ),
         )
 

diff --git a/custom_components/hass_nuki_bt/sensor.py b/custom_components/hass_nuki_bt/sensor.py
@@ -39,7 +39,7 @@ class NukiSensorEntityDescription(SensorEntityDescription):
         icon="mdi:lock",
         device_class=SensorDeviceClass.ENUM,
         entity_category=EntityCategory.DIAGNOSTIC,
-        info_function=lambda slf: slf.device.config[slf.sensor],
+        info_function=lambda slf: slf.device.config.get(slf.sensor),
     ),
     "rssi": NukiSensorEntityDescription(
         key="rssi",