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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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