From 7b14bd89a0a0a73d7815fd73ca99c373cef0dd67 Mon Sep 17 00:00:00 2001 From: G Johansson Date: Mon, 30 Dec 2024 23:15:37 +0000 Subject: [PATCH] Fix template preview --- homeassistant/components/template/config_flow.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/homeassistant/components/template/config_flow.py b/homeassistant/components/template/config_flow.py index e6cc377bc26..bb8ee29d3e2 100644 --- a/homeassistant/components/template/config_flow.py +++ b/homeassistant/components/template/config_flow.py @@ -3,7 +3,9 @@ from __future__ import annotations from collections.abc import Callable, Coroutine, Mapping +from datetime import timedelta from functools import partial +import logging from typing import Any, cast import voluptuous as vol @@ -32,6 +34,7 @@ from homeassistant.const import ( from homeassistant.core import HomeAssistant, callback from homeassistant.exceptions import HomeAssistantError from homeassistant.helpers import entity_registry as er, selector +from homeassistant.helpers.entity_platform import EntityPlatform from homeassistant.helpers.schema_config_entry_flow import ( SchemaCommonFlowHandler, SchemaConfigFlowHandler, @@ -68,6 +71,8 @@ from .sensor import async_create_preview_sensor from .switch import async_create_preview_switch from .template_entity import TemplateEntity +_LOGGER = logging.getLogger(__name__) + _SCHEMA_STATE: dict[vol.Marker, Any] = { vol.Required(CONF_STATE): selector.TemplateSelector(), } @@ -524,6 +529,15 @@ def ws_start_preview( preview_entity = CREATE_PREVIEW_ENTITY[template_type](hass, name, msg["user_input"]) preview_entity.hass = hass preview_entity.registry_entry = entity_registry_entry + preview_entity.platform = EntityPlatform( + hass=hass, + logger=_LOGGER, + domain=template_type, + platform_name=DOMAIN, + platform=None, + scan_interval=timedelta(hours=1), + entity_namespace=None, + ) connection.send_result(msg["id"]) connection.subscriptions[msg["id"]] = preview_entity.async_start_preview(