From 52a8f4df82d2a5fb0a0333c15d3a7b08346394df Mon Sep 17 00:00:00 2001 From: On Freund Date: Sun, 13 Nov 2022 19:53:54 +0200 Subject: [PATCH] Refactor binary sensor creation in Risco integration (#82043) --- homeassistant/components/risco/binary_sensor.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/homeassistant/components/risco/binary_sensor.py b/homeassistant/components/risco/binary_sensor.py index 7c9733e2ab9..423137d88b6 100644 --- a/homeassistant/components/risco/binary_sensor.py +++ b/homeassistant/components/risco/binary_sensor.py @@ -28,16 +28,13 @@ async def async_setup_entry( if is_local(config_entry): local_data: LocalData = hass.data[DOMAIN][config_entry.entry_id] async_add_entities( - RiscoLocalBinarySensor(local_data.system.id, zone_id, zone) - for zone_id, zone in local_data.system.zones.items() - ) - async_add_entities( - RiscoLocalAlarmedBinarySensor(local_data.system.id, zone_id, zone) - for zone_id, zone in local_data.system.zones.items() - ) - async_add_entities( - RiscoLocalArmedBinarySensor(local_data.system.id, zone_id, zone) + entity for zone_id, zone in local_data.system.zones.items() + for entity in ( + RiscoLocalBinarySensor(local_data.system.id, zone_id, zone), + RiscoLocalAlarmedBinarySensor(local_data.system.id, zone_id, zone), + RiscoLocalArmedBinarySensor(local_data.system.id, zone_id, zone), + ) ) else: coordinator: RiscoDataUpdateCoordinator = hass.data[DOMAIN][