Don't create battery sensors for devices which do not report a batter…

…y state. Fixes #61
sebr · Apr 29, 2021 · 0287889 · 0287889
1 parent eb7a245
commit 0287889
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/custom_components/bhyve/sensor.py b/custom_components/bhyve/sensor.py
@@ -34,11 +34,13 @@ async def async_setup_platform(hass, config, async_add_entities, _discovery_info
     devices = await bhyve.devices
     for device in devices:
         if device.get("type") == DEVICE_SPRINKLER:
-            sensors.append(BHyveBatterySensor(hass, bhyve, device))
             sensors.append(BHyveStateSensor(hass, bhyve, device))
             for zone in device.get("zones"):
                 sensors.append(BHyveZoneHistorySensor(hass, bhyve, device, zone))
 
+            if device.get("battery", None) is not None:
+                sensors.append(BHyveBatterySensor(hass, bhyve, device))
+
     async_add_entities(sensors, True)