Allow top level non-trigger template entities (#48976)

This commit is contained in:
Paulus Schoutsen 2021-04-12 17:15:50 -07:00 committed by GitHub
parent 93c68f8be6
commit 65126cec3e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 211 additions and 157 deletions

View file

@ -6,7 +6,6 @@ import logging
from typing import Callable from typing import Callable
from homeassistant import config as conf_util from homeassistant import config as conf_util
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
from homeassistant.const import EVENT_HOMEASSISTANT_START, SERVICE_RELOAD from homeassistant.const import EVENT_HOMEASSISTANT_START, SERVICE_RELOAD
from homeassistant.core import CoreState, Event, callback from homeassistant.core import CoreState, Event, callback
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
@ -57,23 +56,41 @@ async def async_setup(hass, config):
return True return True
async def _process_config(hass, config): async def _process_config(hass, hass_config):
"""Process config.""" """Process config."""
coordinators: list[TriggerUpdateCoordinator] | None = hass.data.get(DOMAIN) coordinators: list[TriggerUpdateCoordinator] | None = hass.data.pop(DOMAIN, None)
# Remove old ones # Remove old ones
if coordinators: if coordinators:
for coordinator in coordinators: for coordinator in coordinators:
coordinator.async_remove() coordinator.async_remove()
async def init_coordinator(hass, conf): async def init_coordinator(hass, conf_section):
coordinator = TriggerUpdateCoordinator(hass, conf) coordinator = TriggerUpdateCoordinator(hass, conf_section)
await coordinator.async_setup(config) await coordinator.async_setup(hass_config)
return coordinator return coordinator
hass.data[DOMAIN] = await asyncio.gather( coordinator_tasks = []
*[init_coordinator(hass, conf) for conf in config[DOMAIN]]
) for conf_section in hass_config[DOMAIN]:
if CONF_TRIGGER in conf_section:
coordinator_tasks.append(init_coordinator(hass, conf_section))
continue
for platform_domain in PLATFORMS:
if platform_domain in conf_section:
hass.async_create_task(
discovery.async_load_platform(
hass,
platform_domain,
DOMAIN,
{"entities": conf_section[platform_domain]},
hass_config,
)
)
if coordinator_tasks:
hass.data[DOMAIN] = await asyncio.gather(*coordinator_tasks)
class TriggerUpdateCoordinator(update_coordinator.DataUpdateCoordinator): class TriggerUpdateCoordinator(update_coordinator.DataUpdateCoordinator):
@ -110,16 +127,17 @@ class TriggerUpdateCoordinator(update_coordinator.DataUpdateCoordinator):
EVENT_HOMEASSISTANT_START, self._attach_triggers EVENT_HOMEASSISTANT_START, self._attach_triggers
) )
for platform_domain in (SENSOR_DOMAIN,): for platform_domain in PLATFORMS:
self.hass.async_create_task( if platform_domain in self.config:
discovery.async_load_platform( self.hass.async_create_task(
self.hass, discovery.async_load_platform(
platform_domain, self.hass,
DOMAIN, platform_domain,
{"coordinator": self, "entities": self.config[platform_domain]}, DOMAIN,
hass_config, {"coordinator": self, "entities": self.config[platform_domain]},
hass_config,
)
) )
)
async def _attach_triggers(self, start_event=None) -> None: async def _attach_triggers(self, start_event=None) -> None:
"""Attach the triggers.""" """Attach the triggers."""

View file

@ -3,101 +3,29 @@ import logging
import voluptuous as vol import voluptuous as vol
from homeassistant.components.sensor import ( from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
DEVICE_CLASSES_SCHEMA as SENSOR_DEVICE_CLASSES_SCHEMA,
DOMAIN as SENSOR_DOMAIN,
)
from homeassistant.config import async_log_exception, config_without_domain from homeassistant.config import async_log_exception, config_without_domain
from homeassistant.const import ( from homeassistant.const import CONF_SENSORS, CONF_UNIQUE_ID
CONF_DEVICE_CLASS, from homeassistant.helpers import config_validation as cv
CONF_ENTITY_PICTURE_TEMPLATE,
CONF_FRIENDLY_NAME,
CONF_FRIENDLY_NAME_TEMPLATE,
CONF_ICON,
CONF_ICON_TEMPLATE,
CONF_NAME,
CONF_SENSORS,
CONF_STATE,
CONF_UNIQUE_ID,
CONF_UNIT_OF_MEASUREMENT,
CONF_VALUE_TEMPLATE,
)
from homeassistant.helpers import config_validation as cv, template
from homeassistant.helpers.trigger import async_validate_trigger_config from homeassistant.helpers.trigger import async_validate_trigger_config
from .const import ( from . import sensor as sensor_platform
CONF_ATTRIBUTE_TEMPLATES, from .const import CONF_TRIGGER, DOMAIN
CONF_ATTRIBUTES,
CONF_AVAILABILITY,
CONF_AVAILABILITY_TEMPLATE,
CONF_PICTURE,
CONF_TRIGGER,
DOMAIN,
)
from .sensor import SENSOR_SCHEMA as PLATFORM_SENSOR_SCHEMA
LEGACY_SENSOR = {
CONF_ICON_TEMPLATE: CONF_ICON,
CONF_ENTITY_PICTURE_TEMPLATE: CONF_PICTURE,
CONF_AVAILABILITY_TEMPLATE: CONF_AVAILABILITY,
CONF_ATTRIBUTE_TEMPLATES: CONF_ATTRIBUTES,
CONF_FRIENDLY_NAME_TEMPLATE: CONF_NAME,
CONF_FRIENDLY_NAME: CONF_NAME,
CONF_VALUE_TEMPLATE: CONF_STATE,
}
SENSOR_SCHEMA = vol.Schema(
{
vol.Optional(CONF_NAME): cv.template,
vol.Required(CONF_STATE): cv.template,
vol.Optional(CONF_ICON): cv.template,
vol.Optional(CONF_PICTURE): cv.template,
vol.Optional(CONF_AVAILABILITY): cv.template,
vol.Optional(CONF_ATTRIBUTES): vol.Schema({cv.string: cv.template}),
vol.Optional(CONF_UNIT_OF_MEASUREMENT): cv.string,
vol.Optional(CONF_DEVICE_CLASS): SENSOR_DEVICE_CLASSES_SCHEMA,
vol.Optional(CONF_UNIQUE_ID): cv.string,
}
)
CONFIG_SECTION_SCHEMA = vol.Schema( CONFIG_SECTION_SCHEMA = vol.Schema(
{ {
vol.Optional(CONF_UNIQUE_ID): cv.string, vol.Optional(CONF_UNIQUE_ID): cv.string,
vol.Required(CONF_TRIGGER): cv.TRIGGER_SCHEMA, vol.Optional(CONF_TRIGGER): cv.TRIGGER_SCHEMA,
vol.Optional(SENSOR_DOMAIN): vol.All(cv.ensure_list, [SENSOR_SCHEMA]), vol.Optional(SENSOR_DOMAIN): vol.All(
vol.Optional(CONF_SENSORS): cv.schema_with_slug_keys(PLATFORM_SENSOR_SCHEMA), cv.ensure_list, [sensor_platform.SENSOR_SCHEMA]
),
vol.Optional(CONF_SENSORS): cv.schema_with_slug_keys(
sensor_platform.LEGACY_SENSOR_SCHEMA
),
} }
) )
def _rewrite_legacy_to_modern_trigger_conf(cfg: dict):
"""Rewrite a legacy to a modern trigger-basd conf."""
logging.getLogger(__name__).warning(
"The entity definition format under template: differs from the platform configuration format. See https://www.home-assistant.io/integrations/template#configuration-for-trigger-based-template-sensors"
)
sensor = list(cfg[SENSOR_DOMAIN]) if SENSOR_DOMAIN in cfg else []
for device_id, entity_cfg in cfg[CONF_SENSORS].items():
entity_cfg = {**entity_cfg}
for from_key, to_key in LEGACY_SENSOR.items():
if from_key not in entity_cfg or to_key in entity_cfg:
continue
val = entity_cfg.pop(from_key)
if isinstance(val, str):
val = template.Template(val)
entity_cfg[to_key] = val
if CONF_NAME not in entity_cfg:
entity_cfg[CONF_NAME] = template.Template(device_id)
sensor.append(entity_cfg)
return {**cfg, "sensor": sensor}
async def async_validate_config(hass, config): async def async_validate_config(hass, config):
"""Validate config.""" """Validate config."""
if DOMAIN not in config: if DOMAIN not in config:
@ -108,15 +36,26 @@ async def async_validate_config(hass, config):
for cfg in cv.ensure_list(config[DOMAIN]): for cfg in cv.ensure_list(config[DOMAIN]):
try: try:
cfg = CONFIG_SECTION_SCHEMA(cfg) cfg = CONFIG_SECTION_SCHEMA(cfg)
cfg[CONF_TRIGGER] = await async_validate_trigger_config(
hass, cfg[CONF_TRIGGER] if CONF_TRIGGER in cfg:
) cfg[CONF_TRIGGER] = await async_validate_trigger_config(
hass, cfg[CONF_TRIGGER]
)
except vol.Invalid as err: except vol.Invalid as err:
async_log_exception(err, DOMAIN, cfg, hass) async_log_exception(err, DOMAIN, cfg, hass)
continue continue
if CONF_TRIGGER in cfg and CONF_SENSORS in cfg: if CONF_SENSORS in cfg:
cfg = _rewrite_legacy_to_modern_trigger_conf(cfg) logging.getLogger(__name__).warning(
"The entity definition format under template: differs from the platform "
"configuration format. See "
"https://www.home-assistant.io/integrations/template#configuration-for-trigger-based-template-sensors"
)
sensors = list(cfg[SENSOR_DOMAIN]) if SENSOR_DOMAIN in cfg else []
sensors.extend(
sensor_platform.rewrite_legacy_to_modern_conf(cfg[CONF_SENSORS])
)
cfg = {**cfg, "sensor": sensors}
config_sections.append(cfg) config_sections.append(cfg)

View file

@ -24,3 +24,4 @@ PLATFORMS = [
CONF_AVAILABILITY = "availability" CONF_AVAILABILITY = "availability"
CONF_ATTRIBUTES = "attributes" CONF_ATTRIBUTES = "attributes"
CONF_PICTURE = "picture" CONF_PICTURE = "picture"
CONF_OBJECT_ID = "object_id"

View file

@ -16,23 +16,59 @@ from homeassistant.const import (
CONF_ENTITY_PICTURE_TEMPLATE, CONF_ENTITY_PICTURE_TEMPLATE,
CONF_FRIENDLY_NAME, CONF_FRIENDLY_NAME,
CONF_FRIENDLY_NAME_TEMPLATE, CONF_FRIENDLY_NAME_TEMPLATE,
CONF_ICON,
CONF_ICON_TEMPLATE, CONF_ICON_TEMPLATE,
CONF_NAME,
CONF_SENSORS, CONF_SENSORS,
CONF_STATE, CONF_STATE,
CONF_UNIQUE_ID, CONF_UNIQUE_ID,
CONF_UNIT_OF_MEASUREMENT, CONF_UNIT_OF_MEASUREMENT,
CONF_VALUE_TEMPLATE, CONF_VALUE_TEMPLATE,
) )
from homeassistant.core import callback from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import TemplateError from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv from homeassistant.helpers import config_validation as cv, template
from homeassistant.helpers.entity import async_generate_entity_id from homeassistant.helpers.entity import async_generate_entity_id
from .const import CONF_ATTRIBUTE_TEMPLATES, CONF_AVAILABILITY_TEMPLATE, CONF_TRIGGER from .const import (
CONF_ATTRIBUTE_TEMPLATES,
CONF_ATTRIBUTES,
CONF_AVAILABILITY,
CONF_AVAILABILITY_TEMPLATE,
CONF_OBJECT_ID,
CONF_PICTURE,
CONF_TRIGGER,
)
from .template_entity import TemplateEntity from .template_entity import TemplateEntity
from .trigger_entity import TriggerEntity from .trigger_entity import TriggerEntity
SENSOR_SCHEMA = vol.All( LEGACY_FIELDS = {
CONF_ICON_TEMPLATE: CONF_ICON,
CONF_ENTITY_PICTURE_TEMPLATE: CONF_PICTURE,
CONF_AVAILABILITY_TEMPLATE: CONF_AVAILABILITY,
CONF_ATTRIBUTE_TEMPLATES: CONF_ATTRIBUTES,
CONF_FRIENDLY_NAME_TEMPLATE: CONF_NAME,
CONF_FRIENDLY_NAME: CONF_NAME,
CONF_VALUE_TEMPLATE: CONF_STATE,
}
SENSOR_SCHEMA = vol.Schema(
{
vol.Optional(CONF_NAME): cv.template,
vol.Required(CONF_STATE): cv.template,
vol.Optional(CONF_ICON): cv.template,
vol.Optional(CONF_PICTURE): cv.template,
vol.Optional(CONF_AVAILABILITY): cv.template,
vol.Optional(CONF_ATTRIBUTES): vol.Schema({cv.string: cv.template}),
vol.Optional(CONF_UNIT_OF_MEASUREMENT): cv.string,
vol.Optional(CONF_DEVICE_CLASS): DEVICE_CLASSES_SCHEMA,
vol.Optional(CONF_UNIQUE_ID): cv.string,
}
)
LEGACY_SENSOR_SCHEMA = vol.All(
cv.deprecated(ATTR_ENTITY_ID), cv.deprecated(ATTR_ENTITY_ID),
vol.Schema( vol.Schema(
{ {
@ -54,50 +90,78 @@ SENSOR_SCHEMA = vol.All(
) )
def trigger_warning(val): def extra_validation_checks(val):
"""Warn if a trigger is defined.""" """Run extra validation checks."""
if CONF_TRIGGER in val: if CONF_TRIGGER in val:
raise vol.Invalid( raise vol.Invalid(
"You can only add triggers to template entities if they are defined under `template:`. " "You can only add triggers to template entities if they are defined under `template:`. "
"See the template documentation for more information: https://www.home-assistant.io/integrations/template/" "See the template documentation for more information: https://www.home-assistant.io/integrations/template/"
) )
if CONF_SENSORS not in val and SENSOR_DOMAIN not in val:
raise vol.Invalid(f"Required key {SENSOR_DOMAIN} not defined")
return val return val
def rewrite_legacy_to_modern_conf(cfg: dict[str, dict]) -> list[dict]:
"""Rewrite a legacy sensor definitions to modern ones."""
sensors = []
for object_id, entity_cfg in cfg.items():
entity_cfg = {**entity_cfg, CONF_OBJECT_ID: object_id}
for from_key, to_key in LEGACY_FIELDS.items():
if from_key not in entity_cfg or to_key in entity_cfg:
continue
val = entity_cfg.pop(from_key)
if isinstance(val, str):
val = template.Template(val)
entity_cfg[to_key] = val
if CONF_NAME not in entity_cfg:
entity_cfg[CONF_NAME] = template.Template(object_id)
sensors.append(entity_cfg)
return sensors
PLATFORM_SCHEMA = vol.All( PLATFORM_SCHEMA = vol.All(
PLATFORM_SCHEMA.extend( PLATFORM_SCHEMA.extend(
{ {
vol.Optional(CONF_TRIGGER): cv.match_all, # to raise custom warning vol.Optional(CONF_TRIGGER): cv.match_all, # to raise custom warning
vol.Required(CONF_SENSORS): cv.schema_with_slug_keys(SENSOR_SCHEMA), vol.Required(CONF_SENSORS): cv.schema_with_slug_keys(LEGACY_SENSOR_SCHEMA),
} }
), ),
trigger_warning, extra_validation_checks,
) )
@callback @callback
def _async_create_template_tracking_entities(hass, config): def _async_create_template_tracking_entities(async_add_entities, hass, definitions):
"""Create the template sensors.""" """Create the template sensors."""
sensors = [] sensors = []
for device, device_config in config[CONF_SENSORS].items(): for entity_conf in definitions:
state_template = device_config[CONF_VALUE_TEMPLATE] # Still available on legacy
icon_template = device_config.get(CONF_ICON_TEMPLATE) object_id = entity_conf.get(CONF_OBJECT_ID)
entity_picture_template = device_config.get(CONF_ENTITY_PICTURE_TEMPLATE)
availability_template = device_config.get(CONF_AVAILABILITY_TEMPLATE) state_template = entity_conf[CONF_STATE]
friendly_name = device_config.get(CONF_FRIENDLY_NAME, device) icon_template = entity_conf.get(CONF_ICON)
friendly_name_template = device_config.get(CONF_FRIENDLY_NAME_TEMPLATE) entity_picture_template = entity_conf.get(CONF_PICTURE)
unit_of_measurement = device_config.get(CONF_UNIT_OF_MEASUREMENT) availability_template = entity_conf.get(CONF_AVAILABILITY)
device_class = device_config.get(CONF_DEVICE_CLASS) friendly_name_template = entity_conf.get(CONF_NAME)
attribute_templates = device_config.get(CONF_ATTRIBUTE_TEMPLATES, {}) unit_of_measurement = entity_conf.get(CONF_UNIT_OF_MEASUREMENT)
unique_id = device_config.get(CONF_UNIQUE_ID) device_class = entity_conf.get(CONF_DEVICE_CLASS)
attribute_templates = entity_conf.get(CONF_ATTRIBUTES, {})
unique_id = entity_conf.get(CONF_UNIQUE_ID)
sensors.append( sensors.append(
SensorTemplate( SensorTemplate(
hass, hass,
device, object_id,
friendly_name,
friendly_name_template, friendly_name_template,
unit_of_measurement, unit_of_measurement,
state_template, state_template,
@ -110,18 +174,29 @@ def _async_create_template_tracking_entities(hass, config):
) )
) )
return sensors async_add_entities(sensors)
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
"""Set up the template sensors.""" """Set up the template sensors."""
if discovery_info is None: if discovery_info is None:
async_add_entities(_async_create_template_tracking_entities(hass, config)) _async_create_template_tracking_entities(
else: async_add_entities,
hass,
rewrite_legacy_to_modern_conf(config[CONF_SENSORS]),
)
return
if "coordinator" in discovery_info:
async_add_entities( async_add_entities(
TriggerSensorEntity(hass, discovery_info["coordinator"], config) TriggerSensorEntity(hass, discovery_info["coordinator"], config)
for config in discovery_info["entities"] for config in discovery_info["entities"]
) )
return
_async_create_template_tracking_entities(
async_add_entities, hass, discovery_info["entities"]
)
class SensorTemplate(TemplateEntity, SensorEntity): class SensorTemplate(TemplateEntity, SensorEntity):
@ -129,18 +204,17 @@ class SensorTemplate(TemplateEntity, SensorEntity):
def __init__( def __init__(
self, self,
hass, hass: HomeAssistant,
device_id, object_id: str | None,
friendly_name, friendly_name_template: template.Template | None,
friendly_name_template, unit_of_measurement: str | None,
unit_of_measurement, state_template: template.Template,
state_template, icon_template: template.Template | None,
icon_template, entity_picture_template: template.Template | None,
entity_picture_template, availability_template: template.Template | None,
availability_template, device_class: str | None,
device_class, attribute_templates: dict[str, template.Template],
attribute_templates, unique_id: str | None,
unique_id,
): ):
"""Initialize the sensor.""" """Initialize the sensor."""
super().__init__( super().__init__(
@ -149,11 +223,22 @@ class SensorTemplate(TemplateEntity, SensorEntity):
icon_template=icon_template, icon_template=icon_template,
entity_picture_template=entity_picture_template, entity_picture_template=entity_picture_template,
) )
self.entity_id = async_generate_entity_id( if object_id is not None:
ENTITY_ID_FORMAT, device_id, hass=hass self.entity_id = async_generate_entity_id(
) ENTITY_ID_FORMAT, object_id, hass=hass
self._name = friendly_name )
self._name: str | None = None
self._friendly_name_template = friendly_name_template self._friendly_name_template = friendly_name_template
# Try to render the name as it can influence the entity ID
if friendly_name_template:
friendly_name_template.hass = hass
try:
self._name = friendly_name_template.async_render(parse_result=False)
except template.TemplateError:
pass
self._unit_of_measurement = unit_of_measurement self._unit_of_measurement = unit_of_measurement
self._template = state_template self._template = state_template
self._state = None self._state = None
@ -164,7 +249,7 @@ class SensorTemplate(TemplateEntity, SensorEntity):
async def async_added_to_hass(self): async def async_added_to_hass(self):
"""Register callbacks.""" """Register callbacks."""
self.add_template_attribute("_state", self._template, None, self._update_state) self.add_template_attribute("_state", self._template, None, self._update_state)
if self._friendly_name_template is not None: if self._friendly_name_template and not self._friendly_name_template.is_static:
self.add_template_attribute("_name", self._friendly_name_template) self.add_template_attribute("_name", self._friendly_name_template)
await super().async_added_to_hass() await super().async_added_to_hass()

View file

@ -28,26 +28,37 @@ async def test_reloadable(hass):
}, },
}, },
}, },
"template": { "template": [
"trigger": {"platform": "event", "event_type": "event_1"}, {
"sensor": { "trigger": {"platform": "event", "event_type": "event_1"},
"name": "top level", "sensor": {
"state": "{{ trigger.event.data.source }}", "name": "top level",
"state": "{{ trigger.event.data.source }}",
},
}, },
}, {
"sensor": {
"name": "top level state",
"state": "{{ states.sensor.top_level.state }} + 2",
},
},
],
}, },
) )
await hass.async_block_till_done() await hass.async_block_till_done()
await hass.async_start() await hass.async_start()
await hass.async_block_till_done() await hass.async_block_till_done()
assert hass.states.get("sensor.top_level_state").state == "unknown + 2"
hass.bus.async_fire("event_1", {"source": "init"}) hass.bus.async_fire("event_1", {"source": "init"})
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 4
assert hass.states.get("sensor.state").state == "mytest" assert hass.states.get("sensor.state").state == "mytest"
assert hass.states.get("sensor.top_level").state == "init" assert hass.states.get("sensor.top_level").state == "init"
await hass.async_block_till_done()
assert hass.states.get("sensor.top_level_state").state == "init + 2"
yaml_path = path.join( yaml_path = path.join(
_get_fixtures_base_path(), _get_fixtures_base_path(),

View file

@ -26,7 +26,7 @@ import homeassistant.util.dt as dt_util
from tests.common import assert_setup_component, async_fire_time_changed from tests.common import assert_setup_component, async_fire_time_changed
async def test_template(hass): async def test_template_legacy(hass):
"""Test template.""" """Test template."""
with assert_setup_component(1, sensor.DOMAIN): with assert_setup_component(1, sensor.DOMAIN):
assert await async_setup_component( assert await async_setup_component(