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

Migrate proximity to use async_track_state_change_event (#115002)

Browse Source
This commit is contained in:
J. Nick Koston 2024-04-06 00:00:30 -10:00 committed by GitHub
parent 34bf4dc962
commit b6d0c9d1c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/proximity/__init__.py
View File

@ -20,7 +20,7 @@ from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.event import ( from homeassistant.helpers.event import (
async_track_entity_registry_updated_event, async_track_entity_registry_updated_event,
async_track_state_change, async_track_state_change_event,
) )
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType from homeassistant.helpers.typing import ConfigType
@ -142,7 +142,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
coordinator = ProximityDataUpdateCoordinator(hass, entry.title, dict(entry.data)) coordinator = ProximityDataUpdateCoordinator(hass, entry.title, dict(entry.data))
entry.async_on_unload( entry.async_on_unload(
async_track_state_change( async_track_state_change_event(
hass, hass,
entry.data[CONF_TRACKED_ENTITIES], entry.data[CONF_TRACKED_ENTITIES],
coordinator.async_check_proximity_state_change, coordinator.async_check_proximity_state_change,

16
homeassistant/components/proximity/coordinator.py
View File

@ -15,7 +15,13 @@ from homeassistant.const import (
CONF_ZONE, CONF_ZONE,
UnitOfLength, UnitOfLength,
) )
from homeassistant.core import Event, HomeAssistant, State, callback from homeassistant.core import (
Event,
EventStateChangedData,
HomeAssistant,
State,
callback,
)
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType from homeassistant.helpers.typing import ConfigType
@ -100,10 +106,14 @@ class ProximityDataUpdateCoordinator(DataUpdateCoordinator[ProximityData]):
self.entity_mapping[tracked_entity_id].append(entity_id) self.entity_mapping[tracked_entity_id].append(entity_id)
async def async_check_proximity_state_change( async def async_check_proximity_state_change(
self, entity: str, old_state: State | None, new_state: State | None self,
event: Event[EventStateChangedData],
) -> None: ) -> None:
"""Fetch and process state change event.""" """Fetch and process state change event."""
self.state_change_data = StateChangedData(entity, old_state, new_state) data = event.data
self.state_change_data = StateChangedData(
data["entity_id"], data["old_state"], data["new_state"]
)
await self.async_refresh() await self.async_refresh()
async def async_check_tracked_entity_change( async def async_check_tracked_entity_change(