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 <marhje52@gmail.com>
This commit is contained in:
Pascal Reeb
2024-02-05 17:09:18 +01:00
committed by GitHub
parent 6f28d79651
commit be6399410e

View File

@@ -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):