From 509c1ca99cce30cac828e43938699074f1426163 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Mon, 3 Apr 2023 08:42:24 -0600 Subject: [PATCH] Fix missing battery sensors for SimpliSafe locks (#90722) --- homeassistant/components/simplisafe/binary_sensor.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/simplisafe/binary_sensor.py b/homeassistant/components/simplisafe/binary_sensor.py index 6849c9648d0..6b26eb7854b 100644 --- a/homeassistant/components/simplisafe/binary_sensor.py +++ b/homeassistant/components/simplisafe/binary_sensor.py @@ -1,7 +1,7 @@ """Support for SimpliSafe binary sensors.""" from __future__ import annotations -from simplipy.device import DeviceTypes +from simplipy.device import DeviceTypes, DeviceV3 from simplipy.device.sensor.v3 import SensorV3 from simplipy.system.v3 import SystemV3 @@ -67,6 +67,9 @@ async def async_setup_entry( if sensor.type in SUPPORTED_BATTERY_SENSOR_TYPES: sensors.append(BatteryBinarySensor(simplisafe, system, sensor)) + for lock in system.locks.values(): + sensors.append(BatteryBinarySensor(simplisafe, system, lock)) + async_add_entities(sensors) @@ -99,14 +102,14 @@ class BatteryBinarySensor(SimpliSafeEntity, BinarySensorEntity): _attr_entity_category = EntityCategory.DIAGNOSTIC def __init__( - self, simplisafe: SimpliSafe, system: SystemV3, sensor: SensorV3 + self, simplisafe: SimpliSafe, system: SystemV3, device: DeviceV3 ) -> None: """Initialize.""" - super().__init__(simplisafe, system, device=sensor) + super().__init__(simplisafe, system, device=device) self._attr_name = "Battery" self._attr_unique_id = f"{super().unique_id}-battery" - self._device: SensorV3 + self._device: DeviceV3 @callback def async_update_from_rest_api(self) -> None: