From 8f56d170b9f1b4f7c840dadbb3468d5718651089 Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Tue, 23 Apr 2024 09:48:17 +0200 Subject: [PATCH] Use generator expression in totalconnect (#116020) --- .../totalconnect/alarm_control_panel.py | 20 ++++++++----------- 1 file changed, 8 insertions(+), 12 deletions(-) diff --git a/homeassistant/components/totalconnect/alarm_control_panel.py b/homeassistant/components/totalconnect/alarm_control_panel.py index 9b2abedbf52..b0ad2f19069 100644 --- a/homeassistant/components/totalconnect/alarm_control_panel.py +++ b/homeassistant/components/totalconnect/alarm_control_panel.py @@ -38,21 +38,17 @@ async def async_setup_entry( hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback ) -> None: """Set up TotalConnect alarm panels based on a config entry.""" - alarms: list[TotalConnectAlarm] = [] - coordinator: TotalConnectDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id] - for location in coordinator.client.locations.values(): - alarms.extend( - TotalConnectAlarm( - coordinator, - location, - partition_id, - ) - for partition_id in location.partitions + async_add_entities( + TotalConnectAlarm( + coordinator, + location, + partition_id, ) - - async_add_entities(alarms) + for location in coordinator.client.locations.values() + for partition_id in location.partitions + ) # Set up services platform = entity_platform.async_get_current_platform()