Electric-Special/ha-core
1
0
Fork 0
mirror of https://github.com/Electric-Special/ha-core.git synced 2026-03-21 06:05:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add new template entity framework to event platform (#162228)

Browse Source
This commit is contained in:
Petro31
2026-02-09 01:54:40 -05:00
committed by GitHub
parent 5c5f5d064a
commit 3c266183e1
1 changed files with 13 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
homeassistant/components/template/event.py
View File

@@ -124,14 +124,24 @@ class AbstractTemplateEvent(AbstractTemplateEntity, EventEntity):
# This ensures that the __init__ on AbstractTemplateEntity is not called twice.
def __init__(self, config: dict[str, Any]) -> None: # pylint: disable=super-init-not-called
"""Initialize the features."""
self._event_type_template = config[CONF_EVENT_TYPE]
self._event_types_template = config[CONF_EVENT_TYPES]
self._attr_device_class = config.get(CONF_DEVICE_CLASS)
self._event_type = None
self._attr_event_types = []
self.setup_template(
CONF_EVENT_TYPES,
"_attr_event_types",
None,
self._update_event_types,
)
self.setup_template(
CONF_EVENT_TYPE,
"_event_type",
None,
self._update_event_type,
)
@callback
def _update_event_types(self, event_types: Any) -> None:
"""Update the event types from the template."""
@@ -179,25 +189,6 @@ class StateEventEntity(TemplateEntity, AbstractTemplateEvent):
TemplateEntity.__init__(self, hass, config, unique_id)
AbstractTemplateEvent.__init__(self, config)
@callback
def _async_setup_templates(self) -> None:
"""Set up templates."""
self.add_template_attribute(
"_attr_event_types",
self._event_types_template,
None,
self._update_event_types,
none_on_template_error=True,
)
self.add_template_attribute(
"_event_type",
self._event_type_template,
None,
self._update_event_type,
none_on_template_error=True,
)
super()._async_setup_templates()
class TriggerEventEntity(TriggerEntity, AbstractTemplateEvent, RestoreEntity):
"""Event entity based on trigger data."""
@@ -217,20 +208,3 @@ class TriggerEventEntity(TriggerEntity, AbstractTemplateEvent, RestoreEntity):
"""Initialize the entity."""
TriggerEntity.__init__(self, hass, coordinator, config)
AbstractTemplateEvent.__init__(self, config)
@callback
def _handle_coordinator_update(self) -> None:
"""Handle update of the data."""
self._process_data()
if not self.available:
return
for key, updater in (
(CONF_EVENT_TYPES, self._update_event_types),
(CONF_EVENT_TYPE, self._update_event_type),
):
updater(self._rendered[key])
self.async_set_context(self.coordinator.data["context"])
self.async_write_ha_state()