mirror of https://github.com/esphome/esphome.git
205 lines
8.8 KiB
Python
205 lines
8.8 KiB
Python
import voluptuous as vol
|
|
|
|
import esphomeyaml.config_validation as cv
|
|
from esphomeyaml import automation
|
|
from esphomeyaml.const import CONF_ABOVE, CONF_ACCURACY_DECIMALS, CONF_ALPHA, CONF_BELOW, \
|
|
CONF_DEBOUNCE, CONF_DELTA, CONF_EXPIRE_AFTER, CONF_EXPONENTIAL_MOVING_AVERAGE, CONF_FILTERS, \
|
|
CONF_FILTER_NAN, CONF_FILTER_OUT, CONF_HEARTBEAT, CONF_ICON, CONF_ID, CONF_LAMBDA, \
|
|
CONF_MQTT_ID, CONF_MULTIPLY, CONF_NAME, CONF_OFFSET, CONF_ON_RAW_VALUE, CONF_ON_VALUE, \
|
|
CONF_ON_VALUE_RANGE, CONF_OR, CONF_SEND_EVERY, CONF_SLIDING_WINDOW_MOVING_AVERAGE, \
|
|
CONF_THROTTLE, CONF_TRIGGER_ID, CONF_UNIQUE, CONF_UNIT_OF_MEASUREMENT, CONF_WINDOW_SIZE, \
|
|
CONF_INTERNAL
|
|
from esphomeyaml.helpers import App, ArrayInitializer, Pvariable, add, esphomelib_ns, float_, \
|
|
process_lambda, setup_mqtt_component, templatable, add_job
|
|
|
|
PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend({
|
|
|
|
})
|
|
|
|
|
|
def validate_recursive_filter(value):
|
|
return FILTERS_SCHEMA(value)
|
|
|
|
|
|
FILTER_KEYS = [CONF_OFFSET, CONF_MULTIPLY, CONF_FILTER_OUT, CONF_FILTER_NAN,
|
|
CONF_SLIDING_WINDOW_MOVING_AVERAGE, CONF_EXPONENTIAL_MOVING_AVERAGE, CONF_LAMBDA,
|
|
CONF_THROTTLE, CONF_DELTA, CONF_UNIQUE, CONF_HEARTBEAT, CONF_DEBOUNCE, CONF_OR]
|
|
|
|
FILTERS_SCHEMA = vol.All(cv.ensure_list, [vol.All({
|
|
vol.Optional(CONF_OFFSET): vol.Coerce(float),
|
|
vol.Optional(CONF_MULTIPLY): vol.Coerce(float),
|
|
vol.Optional(CONF_FILTER_OUT): vol.Coerce(float),
|
|
vol.Optional(CONF_FILTER_NAN): None,
|
|
vol.Optional(CONF_SLIDING_WINDOW_MOVING_AVERAGE): vol.Schema({
|
|
vol.Required(CONF_WINDOW_SIZE): cv.positive_not_null_int,
|
|
vol.Required(CONF_SEND_EVERY): cv.positive_not_null_int,
|
|
}),
|
|
vol.Optional(CONF_EXPONENTIAL_MOVING_AVERAGE): vol.Schema({
|
|
vol.Required(CONF_ALPHA): cv.positive_float,
|
|
vol.Required(CONF_SEND_EVERY): cv.positive_not_null_int,
|
|
}),
|
|
vol.Optional(CONF_LAMBDA): cv.lambda_,
|
|
vol.Optional(CONF_THROTTLE): cv.positive_time_period_milliseconds,
|
|
vol.Optional(CONF_DELTA): vol.Coerce(float),
|
|
vol.Optional(CONF_UNIQUE): None,
|
|
vol.Optional(CONF_HEARTBEAT): cv.positive_time_period_milliseconds,
|
|
vol.Optional(CONF_DEBOUNCE): cv.positive_time_period_milliseconds,
|
|
vol.Optional(CONF_OR): validate_recursive_filter,
|
|
}, cv.has_exactly_one_key(*FILTER_KEYS))])
|
|
|
|
# pylint: disable=invalid-name
|
|
sensor_ns = esphomelib_ns.namespace('sensor')
|
|
Sensor = sensor_ns.Sensor
|
|
MQTTSensorComponent = sensor_ns.MQTTSensorComponent
|
|
OffsetFilter = sensor_ns.OffsetFilter
|
|
MultiplyFilter = sensor_ns.MultiplyFilter
|
|
FilterOutValueFilter = sensor_ns.FilterOutValueFilter
|
|
FilterOutNANFilter = sensor_ns.FilterOutNANFilter
|
|
SlidingWindowMovingAverageFilter = sensor_ns.SlidingWindowMovingAverageFilter
|
|
ExponentialMovingAverageFilter = sensor_ns.ExponentialMovingAverageFilter
|
|
LambdaFilter = sensor_ns.LambdaFilter
|
|
ThrottleFilter = sensor_ns.ThrottleFilter
|
|
DeltaFilter = sensor_ns.DeltaFilter
|
|
OrFilter = sensor_ns.OrFilter
|
|
HeartbeatFilter = sensor_ns.HeartbeatFilter
|
|
DebounceFilter = sensor_ns.DebounceFilter
|
|
UniqueFilter = sensor_ns.UniqueFilter
|
|
|
|
SensorValueTrigger = sensor_ns.SensorValueTrigger
|
|
RawSensorValueTrigger = sensor_ns.RawSensorValueTrigger
|
|
ValueRangeTrigger = sensor_ns.ValueRangeTrigger
|
|
|
|
SENSOR_SCHEMA = cv.MQTT_COMPONENT_SCHEMA.extend({
|
|
cv.GenerateID(CONF_MQTT_ID): cv.declare_variable_id(MQTTSensorComponent),
|
|
cv.GenerateID(): cv.declare_variable_id(Sensor),
|
|
vol.Required(CONF_NAME): cv.string,
|
|
vol.Optional(CONF_UNIT_OF_MEASUREMENT): cv.string_strict,
|
|
vol.Optional(CONF_ICON): cv.icon,
|
|
vol.Optional(CONF_ACCURACY_DECIMALS): vol.Coerce(int),
|
|
vol.Optional(CONF_EXPIRE_AFTER): vol.Any(None, cv.positive_time_period_milliseconds),
|
|
vol.Optional(CONF_FILTERS): FILTERS_SCHEMA,
|
|
vol.Optional(CONF_ON_VALUE): vol.All(cv.ensure_list, [automation.AUTOMATION_SCHEMA.extend({
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_variable_id(SensorValueTrigger),
|
|
})]),
|
|
vol.Optional(CONF_ON_RAW_VALUE): vol.All(cv.ensure_list, [automation.AUTOMATION_SCHEMA.extend({
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_variable_id(RawSensorValueTrigger),
|
|
})]),
|
|
vol.Optional(CONF_ON_VALUE_RANGE): vol.All(cv.ensure_list, [vol.All(
|
|
automation.AUTOMATION_SCHEMA.extend({
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_variable_id(ValueRangeTrigger),
|
|
vol.Optional(CONF_ABOVE): vol.Coerce(float),
|
|
vol.Optional(CONF_BELOW): vol.Coerce(float),
|
|
}), cv.has_at_least_one_key(CONF_ABOVE, CONF_BELOW))]),
|
|
})
|
|
|
|
SENSOR_PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(SENSOR_SCHEMA.schema)
|
|
|
|
|
|
def setup_filter(config):
|
|
if CONF_OFFSET in config:
|
|
yield OffsetFilter.new(config[CONF_OFFSET])
|
|
elif CONF_MULTIPLY in config:
|
|
yield MultiplyFilter.new(config[CONF_MULTIPLY])
|
|
elif CONF_FILTER_OUT in config:
|
|
yield FilterOutValueFilter.new(config[CONF_FILTER_OUT])
|
|
elif CONF_FILTER_NAN in config:
|
|
yield FilterOutNANFilter.new()
|
|
elif CONF_SLIDING_WINDOW_MOVING_AVERAGE in config:
|
|
conf = config[CONF_SLIDING_WINDOW_MOVING_AVERAGE]
|
|
yield SlidingWindowMovingAverageFilter.new(conf[CONF_WINDOW_SIZE], conf[CONF_SEND_EVERY])
|
|
elif CONF_EXPONENTIAL_MOVING_AVERAGE in config:
|
|
conf = config[CONF_EXPONENTIAL_MOVING_AVERAGE]
|
|
yield ExponentialMovingAverageFilter.new(conf[CONF_ALPHA], conf[CONF_SEND_EVERY])
|
|
elif CONF_LAMBDA in config:
|
|
lambda_ = None
|
|
for lambda_ in process_lambda(config[CONF_LAMBDA], [(float_, 'x')]):
|
|
yield None
|
|
yield LambdaFilter.new(lambda_)
|
|
elif CONF_THROTTLE in config:
|
|
yield ThrottleFilter.new(config[CONF_THROTTLE])
|
|
elif CONF_DELTA in config:
|
|
yield DeltaFilter.new(config[CONF_DELTA])
|
|
elif CONF_OR in config:
|
|
filters = None
|
|
for filters in setup_filters(config[CONF_OR]):
|
|
yield None
|
|
yield OrFilter.new(filters)
|
|
elif CONF_HEARTBEAT in config:
|
|
yield App.register_component(HeartbeatFilter.new(config[CONF_HEARTBEAT]))
|
|
elif CONF_DEBOUNCE in config:
|
|
yield App.register_component(DebounceFilter.new(config[CONF_DEBOUNCE]))
|
|
elif CONF_UNIQUE in config:
|
|
yield UniqueFilter.new()
|
|
|
|
|
|
def setup_filters(config):
|
|
filters = []
|
|
for conf in config:
|
|
filter = None
|
|
for filter in setup_filter(conf):
|
|
yield None
|
|
filters.append(filter)
|
|
yield ArrayInitializer(*filters)
|
|
|
|
|
|
def setup_sensor_core_(sensor_var, mqtt_var, config):
|
|
if CONF_INTERNAL in config:
|
|
add(sensor_var.set_internal(config[CONF_INTERNAL]))
|
|
if CONF_UNIT_OF_MEASUREMENT in config:
|
|
add(sensor_var.set_unit_of_measurement(config[CONF_UNIT_OF_MEASUREMENT]))
|
|
if CONF_ICON in config:
|
|
add(sensor_var.set_icon(config[CONF_ICON]))
|
|
if CONF_ACCURACY_DECIMALS in config:
|
|
add(sensor_var.set_accuracy_decimals(config[CONF_ACCURACY_DECIMALS]))
|
|
if CONF_FILTERS in config:
|
|
filters = None
|
|
for filters in setup_filters(config[CONF_FILTERS]):
|
|
yield
|
|
add(sensor_var.set_filters(filters))
|
|
|
|
for conf in config.get(CONF_ON_VALUE, []):
|
|
rhs = sensor_var.make_value_trigger()
|
|
trigger = Pvariable(conf[CONF_TRIGGER_ID], rhs)
|
|
automation.build_automation(trigger, float_, conf)
|
|
for conf in config.get(CONF_ON_RAW_VALUE, []):
|
|
rhs = sensor_var.make_raw_value_trigger()
|
|
trigger = Pvariable(conf[CONF_TRIGGER_ID], rhs)
|
|
automation.build_automation(trigger, float_, conf)
|
|
for conf in config.get(CONF_ON_VALUE_RANGE, []):
|
|
rhs = sensor_var.make_value_range_trigger()
|
|
trigger = Pvariable(conf[CONF_TRIGGER_ID], rhs)
|
|
if CONF_ABOVE in conf:
|
|
template_ = None
|
|
for template_ in templatable(conf[CONF_ABOVE], float_, float_):
|
|
yield
|
|
trigger.set_min(template_)
|
|
if CONF_BELOW in conf:
|
|
template_ = None
|
|
for template_ in templatable(conf[CONF_BELOW], float_, float_):
|
|
yield
|
|
trigger.set_max(template_)
|
|
automation.build_automation(trigger, float_, conf)
|
|
|
|
if CONF_EXPIRE_AFTER in config:
|
|
if config[CONF_EXPIRE_AFTER] is None:
|
|
add(mqtt_var.disable_expire_after())
|
|
else:
|
|
add(mqtt_var.set_expire_after(config[CONF_EXPIRE_AFTER]))
|
|
setup_mqtt_component(mqtt_var, config)
|
|
|
|
|
|
def setup_sensor(sensor_obj, mqtt_obj, config):
|
|
sensor_var = Pvariable(config[CONF_ID], sensor_obj, has_side_effects=False)
|
|
mqtt_var = Pvariable(config[CONF_MQTT_ID], mqtt_obj, has_side_effects=False)
|
|
add_job(setup_sensor_core_, sensor_var, mqtt_var, config)
|
|
|
|
|
|
def register_sensor(var, config):
|
|
sensor_var = Pvariable(config[CONF_ID], var, has_side_effects=True)
|
|
rhs = App.register_sensor(sensor_var)
|
|
mqtt_var = Pvariable(config[CONF_MQTT_ID], rhs, has_side_effects=True)
|
|
add_job(setup_sensor_core_, sensor_var, mqtt_var, config)
|
|
|
|
|
|
BUILD_FLAGS = '-DUSE_SENSOR'
|