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

Migrate rfxtrx to use async_track_state_change_event (#115556)

Browse Source
This commit is contained in:
J. Nick Koston 2024-04-13 20:37:01 -05:00 committed by GitHub
parent 41f5325ce3
commit 0feea624f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
homeassistant/components/rfxtrx/config_flow.py
View File

@ -30,14 +30,14 @@ from homeassistant.const import (
CONF_PORT, CONF_PORT,
CONF_TYPE, CONF_TYPE,
) )
from homeassistant.core import State, callback from homeassistant.core import Event, EventStateChangedData, callback
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
device_registry as dr, device_registry as dr,
entity_registry as er, entity_registry as er,
) )
from homeassistant.helpers.event import async_track_state_change from homeassistant.helpers.event import async_track_state_change_event
from . import ( from . import (
DOMAIN, DOMAIN,
@ -353,10 +353,10 @@ class RfxtrxOptionsFlow(OptionsFlow):
entity_migration_map[new_entity_id] = entry entity_migration_map[new_entity_id] = entry
@callback @callback
def _handle_state_removed( def _handle_state_removed(event: Event[EventStateChangedData]) -> None:
entity_id: str, old_state: State | None, new_state: State | None
) -> None:
# Wait for entities to finish cleanup # Wait for entities to finish cleanup
new_state = event.data["new_state"]
entity_id = event.data["entity_id"]
if new_state is None and entity_id in entities_to_be_removed: if new_state is None and entity_id in entities_to_be_removed:
entities_to_be_removed.remove(entity_id) entities_to_be_removed.remove(entity_id)
if not entities_to_be_removed: if not entities_to_be_removed:
@ -370,7 +370,7 @@ class RfxtrxOptionsFlow(OptionsFlow):
if not self.hass.states.async_available(entry.entity_id) if not self.hass.states.async_available(entry.entity_id)
} }
wait_for_entities = asyncio.Event() wait_for_entities = asyncio.Event()
remove_track_state_changes = async_track_state_change( remove_track_state_changes = async_track_state_change_event(
self.hass, entities_to_be_removed, _handle_state_removed self.hass, entities_to_be_removed, _handle_state_removed
) )
@ -384,10 +384,10 @@ class RfxtrxOptionsFlow(OptionsFlow):
remove_track_state_changes() remove_track_state_changes()
@callback @callback
def _handle_state_added( def _handle_state_added(event: Event[EventStateChangedData]) -> None:
entity_id: str, old_state: State | None, new_state: State | None
) -> None:
# Wait for entities to be added # Wait for entities to be added
old_state = event.data["old_state"]
entity_id = event.data["entity_id"]
if old_state is None and entity_id in entities_to_be_added: if old_state is None and entity_id in entities_to_be_added:
entities_to_be_added.remove(entity_id) entities_to_be_added.remove(entity_id)
if not entities_to_be_added: if not entities_to_be_added:
@ -400,7 +400,7 @@ class RfxtrxOptionsFlow(OptionsFlow):
if self.hass.states.async_available(entry.entity_id) if self.hass.states.async_available(entry.entity_id)
} }
wait_for_entities = asyncio.Event() wait_for_entities = asyncio.Event()
remove_track_state_changes = async_track_state_change( remove_track_state_changes = async_track_state_change_event(
self.hass, entities_to_be_added, _handle_state_added self.hass, entities_to_be_added, _handle_state_added
) )