From be6399410efd5911992fb8dab793b480ad16fbc5 Mon Sep 17 00:00:00 2001 From: Pascal Reeb Date: Mon, 5 Feb 2024 17:09:18 +0100 Subject: [PATCH] Use a single call to add entities in Nuki (#109696) * Nuki: use a single call to add entities * Clean up list addition --------- Co-authored-by: Martin Hjelmare --- homeassistant/components/nuki/binary_sensor.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/nuki/binary_sensor.py b/homeassistant/components/nuki/binary_sensor.py index e3b2d129017..f1da14bdd35 100644 --- a/homeassistant/components/nuki/binary_sensor.py +++ b/homeassistant/components/nuki/binary_sensor.py @@ -22,19 +22,20 @@ async def async_setup_entry( """Set up the Nuki binary sensors.""" entry_data: NukiEntryData = hass.data[NUKI_DOMAIN][entry.entry_id] - lock_entities = [] - opener_entities = [] + entities: list[NukiEntity] = [] for lock in entry_data.locks: if lock.is_door_sensor_activated: - lock_entities.extend([NukiDoorsensorEntity(entry_data.coordinator, lock)]) + entities.append(NukiDoorsensorEntity(entry_data.coordinator, lock)) - async_add_entities(lock_entities) + entities.extend( + [ + NukiRingactionEntity(entry_data.coordinator, opener) + for opener in entry_data.openers + ] + ) - for opener in entry_data.openers: - opener_entities.extend([NukiRingactionEntity(entry_data.coordinator, opener)]) - - async_add_entities(opener_entities) + async_add_entities(entities) class NukiDoorsensorEntity(NukiEntity[NukiDevice], BinarySensorEntity):