Add config flow for template sensor (#98970)
* Add config flow for template sensor * Tweak error reporting * Improve validation * Fix test * Rename translation strings * Improve validation * Fix sensor async_setup_entry * Avoid duplicating sensor device class translations * Avoid duplicating sensor device class translations * Add config flow tests * Include all units from DEVICE_CLASS_UNITS in unit_of_measurement select * Address review comments
This commit is contained in:
parent
bc5f934f35
commit
63c538b024
22 changed files with 1174 additions and 54 deletions
|
@ -17,6 +17,7 @@ from homeassistant.components.sensor import (
|
|||
SensorEntity,
|
||||
)
|
||||
from homeassistant.components.sensor.helpers import async_parse_date_datetime
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import (
|
||||
ATTR_ENTITY_ID,
|
||||
CONF_DEVICE_CLASS,
|
||||
|
@ -195,6 +196,28 @@ async def async_setup_platform(
|
|||
)
|
||||
|
||||
|
||||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
config_entry: ConfigEntry,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Initialize config entry."""
|
||||
_options = dict(config_entry.options)
|
||||
_options.pop("template_type")
|
||||
validated_config = SENSOR_SCHEMA(_options)
|
||||
async_add_entities([SensorTemplate(hass, validated_config, config_entry.entry_id)])
|
||||
|
||||
|
||||
@callback
|
||||
def async_create_preview_sensor(
|
||||
hass: HomeAssistant, name: str, config: dict[str, Any]
|
||||
) -> SensorTemplate:
|
||||
"""Create a preview sensor."""
|
||||
validated_config = SENSOR_SCHEMA(config | {CONF_NAME: name})
|
||||
entity = SensorTemplate(hass, validated_config, None)
|
||||
return entity
|
||||
|
||||
|
||||
class SensorTemplate(TemplateEntity, SensorEntity):
|
||||
"""Representation of a Template Sensor."""
|
||||
|
||||
|
@ -217,13 +240,14 @@ class SensorTemplate(TemplateEntity, SensorEntity):
|
|||
ENTITY_ID_FORMAT, object_id, hass=hass
|
||||
)
|
||||
|
||||
async def async_added_to_hass(self) -> None:
|
||||
"""Register callbacks."""
|
||||
@callback
|
||||
def _async_setup_templates(self) -> None:
|
||||
"""Set up templates."""
|
||||
self.add_template_attribute(
|
||||
"_attr_native_value", self._template, None, self._update_state
|
||||
)
|
||||
|
||||
await super().async_added_to_hass()
|
||||
super()._async_setup_templates()
|
||||
|
||||
@callback
|
||||
def _update_state(self, result):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue