Use generator expression in totalconnect (#116020)

This commit is contained in:
Joost Lekkerkerker 2024-04-23 09:48:17 +02:00 committed by GitHub
parent e2b401397d
commit 8f56d170b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -38,22 +38,18 @@ async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None: ) -> None:
"""Set up TotalConnect alarm panels based on a config entry.""" """Set up TotalConnect alarm panels based on a config entry."""
alarms: list[TotalConnectAlarm] = []
coordinator: TotalConnectDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id] coordinator: TotalConnectDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
for location in coordinator.client.locations.values(): async_add_entities(
alarms.extend(
TotalConnectAlarm( TotalConnectAlarm(
coordinator, coordinator,
location, location,
partition_id, partition_id,
) )
for location in coordinator.client.locations.values()
for partition_id in location.partitions for partition_id in location.partitions
) )
async_add_entities(alarms)
# Set up services # Set up services
platform = entity_platform.async_get_current_platform() platform = entity_platform.async_get_current_platform()