mirror of
https://github.com/home-assistant/core.git
synced 2025-11-09 02:49:40 +00:00
Dynamically add Airzone entities (#121891)
* airzone: reload entry on new devices
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* Revert "airzone: reload entry on new devices"
This reverts commit 3ecc0844e4.
* airzone: sensor: dynamically add new entities
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* tests: restore reverted airzone tests
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: sensor: code fixes
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: water_heater: dynamically add entities
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: binary_sensor: dynamically add entities
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: select: dynamically add entities
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: climate: dynamically add entities
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* tests: airzone: use freezer
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: call async_add_entities once
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: wrap async_add_listener on async_on_unload
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
* airzone: reduce number of entity listeners
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
---------
Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
This commit is contained in:
committed by
GitHub
parent
3111951757
commit
c044417837
@@ -82,33 +82,54 @@ async def async_setup_entry(
|
||||
"""Add Airzone binary sensors from a config_entry."""
|
||||
coordinator = entry.runtime_data
|
||||
|
||||
binary_sensors: list[AirzoneBinarySensor] = [
|
||||
AirzoneSystemBinarySensor(
|
||||
coordinator,
|
||||
description,
|
||||
entry,
|
||||
system_id,
|
||||
system_data,
|
||||
)
|
||||
for system_id, system_data in coordinator.data[AZD_SYSTEMS].items()
|
||||
for description in SYSTEM_BINARY_SENSOR_TYPES
|
||||
if description.key in system_data
|
||||
]
|
||||
added_systems: set[str] = set()
|
||||
added_zones: set[str] = set()
|
||||
|
||||
binary_sensors.extend(
|
||||
AirzoneZoneBinarySensor(
|
||||
coordinator,
|
||||
description,
|
||||
entry,
|
||||
system_zone_id,
|
||||
zone_data,
|
||||
)
|
||||
for system_zone_id, zone_data in coordinator.data[AZD_ZONES].items()
|
||||
for description in ZONE_BINARY_SENSOR_TYPES
|
||||
if description.key in zone_data
|
||||
)
|
||||
def _async_entity_listener() -> None:
|
||||
"""Handle additions of binary sensors."""
|
||||
|
||||
async_add_entities(binary_sensors)
|
||||
entities: list[AirzoneBinarySensor] = []
|
||||
|
||||
systems_data = coordinator.data.get(AZD_SYSTEMS, {})
|
||||
received_systems = set(systems_data)
|
||||
new_systems = received_systems - added_systems
|
||||
if new_systems:
|
||||
entities.extend(
|
||||
AirzoneSystemBinarySensor(
|
||||
coordinator,
|
||||
description,
|
||||
entry,
|
||||
system_zone_id,
|
||||
systems_data.get(system_zone_id),
|
||||
)
|
||||
for system_zone_id in new_systems
|
||||
for description in SYSTEM_BINARY_SENSOR_TYPES
|
||||
if description.key in systems_data.get(system_zone_id)
|
||||
)
|
||||
added_systems.update(new_systems)
|
||||
|
||||
zones_data = coordinator.data.get(AZD_ZONES, {})
|
||||
received_zones = set(zones_data)
|
||||
new_zones = received_zones - added_zones
|
||||
if new_zones:
|
||||
entities.extend(
|
||||
AirzoneZoneBinarySensor(
|
||||
coordinator,
|
||||
description,
|
||||
entry,
|
||||
system_zone_id,
|
||||
zones_data.get(system_zone_id),
|
||||
)
|
||||
for system_zone_id in new_zones
|
||||
for description in ZONE_BINARY_SENSOR_TYPES
|
||||
if description.key in zones_data.get(system_zone_id)
|
||||
)
|
||||
added_zones.update(new_zones)
|
||||
|
||||
async_add_entities(entities)
|
||||
|
||||
entry.async_on_unload(coordinator.async_add_listener(_async_entity_listener))
|
||||
_async_entity_listener()
|
||||
|
||||
|
||||
class AirzoneBinarySensor(AirzoneEntity, BinarySensorEntity):
|
||||
|
||||
Reference in New Issue
Block a user