mirror of
https://github.com/esphome/esphome.git
synced 2024-09-07 00:59:08 +02:00
513 lines
21 KiB
Python
513 lines
21 KiB
Python
import voluptuous as vol
|
|
|
|
import esphomeyaml.config_validation as cv
|
|
from esphomeyaml import core
|
|
from esphomeyaml.components import cover, deep_sleep, fan, output
|
|
from esphomeyaml.const import CONF_ABOVE, CONF_ACTION_ID, CONF_AND, CONF_AUTOMATION_ID, \
|
|
CONF_BELOW, CONF_BLUE, CONF_BRIGHTNESS, CONF_CONDITION, CONF_CONDITION_ID, CONF_DELAY, \
|
|
CONF_EFFECT, CONF_ELSE, CONF_FLASH_LENGTH, CONF_GREEN, CONF_ID, CONF_IF, CONF_LAMBDA, \
|
|
CONF_LEVEL, CONF_OR, CONF_OSCILLATING, CONF_PAYLOAD, CONF_QOS, CONF_RANGE, CONF_RED, \
|
|
CONF_RETAIN, CONF_SPEED, CONF_THEN, CONF_TOPIC, CONF_TRANSITION_LENGTH, CONF_TRIGGER_ID, \
|
|
CONF_WHITE
|
|
from esphomeyaml.core import ESPHomeYAMLError
|
|
from esphomeyaml.helpers import App, ArrayInitializer, Pvariable, TemplateArguments, add, add_job, \
|
|
bool_, esphomelib_ns, float_, get_variable, process_lambda, std_string, templatable, uint32, \
|
|
uint8
|
|
|
|
CONF_MQTT_PUBLISH = 'mqtt.publish'
|
|
CONF_LIGHT_TOGGLE = 'light.toggle'
|
|
CONF_LIGHT_TURN_OFF = 'light.turn_off'
|
|
CONF_LIGHT_TURN_ON = 'light.turn_on'
|
|
CONF_SWITCH_TOGGLE = 'switch.toggle'
|
|
CONF_SWITCH_TURN_OFF = 'switch.turn_off'
|
|
CONF_SWITCH_TURN_ON = 'switch.turn_on'
|
|
CONF_COVER_OPEN = 'cover.open'
|
|
CONF_COVER_CLOSE = 'cover.close'
|
|
CONF_COVER_STOP = 'cover.stop'
|
|
CONF_FAN_TOGGLE = 'fan.toggle'
|
|
CONF_FAN_TURN_OFF = 'fan.turn_off'
|
|
CONF_FAN_TURN_ON = 'fan.turn_on'
|
|
CONF_OUTPUT_TURN_ON = 'output.turn_on'
|
|
CONF_OUTPUT_TURN_OFF = 'output.turn_off'
|
|
CONF_OUTPUT_SET_LEVEL = 'output.set_level'
|
|
CONF_DEEP_SLEEP_ENTER = 'deep_sleep.enter'
|
|
CONF_DEEP_SLEEP_PREVENT = 'deep_sleep.prevent'
|
|
|
|
|
|
def maybe_simple_id(*validators):
|
|
validator = vol.All(*validators)
|
|
|
|
def validate(value):
|
|
if isinstance(value, dict):
|
|
return validator(value)
|
|
return validator({CONF_ID: value})
|
|
|
|
return validate
|
|
|
|
|
|
def validate_recursive_condition(value):
|
|
return CONDITIONS_SCHEMA(value)
|
|
|
|
|
|
def validate_recursive_action(value):
|
|
return ACTIONS_SCHEMA(value)
|
|
|
|
|
|
ACTION_KEYS = [CONF_DELAY, CONF_MQTT_PUBLISH, CONF_LIGHT_TOGGLE, CONF_LIGHT_TURN_OFF,
|
|
CONF_LIGHT_TURN_ON, CONF_SWITCH_TOGGLE, CONF_SWITCH_TURN_OFF, CONF_SWITCH_TURN_ON,
|
|
CONF_LAMBDA, CONF_COVER_OPEN, CONF_COVER_CLOSE, CONF_COVER_STOP, CONF_FAN_TOGGLE,
|
|
CONF_FAN_TURN_OFF, CONF_FAN_TURN_ON, CONF_OUTPUT_TURN_ON, CONF_OUTPUT_TURN_OFF,
|
|
CONF_OUTPUT_SET_LEVEL, CONF_IF, CONF_DEEP_SLEEP_ENTER, CONF_DEEP_SLEEP_PREVENT]
|
|
|
|
ACTIONS_SCHEMA = vol.All(cv.ensure_list, [vol.All({
|
|
cv.GenerateID(CONF_ACTION_ID): cv.declare_variable_id(None),
|
|
vol.Optional(CONF_DELAY): cv.templatable(cv.positive_time_period_milliseconds),
|
|
vol.Optional(CONF_MQTT_PUBLISH): vol.Schema({
|
|
vol.Required(CONF_TOPIC): cv.templatable(cv.publish_topic),
|
|
vol.Required(CONF_PAYLOAD): cv.templatable(cv.mqtt_payload),
|
|
vol.Optional(CONF_QOS): cv.templatable(cv.mqtt_qos),
|
|
vol.Optional(CONF_RETAIN): cv.templatable(cv.boolean),
|
|
}),
|
|
vol.Optional(CONF_LIGHT_TOGGLE): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
vol.Optional(CONF_TRANSITION_LENGTH): cv.templatable(cv.positive_time_period_milliseconds),
|
|
}),
|
|
vol.Optional(CONF_LIGHT_TURN_OFF): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
vol.Optional(CONF_TRANSITION_LENGTH): cv.templatable(cv.positive_time_period_milliseconds),
|
|
}),
|
|
vol.Optional(CONF_LIGHT_TURN_ON): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
vol.Exclusive(CONF_TRANSITION_LENGTH, 'transformer'):
|
|
cv.templatable(cv.positive_time_period_milliseconds),
|
|
vol.Exclusive(CONF_FLASH_LENGTH, 'transformer'):
|
|
cv.templatable(cv.positive_time_period_milliseconds),
|
|
vol.Optional(CONF_BRIGHTNESS): cv.templatable(cv.percentage),
|
|
vol.Optional(CONF_RED): cv.templatable(cv.percentage),
|
|
vol.Optional(CONF_GREEN): cv.templatable(cv.percentage),
|
|
vol.Optional(CONF_BLUE): cv.templatable(cv.percentage),
|
|
vol.Optional(CONF_WHITE): cv.templatable(cv.percentage),
|
|
vol.Optional(CONF_EFFECT): cv.templatable(cv.string),
|
|
}),
|
|
vol.Optional(CONF_SWITCH_TOGGLE): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_SWITCH_TURN_OFF): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_SWITCH_TURN_ON): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_COVER_OPEN): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_COVER_CLOSE): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_COVER_STOP): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_COVER_OPEN): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_COVER_CLOSE): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_COVER_STOP): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_FAN_TOGGLE): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_FAN_TURN_OFF): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_FAN_TURN_ON): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
vol.Optional(CONF_OSCILLATING): cv.templatable(cv.boolean),
|
|
vol.Optional(CONF_SPEED): cv.templatable(fan.validate_fan_speed),
|
|
}),
|
|
vol.Optional(CONF_OUTPUT_TURN_OFF): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
}),
|
|
vol.Optional(CONF_OUTPUT_TURN_ON): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(None)
|
|
}),
|
|
vol.Optional(CONF_OUTPUT_SET_LEVEL): {
|
|
vol.Required(CONF_ID): cv.use_variable_id(None),
|
|
vol.Required(CONF_LEVEL): cv.percentage,
|
|
},
|
|
vol.Optional(CONF_DEEP_SLEEP_ENTER): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(deep_sleep.DeepSleepComponent),
|
|
}),
|
|
vol.Optional(CONF_DEEP_SLEEP_PREVENT): maybe_simple_id({
|
|
vol.Required(CONF_ID): cv.use_variable_id(deep_sleep.DeepSleepComponent),
|
|
}),
|
|
vol.Optional(CONF_IF): vol.All({
|
|
vol.Required(CONF_CONDITION): validate_recursive_condition,
|
|
vol.Optional(CONF_THEN): validate_recursive_action,
|
|
vol.Optional(CONF_ELSE): validate_recursive_action,
|
|
}, cv.has_at_least_one_key(CONF_THEN, CONF_ELSE)),
|
|
vol.Optional(CONF_LAMBDA): cv.lambda_,
|
|
}, cv.has_exactly_one_key(*ACTION_KEYS))])
|
|
|
|
# pylint: disable=invalid-name
|
|
DelayAction = esphomelib_ns.DelayAction
|
|
LambdaAction = esphomelib_ns.LambdaAction
|
|
IfAction = esphomelib_ns.IfAction
|
|
Automation = esphomelib_ns.Automation
|
|
|
|
CONDITIONS_SCHEMA = vol.All(cv.ensure_list, [cv.templatable({
|
|
cv.GenerateID(CONF_CONDITION_ID): cv.declare_variable_id(None),
|
|
vol.Optional(CONF_AND): validate_recursive_condition,
|
|
vol.Optional(CONF_OR): validate_recursive_condition,
|
|
vol.Optional(CONF_RANGE): vol.All(vol.Schema({
|
|
vol.Optional(CONF_ABOVE): vol.Coerce(float),
|
|
vol.Optional(CONF_BELOW): vol.Coerce(float),
|
|
}), cv.has_at_least_one_key(CONF_ABOVE, CONF_BELOW)),
|
|
vol.Optional(CONF_LAMBDA): cv.lambda_,
|
|
})])
|
|
|
|
# pylint: disable=invalid-name
|
|
AndCondition = esphomelib_ns.AndCondition
|
|
OrCondition = esphomelib_ns.OrCondition
|
|
RangeCondition = esphomelib_ns.RangeCondition
|
|
LambdaCondition = esphomelib_ns.LambdaCondition
|
|
|
|
|
|
def validate_automation(extra_schema=None):
|
|
schema = AUTOMATION_SCHEMA.extend(extra_schema or {})
|
|
|
|
def validator(value):
|
|
if isinstance(value, list):
|
|
return schema({CONF_THEN: value})
|
|
elif isinstance(value, dict):
|
|
if CONF_THEN in value:
|
|
return schema(value)
|
|
return schema({CONF_THEN: value})
|
|
return schema(value)
|
|
|
|
return validator
|
|
|
|
|
|
AUTOMATION_SCHEMA = vol.Schema({
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_variable_id(None),
|
|
cv.GenerateID(CONF_AUTOMATION_ID): cv.declare_variable_id(None),
|
|
vol.Optional(CONF_IF): CONDITIONS_SCHEMA,
|
|
vol.Required(CONF_THEN): ACTIONS_SCHEMA,
|
|
})
|
|
|
|
|
|
def build_condition(config, arg_type):
|
|
template_arg = TemplateArguments(arg_type)
|
|
if isinstance(config, core.Lambda):
|
|
lambda_ = None
|
|
for lambda_ in process_lambda(config, [(arg_type, 'x')]):
|
|
yield
|
|
yield LambdaCondition.new(template_arg, lambda_)
|
|
elif CONF_AND in config:
|
|
yield AndCondition.new(template_arg, build_conditions(config[CONF_AND], template_arg))
|
|
elif CONF_OR in config:
|
|
yield OrCondition.new(template_arg, build_conditions(config[CONF_OR], template_arg))
|
|
elif CONF_LAMBDA in config:
|
|
lambda_ = None
|
|
for lambda_ in process_lambda(config[CONF_LAMBDA], [(arg_type, 'x')]):
|
|
yield
|
|
yield LambdaCondition.new(template_arg, lambda_)
|
|
elif CONF_RANGE in config:
|
|
conf = config[CONF_RANGE]
|
|
rhs = RangeCondition.new(template_arg)
|
|
type = RangeCondition.template(template_arg)
|
|
condition = Pvariable(config[CONF_CONDITION_ID], rhs, type=type)
|
|
if CONF_ABOVE in conf:
|
|
template_ = None
|
|
for template_ in templatable(conf[CONF_ABOVE], arg_type, float_):
|
|
yield
|
|
condition.set_min(template_)
|
|
if CONF_BELOW in conf:
|
|
template_ = None
|
|
for template_ in templatable(conf[CONF_BELOW], arg_type, float_):
|
|
yield
|
|
condition.set_max(template_)
|
|
yield condition
|
|
else:
|
|
raise ESPHomeYAMLError(u"Unsupported condition {}".format(config))
|
|
|
|
|
|
def build_conditions(config, arg_type):
|
|
conditions = []
|
|
for conf in config:
|
|
condition = None
|
|
for condition in build_condition(conf, arg_type):
|
|
yield None
|
|
conditions.append(condition)
|
|
yield ArrayInitializer(*conditions)
|
|
|
|
|
|
def build_action(full_config, arg_type):
|
|
from esphomeyaml.components import light, mqtt, switch
|
|
|
|
template_arg = TemplateArguments(arg_type)
|
|
# Keep pylint from freaking out
|
|
var = None
|
|
action_id = full_config[CONF_ACTION_ID]
|
|
key, config = next((k, v) for k, v in full_config.items() if k in ACTION_KEYS)
|
|
if key == CONF_DELAY:
|
|
rhs = App.register_component(DelayAction.new(template_arg))
|
|
type = DelayAction.template(template_arg)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
template_ = None
|
|
for template_ in templatable(config, arg_type, uint32):
|
|
yield
|
|
add(action.set_delay(template_))
|
|
yield action
|
|
elif key == CONF_LAMBDA:
|
|
lambda_ = None
|
|
for lambda_ in process_lambda(config, [(arg_type, 'x')]):
|
|
yield None
|
|
rhs = LambdaAction.new(template_arg, lambda_)
|
|
type = LambdaAction.template(template_arg)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_MQTT_PUBLISH:
|
|
rhs = App.Pget_mqtt_client().Pmake_publish_action(template_arg)
|
|
type = mqtt.MQTTPublishAction.template(template_arg)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_TOPIC], arg_type, std_string):
|
|
yield None
|
|
add(action.set_topic(template_))
|
|
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_PAYLOAD], arg_type, std_string):
|
|
yield None
|
|
add(action.set_payload(template_))
|
|
if CONF_QOS in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_QOS], arg_type, uint8):
|
|
yield
|
|
add(action.set_qos(template_))
|
|
if CONF_RETAIN in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_RETAIN], arg_type, bool_):
|
|
yield None
|
|
add(action.set_retain(template_))
|
|
yield action
|
|
elif key == CONF_LIGHT_TOGGLE:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_toggle_action(template_arg)
|
|
type = light.ToggleAction.template(template_arg)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
if CONF_TRANSITION_LENGTH in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_TRANSITION_LENGTH], arg_type, uint32):
|
|
yield None
|
|
add(action.set_transition_length(template_))
|
|
yield action
|
|
elif key == CONF_LIGHT_TURN_OFF:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_off_action(template_arg)
|
|
type = light.TurnOffAction.template(template_arg)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
if CONF_TRANSITION_LENGTH in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_TRANSITION_LENGTH], arg_type, uint32):
|
|
yield None
|
|
add(action.set_transition_length(template_))
|
|
yield action
|
|
elif key == CONF_LIGHT_TURN_ON:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_on_action(template_arg)
|
|
type = light.TurnOnAction.template(template_arg)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
if CONF_TRANSITION_LENGTH in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_TRANSITION_LENGTH], arg_type, uint32):
|
|
yield None
|
|
add(action.set_transition_length(template_))
|
|
if CONF_FLASH_LENGTH in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_FLASH_LENGTH], arg_type, uint32):
|
|
yield None
|
|
add(action.set_flash_length(template_))
|
|
if CONF_BRIGHTNESS in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_BRIGHTNESS], arg_type, float_):
|
|
yield None
|
|
add(action.set_brightness(template_))
|
|
if CONF_RED in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_RED], arg_type, float_):
|
|
yield None
|
|
add(action.set_red(template_))
|
|
if CONF_GREEN in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_GREEN], arg_type, float_):
|
|
yield None
|
|
add(action.set_green(template_))
|
|
if CONF_BLUE in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_BLUE], arg_type, float_):
|
|
yield None
|
|
add(action.set_blue(template_))
|
|
if CONF_WHITE in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_WHITE], arg_type, float_):
|
|
yield None
|
|
add(action.set_white(template_))
|
|
if CONF_EFFECT in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_EFFECT], arg_type, std_string):
|
|
yield None
|
|
add(action.set_effect(template_))
|
|
yield action
|
|
elif key == CONF_SWITCH_TOGGLE:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_toggle_action(template_arg)
|
|
type = switch.ToggleAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_SWITCH_TURN_OFF:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_off_action(template_arg)
|
|
type = switch.TurnOffAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_SWITCH_TURN_ON:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_on_action(template_arg)
|
|
type = switch.TurnOnAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_COVER_OPEN:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_open_action(template_arg)
|
|
type = cover.OpenAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_COVER_CLOSE:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_close_action(template_arg)
|
|
type = cover.CloseAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_COVER_STOP:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_stop_action(template_arg)
|
|
type = cover.StopAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_FAN_TOGGLE:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_toggle_action(template_arg)
|
|
type = fan.ToggleAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_FAN_TURN_OFF:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_off_action(template_arg)
|
|
type = fan.TurnOffAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_FAN_TURN_ON:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_on_action(template_arg)
|
|
type = fan.TurnOnAction.template(arg_type)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
if CONF_OSCILLATING in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_OSCILLATING], arg_type, bool_):
|
|
yield None
|
|
add(action.set_oscillating(template_))
|
|
if CONF_SPEED in config:
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_SPEED], arg_type, fan.FanSpeed):
|
|
yield None
|
|
add(action.set_speed(template_))
|
|
yield action
|
|
elif key == CONF_OUTPUT_TURN_OFF:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_off_action(template_arg)
|
|
type = output.TurnOffAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_OUTPUT_TURN_ON:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_turn_on_action(template_arg)
|
|
type = output.TurnOnAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_OUTPUT_SET_LEVEL:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_set_level_action(template_arg)
|
|
type = output.SetLevelAction.template(arg_type)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
template_ = None
|
|
for template_ in templatable(config[CONF_LEVEL], arg_type, bool_):
|
|
yield None
|
|
add(action.set_level(template_))
|
|
yield action
|
|
elif key == CONF_IF:
|
|
for conditions in build_conditions(config[CONF_CONDITION], arg_type):
|
|
yield None
|
|
rhs = IfAction.new(template_arg, conditions)
|
|
type = IfAction.template(template_arg)
|
|
action = Pvariable(action_id, rhs, type=type)
|
|
if CONF_THEN in config:
|
|
for actions in build_actions(config[CONF_THEN], arg_type):
|
|
yield None
|
|
add(action.add_then(actions))
|
|
if CONF_ELSE in config:
|
|
for actions in build_actions(config[CONF_ELSE], arg_type):
|
|
yield None
|
|
add(action.add_else(actions))
|
|
yield action
|
|
elif key == CONF_DEEP_SLEEP_ENTER:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_enter_deep_sleep_action(template_arg)
|
|
type = deep_sleep.EnterDeepSleepAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
elif key == CONF_DEEP_SLEEP_PREVENT:
|
|
for var in get_variable(config[CONF_ID]):
|
|
yield None
|
|
rhs = var.make_prevent_deep_sleep_action(template_arg)
|
|
type = deep_sleep.PreventDeepSleepAction.template(arg_type)
|
|
yield Pvariable(action_id, rhs, type=type)
|
|
else:
|
|
raise ESPHomeYAMLError(u"Unsupported action {}".format(config))
|
|
|
|
|
|
def build_actions(config, arg_type):
|
|
actions = []
|
|
for conf in config:
|
|
action = None
|
|
for action in build_action(conf, arg_type):
|
|
yield None
|
|
actions.append(action)
|
|
yield ArrayInitializer(*actions, multiline=False)
|
|
|
|
|
|
def build_automation_(trigger, arg_type, config):
|
|
rhs = App.make_automation(TemplateArguments(arg_type), trigger)
|
|
type = Automation.template(arg_type)
|
|
obj = Pvariable(config[CONF_AUTOMATION_ID], rhs, type=type)
|
|
if CONF_IF in config:
|
|
conditions = None
|
|
for conditions in build_conditions(config[CONF_IF], arg_type):
|
|
yield None
|
|
add(obj.add_conditions(conditions))
|
|
actions = None
|
|
for actions in build_actions(config[CONF_THEN], arg_type):
|
|
yield None
|
|
add(obj.add_actions(actions))
|
|
yield obj
|
|
|
|
|
|
def build_automation(trigger, arg_type, config):
|
|
add_job(build_automation_, trigger, arg_type, config)
|