mirror of
https://github.com/esphome/esphome.git
synced 2024-11-14 02:58:11 +01:00
224 lines
10 KiB
Python
224 lines
10 KiB
Python
|
import voluptuous as vol
|
||
|
|
|
||
|
|
import esphomeyaml.config_validation as cv
|
||
|
|
from esphomeyaml.const import CONF_ACTION_ID, CONF_AND, CONF_AUTOMATION_ID, CONF_BLUE, \
|
||
|
|
CONF_BRIGHTNESS, CONF_CONDITION_ID, CONF_DELAY, CONF_EFFECT, CONF_FLASH_LENGTH, CONF_GREEN, \
|
||
|
|
CONF_ID, CONF_IF, CONF_LAMBDA, CONF_MAX, CONF_MIN, CONF_OR, CONF_PAYLOAD, CONF_QOS, \
|
||
|
|
CONF_RANGE, CONF_RED, CONF_RETAIN, 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, \
|
||
|
|
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'
|
||
|
|
|
||
|
|
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]
|
||
|
|
|
||
|
|
ACTIONS_SCHEMA = vol.All(cv.ensure_list, [vol.All({
|
||
|
|
cv.GenerateID('action', CONF_ACTION_ID): cv.register_variable_id,
|
||
|
|
vol.Optional(CONF_DELAY): 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): vol.Schema({
|
||
|
|
vol.Required(CONF_ID): cv.variable_id,
|
||
|
|
vol.Optional(CONF_TRANSITION_LENGTH): cv.templatable(cv.positive_time_period_milliseconds),
|
||
|
|
}),
|
||
|
|
vol.Optional(CONF_LIGHT_TURN_OFF): vol.Schema({
|
||
|
|
vol.Required(CONF_ID): cv.variable_id,
|
||
|
|
vol.Optional(CONF_TRANSITION_LENGTH): cv.templatable(cv.positive_time_period_milliseconds),
|
||
|
|
}),
|
||
|
|
vol.Optional(CONF_LIGHT_TURN_ON): vol.Schema({
|
||
|
|
vol.Required(CONF_ID): cv.variable_id,
|
||
|
|
vol.Optional(CONF_TRANSITION_LENGTH): cv.templatable(cv.positive_time_period_milliseconds),
|
||
|
|
vol.Optional(CONF_FLASH_LENGTH): cv.templatable(cv.positive_time_period_milliseconds),
|
||
|
|
vol.Optional(CONF_BRIGHTNESS): cv.templatable(cv.zero_to_one_float),
|
||
|
|
vol.Optional(CONF_RED): cv.templatable(cv.zero_to_one_float),
|
||
|
|
vol.Optional(CONF_GREEN): cv.templatable(cv.zero_to_one_float),
|
||
|
|
vol.Optional(CONF_BLUE): cv.templatable(cv.zero_to_one_float),
|
||
|
|
vol.Optional(CONF_WHITE): cv.templatable(cv.zero_to_one_float),
|
||
|
|
vol.Optional(CONF_EFFECT): cv.templatable(cv.string),
|
||
|
|
}),
|
||
|
|
vol.Optional(CONF_SWITCH_TOGGLE): vol.Schema({
|
||
|
|
vol.Required(CONF_ID): cv.variable_id,
|
||
|
|
}),
|
||
|
|
vol.Optional(CONF_SWITCH_TURN_OFF): vol.Schema({
|
||
|
|
vol.Required(CONF_ID): cv.variable_id,
|
||
|
|
}),
|
||
|
|
vol.Optional(CONF_SWITCH_TURN_ON): vol.Schema({
|
||
|
|
vol.Required(CONF_ID): cv.variable_id,
|
||
|
|
}),
|
||
|
|
vol.Optional(CONF_LAMBDA): cv.lambda_,
|
||
|
|
}, cv.has_at_exactly_one_key(*ACTION_KEYS))])
|
||
|
|
|
||
|
|
# pylint: disable=invalid-name
|
||
|
|
DelayAction = esphomelib_ns.DelayAction
|
||
|
|
LambdaAction = esphomelib_ns.LambdaAction
|
||
|
|
Automation = esphomelib_ns.Automation
|
||
|
|
|
||
|
|
|
||
|
|
def validate_recursive_condition(value):
|
||
|
|
return CONDITIONS_SCHEMA(value)
|
||
|
|
|
||
|
|
|
||
|
|
CONDITION_KEYS = [CONF_AND, CONF_OR, CONF_RANGE, CONF_LAMBDA]
|
||
|
|
|
||
|
|
CONDITIONS_SCHEMA = vol.All(cv.ensure_list, [vol.All({
|
||
|
|
cv.GenerateID('condition', CONF_CONDITION_ID): cv.register_variable_id,
|
||
|
|
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_MIN): vol.Coerce(float),
|
||
|
|
vol.Optional(CONF_MAX): vol.Coerce(float),
|
||
|
|
}), cv.has_at_least_one_key(CONF_MIN, CONF_MAX)),
|
||
|
|
vol.Optional(CONF_LAMBDA): cv.lambda_,
|
||
|
|
}), cv.has_at_exactly_one_key(*CONDITION_KEYS)])
|
||
|
|
|
||
|
|
# pylint: disable=invalid-name
|
||
|
|
AndCondition = esphomelib_ns.AndCondition
|
||
|
|
OrCondition = esphomelib_ns.OrCondition
|
||
|
|
RangeCondition = esphomelib_ns.RangeCondition
|
||
|
|
LambdaCondition = esphomelib_ns.LambdaCondition
|
||
|
|
|
||
|
|
AUTOMATION_SCHEMA = vol.Schema({
|
||
|
|
cv.GenerateID('trigger', CONF_TRIGGER_ID): cv.register_variable_id,
|
||
|
|
cv.GenerateID('automation', CONF_AUTOMATION_ID): cv.register_variable_id,
|
||
|
|
vol.Optional(CONF_IF): CONDITIONS_SCHEMA,
|
||
|
|
vol.Required(CONF_THEN): ACTIONS_SCHEMA,
|
||
|
|
})
|
||
|
|
|
||
|
|
|
||
|
|
def build_condition(config, arg_type):
|
||
|
|
template_arg = TemplateArguments(arg_type)
|
||
|
|
if CONF_AND in config:
|
||
|
|
return AndCondition.new(template_arg, build_conditions(config[CONF_AND], template_arg))
|
||
|
|
if CONF_OR in config:
|
||
|
|
return OrCondition.new(template_arg, build_conditions(config[CONF_OR], template_arg))
|
||
|
|
if CONF_LAMBDA in config:
|
||
|
|
return LambdaCondition.new(template_arg,
|
||
|
|
process_lambda(config[CONF_LAMBDA], [(arg_type, 'x')]))
|
||
|
|
if CONF_RANGE in config:
|
||
|
|
conf = config[CONF_RANGE]
|
||
|
|
rhs = RangeCondition.new(template_arg)
|
||
|
|
condition = Pvariable(RangeCondition.template(template_arg), config[CONF_CONDITION_ID], rhs)
|
||
|
|
if CONF_MIN in conf:
|
||
|
|
condition.set_min(templatable(conf[CONF_MIN], arg_type, float_))
|
||
|
|
if CONF_MAX in conf:
|
||
|
|
condition.set_max(templatable(conf[CONF_MAX], arg_type, float_))
|
||
|
|
return condition
|
||
|
|
raise ESPHomeYAMLError(u"Unsupported condition {}".format(config))
|
||
|
|
|
||
|
|
|
||
|
|
def build_conditions(config, arg_type):
|
||
|
|
return ArrayInitializer(*[build_condition(x, arg_type) for x in config])
|
||
|
|
|
||
|
|
|
||
|
|
def build_action(config, arg_type):
|
||
|
|
from esphomeyaml.components import light, mqtt, switch
|
||
|
|
|
||
|
|
template_arg = TemplateArguments(arg_type)
|
||
|
|
if CONF_DELAY in config:
|
||
|
|
rhs = App.register_component(DelayAction.new(template_arg))
|
||
|
|
action = Pvariable(DelayAction.template(template_arg), config[CONF_ACTION_ID], rhs)
|
||
|
|
add(action.set_delay(templatable(config[CONF_DELAY], arg_type, uint32)))
|
||
|
|
return action
|
||
|
|
elif CONF_LAMBDA in config:
|
||
|
|
rhs = LambdaAction.new(template_arg, process_lambda(config[CONF_LAMBDA], [(arg_type, 'x')]))
|
||
|
|
return Pvariable(LambdaAction.template(template_arg), config[CONF_ACTION_ID], rhs)
|
||
|
|
elif CONF_MQTT_PUBLISH in config:
|
||
|
|
conf = config[CONF_MQTT_PUBLISH]
|
||
|
|
rhs = App.Pget_mqtt_client().Pmake_publish_action()
|
||
|
|
action = Pvariable(mqtt.MQTTPublishAction.template(template_arg), config[CONF_ACTION_ID],
|
||
|
|
rhs)
|
||
|
|
add(action.set_topic(templatable(conf[CONF_TOPIC], arg_type, std_string)))
|
||
|
|
add(action.set_payload(templatable(conf[CONF_PAYLOAD], arg_type, std_string)))
|
||
|
|
if CONF_QOS in conf:
|
||
|
|
add(action.set_qos(templatable(conf[CONF_QOS], arg_type, uint8)))
|
||
|
|
if CONF_RETAIN in conf:
|
||
|
|
add(action.set_retain(templatable(conf[CONF_RETAIN], arg_type, bool_)))
|
||
|
|
return action
|
||
|
|
elif CONF_LIGHT_TOGGLE in config:
|
||
|
|
conf = config[CONF_LIGHT_TOGGLE]
|
||
|
|
var = get_variable(conf[CONF_ID])
|
||
|
|
rhs = var.make_toggle_action(template_arg)
|
||
|
|
action = Pvariable(light.ToggleAction.template(template_arg), config[CONF_ACTION_ID], rhs)
|
||
|
|
if CONF_TRANSITION_LENGTH in conf:
|
||
|
|
add(action.set_transition_length(
|
||
|
|
templatable(conf[CONF_TRANSITION_LENGTH], arg_type, uint32)
|
||
|
|
))
|
||
|
|
return action
|
||
|
|
elif CONF_LIGHT_TURN_OFF in config:
|
||
|
|
conf = config[CONF_LIGHT_TURN_OFF]
|
||
|
|
var = get_variable(conf[CONF_ID])
|
||
|
|
rhs = var.make_turn_off_action(template_arg)
|
||
|
|
action = Pvariable(light.TurnOffAction.template(template_arg), config[CONF_ACTION_ID], rhs)
|
||
|
|
if CONF_TRANSITION_LENGTH in conf:
|
||
|
|
add(action.set_transition_length(
|
||
|
|
templatable(conf[CONF_TRANSITION_LENGTH], arg_type, uint32)
|
||
|
|
))
|
||
|
|
return action
|
||
|
|
elif CONF_LIGHT_TURN_ON in config:
|
||
|
|
conf = config[CONF_LIGHT_TURN_ON]
|
||
|
|
var = get_variable(conf[CONF_ID])
|
||
|
|
rhs = var.make_turn_on_action(template_arg)
|
||
|
|
action = Pvariable(light.TurnOnAction.template(template_arg), config[CONF_ACTION_ID], rhs)
|
||
|
|
if CONF_TRANSITION_LENGTH in conf:
|
||
|
|
add(action.set_transition_length(
|
||
|
|
templatable(conf[CONF_TRANSITION_LENGTH], arg_type, uint32)
|
||
|
|
))
|
||
|
|
if CONF_FLASH_LENGTH in conf:
|
||
|
|
add(action.set_flash_length(templatable(conf[CONF_FLASH_LENGTH], arg_type, uint32)))
|
||
|
|
if CONF_BRIGHTNESS in conf:
|
||
|
|
add(action.set_brightness(templatable(conf[CONF_BRIGHTNESS], arg_type, float_)))
|
||
|
|
if CONF_RED in conf:
|
||
|
|
add(action.set_red(templatable(conf[CONF_RED], arg_type, float_)))
|
||
|
|
if CONF_GREEN in conf:
|
||
|
|
add(action.set_green(templatable(conf[CONF_GREEN], arg_type, float_)))
|
||
|
|
if CONF_BLUE in conf:
|
||
|
|
add(action.set_blue(templatable(conf[CONF_BLUE], arg_type, float_)))
|
||
|
|
if CONF_WHITE in conf:
|
||
|
|
add(action.set_white(templatable(conf[CONF_WHITE], arg_type, float_)))
|
||
|
|
if CONF_EFFECT in conf:
|
||
|
|
add(action.set_effect(templatable(conf[CONF_EFFECT], arg_type, std_string)))
|
||
|
|
return action
|
||
|
|
elif CONF_SWITCH_TOGGLE in config:
|
||
|
|
conf = config[CONF_SWITCH_TOGGLE]
|
||
|
|
var = get_variable(conf[CONF_ID])
|
||
|
|
rhs = var.make_toggle_action(template_arg)
|
||
|
|
return Pvariable(switch.ToggleAction.template(arg_type), config[CONF_ACTION_ID], rhs)
|
||
|
|
elif CONF_SWITCH_TURN_OFF in config:
|
||
|
|
conf = config[CONF_SWITCH_TURN_OFF]
|
||
|
|
var = get_variable(conf[CONF_ID])
|
||
|
|
rhs = var.make_turn_off_action(template_arg)
|
||
|
|
return Pvariable(switch.TurnOffAction.template(arg_type), config[CONF_ACTION_ID], rhs)
|
||
|
|
elif CONF_SWITCH_TURN_ON in config:
|
||
|
|
conf = config[CONF_SWITCH_TURN_ON]
|
||
|
|
var = get_variable(conf[CONF_ID])
|
||
|
|
rhs = var.make_turn_on_action(template_arg)
|
||
|
|
return Pvariable(switch.TurnOnAction.template(arg_type), config[CONF_ACTION_ID], rhs)
|
||
|
|
raise ESPHomeYAMLError(u"Unsupported action {}".format(config))
|
||
|
|
|
||
|
|
|
||
|
|
def build_actions(config, arg_type):
|
||
|
|
return ArrayInitializer(*[build_action(x, arg_type) for x in config])
|
||
|
|
|
||
|
|
|
||
|
|
def build_automation(trigger, arg_type, config):
|
||
|
|
rhs = App.make_automation(trigger)
|
||
|
|
obj = Pvariable(Automation.template(arg_type), config[CONF_AUTOMATION_ID], rhs)
|
||
|
|
if CONF_IF in config:
|
||
|
|
add(obj.add_conditions(build_conditions(config[CONF_IF], arg_type)))
|
||
|
|
add(obj.add_actions(build_actions(config[CONF_THEN], arg_type)))
|