evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve async_track_template_result callback typing (#97135)

Browse source
This commit is contained in:
Marc Mueller 2023-07-24 12:42:17 +02:00 • committed by GitHub
parent c0da6b822e
commit 582499a260
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 228 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

13
homeassistant/components/bayesian/binary_sensor.py
View file

@ -27,7 +27,7 @@ from homeassistant.const import (
STATE_UNAVAILABLE,
STATE_UNKNOWN,
)
from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ConditionError, TemplateError
from homeassistant.helpers import condition
import homeassistant.helpers.config_validation as cv
@ -259,14 +259,13 @@ class BayesianBinarySensor(BinarySensorEntity):
@callback
def _async_template_result_changed(
event: Event | None, updates: list[TrackTemplateResult]
event: EventType[EventStateChangedData] | None,
updates: list[TrackTemplateResult],
) -> None:
track_template_result = updates.pop()
template = track_template_result.template
result = track_template_result.result
entity: str | None = (
None if event is None else event.data.get(CONF_ENTITY_ID)
)
entity_id = None if event is None else event.data["entity_id"]
if isinstance(result, TemplateError):
_LOGGER.error(
"TemplateError('%s') while processing template '%s' in entity '%s'",
@ -283,8 +282,8 @@ class BayesianBinarySensor(BinarySensorEntity):
observation.observed = observed
# in some cases a template may update because of the absence of an entity
if entity is not None:
observation.entity_id = entity
if entity_id is not None:
observation.entity_id = entity_id
self.current_observations[observation.id] = observation