github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-10-04 07:58:43 +02:00
Code Releases Activity

Improve generic event typing [recorder] (#114736)

Browse Source
This commit is contained in:
Marc Mueller 2024-04-06 21:35:42 +02:00 committed by GitHub
parent 2e3cb1a767
commit 8f425b9ea7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
homeassistant/components/recorder/entity_registry.py
View File

@ -1,8 +1,7 @@
"""Recorder entity registry helper."""
from collections.abc import Mapping
import logging
from typing import Any
from typing import TYPE_CHECKING
from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
@ -19,10 +18,14 @@ def async_setup(hass: HomeAssistant) -> None:
"""Set up the entity hooks."""
@callback
def _async_entity_id_changed(event: Event) -> None:
def _async_entity_id_changed(
event: Event[er.EventEntityRegistryUpdatedData],
) -> None:
instance = get_instance(hass)
old_entity_id: str = event.data["old_entity_id"]
new_entity_id: str = event.data["entity_id"]
if TYPE_CHECKING:
assert event.data["action"] == "update" and "old_entity_id" in event.data
old_entity_id = event.data["old_entity_id"]
new_entity_id = event.data["entity_id"]
instance.async_update_statistics_metadata(
old_entity_id, new_statistic_id=new_entity_id
)
@ -31,7 +34,9 @@ def async_setup(hass: HomeAssistant) -> None:
)
@callback
def entity_registry_changed_filter(event_data: Mapping[str, Any]) -> bool:
def entity_registry_changed_filter(
event_data: er.EventEntityRegistryUpdatedData,
) -> bool:
"""Handle entity_id changed filter."""
return event_data["action"] == "update" and "old_entity_id" in event_data