8ef542927f
* Add automation config validation * Remove unnecessary dict validator * Downgrade voluptuous to 0.8.9 * Fix linting * Address issues
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
"""
|
|
Offer template automation rules.
|
|
|
|
For more details about this automation rule, please refer to the documentation
|
|
at https://home-assistant.io/components/automation/#template-trigger
|
|
"""
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import (
|
|
CONF_VALUE_TEMPLATE, EVENT_STATE_CHANGED, CONF_PLATFORM)
|
|
from homeassistant.exceptions import TemplateError
|
|
from homeassistant.helpers import template
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
TRIGGER_SCHEMA = IF_ACTION_SCHEMA = vol.Schema({
|
|
vol.Required(CONF_PLATFORM): 'template',
|
|
vol.Required(CONF_VALUE_TEMPLATE): cv.template,
|
|
})
|
|
|
|
|
|
def trigger(hass, config, action):
|
|
"""Listen for state changes based on configuration."""
|
|
value_template = config.get(CONF_VALUE_TEMPLATE)
|
|
|
|
# Local variable to keep track of if the action has already been triggered
|
|
already_triggered = False
|
|
|
|
def event_listener(event):
|
|
"""Listen for state changes and calls action."""
|
|
nonlocal already_triggered
|
|
template_result = _check_template(hass, value_template)
|
|
|
|
# Check to see if template returns true
|
|
if template_result and not already_triggered:
|
|
already_triggered = True
|
|
action()
|
|
elif not template_result:
|
|
already_triggered = False
|
|
|
|
hass.bus.listen(EVENT_STATE_CHANGED, event_listener)
|
|
return True
|
|
|
|
|
|
def if_action(hass, config):
|
|
"""Wrap action method with state based condition."""
|
|
value_template = config.get(CONF_VALUE_TEMPLATE)
|
|
|
|
return lambda: _check_template(hass, value_template)
|
|
|
|
|
|
def _check_template(hass, value_template):
|
|
"""Check if result of template is true."""
|
|
try:
|
|
value = template.render(hass, value_template, {})
|
|
except TemplateError as ex:
|
|
if ex.args and ex.args[0].startswith(
|
|
"UndefinedError: 'None' has no attribute"):
|
|
# Common during HA startup - so just a warning
|
|
_LOGGER.warning(ex)
|
|
else:
|
|
_LOGGER.error(ex)
|
|
return False
|
|
|
|
return value.lower() == 'true'
|