jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 12:17:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bring Withings activity sensor creation in line with the others (#102771)

Browse Source
This commit is contained in:
Joost Lekkerkerker 2023-10-25 12:14:03 +02:00 committed by GitHub
parent 45e4f71d1a
commit 4812d62ccf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
homeassistant/components/withings/sensor.py
View File

@ -706,26 +706,28 @@ async def async_setup_entry(
activity_coordinator = withings_data.activity_coordinator activity_coordinator = withings_data.activity_coordinator
activity_callback: Callable[[], None] | None = None
activity_entities_setup_before = ent_reg.async_get_entity_id( activity_entities_setup_before = ent_reg.async_get_entity_id(
Platform.SENSOR, DOMAIN, f"withings_{entry.unique_id}_activity_steps_today" Platform.SENSOR, DOMAIN, f"withings_{entry.unique_id}_activity_steps_today"
) )
def _async_add_activity_entities() -> None:
"""Add activity entities."""
if activity_coordinator.data is not None or activity_entities_setup_before:
async_add_entities(
WithingsActivitySensor(activity_coordinator, attribute)
for attribute in ACTIVITY_SENSORS
)
if activity_callback:
activity_callback()
if activity_coordinator.data is not None or activity_entities_setup_before: if activity_coordinator.data is not None or activity_entities_setup_before:
_async_add_activity_entities() entities.extend(
WithingsActivitySensor(activity_coordinator, attribute)
for attribute in ACTIVITY_SENSORS
)
else: else:
activity_callback = activity_coordinator.async_add_listener( remove_activity_listener: Callable[[], None]
def _async_add_activity_entities() -> None:
"""Add activity entities."""
if activity_coordinator.data is not None:
async_add_entities(
WithingsActivitySensor(activity_coordinator, attribute)
for attribute in ACTIVITY_SENSORS
)
remove_activity_listener()
remove_activity_listener = activity_coordinator.async_add_listener(
_async_add_activity_entities _async_add_activity_entities
) )