Added Averager
This commit is contained in:
@@ -1,11 +1,13 @@
|
||||
import esphome.codegen as cg
|
||||
import esphome.config_validation as cv
|
||||
from esphome import automation
|
||||
from esphome.components import sensor
|
||||
from esphome.components.adc import sensor as adc
|
||||
from esphome.const import (
|
||||
CONF_ID,
|
||||
DEVICE_CLASS_VOLTAGE,
|
||||
STATE_CLASS_MEASUREMENT
|
||||
STATE_CLASS_MEASUREMENT,
|
||||
CONF_TRIGGER_ID
|
||||
)
|
||||
|
||||
chlorine_sensor_ns = cg.esphome_ns.namespace('analog_orp')
|
||||
@@ -15,11 +17,27 @@ CONF_SENSOR_PIN = "pin"
|
||||
CONF_INVERTED = "inverted"
|
||||
CONF_PRINT_RAW = "print_raw"
|
||||
CONF_ZERO_POINT = "zero_point"
|
||||
CONF_AVERAGE = "average"
|
||||
CONF_AVERAGE_MESUREMENTS="mesurements"
|
||||
CONF_SEND_STATE_EVERY="send_state_every"
|
||||
CONF_ON_READ_VALUE = "on_value_read"
|
||||
CONF_AVERAGE_ON_VALUE = "on_value"
|
||||
CONF_AVERAGE_NEW = "on_completely_new_value"
|
||||
|
||||
UNIT_MILLI_VOLT = "mV"
|
||||
|
||||
ICON_LIGHTNING_BOLT="mdi:lightning-bolt"
|
||||
|
||||
ChlorineValueRead = chlorine_sensor_ns.class_(
|
||||
"ChlorineValueReadTrigger", automation.Trigger.template(cg.float_)
|
||||
)
|
||||
ChlorineAverageChange = chlorine_sensor_ns.class_(
|
||||
"ChlorineAverageChangeTrigger", automation.Trigger.template(cg.float_)
|
||||
)
|
||||
ChlorineNewValue = chlorine_sensor_ns.class_(
|
||||
"ChlorineNewValueTrigger", automation.Trigger.template(cg.float_)
|
||||
)
|
||||
|
||||
CONFIG_SCHEMA = cv.All(
|
||||
sensor.sensor_schema(
|
||||
ChlorineSensor,
|
||||
@@ -32,19 +50,50 @@ CONFIG_SCHEMA = cv.All(
|
||||
cv.Required(CONF_SENSOR_PIN): adc.validate_adc_pin,
|
||||
cv.Required(CONF_ZERO_POINT): cv.Range(min=0, max=1023),
|
||||
cv.Optional(CONF_INVERTED, False): cv.boolean,
|
||||
cv.Optional(CONF_PRINT_RAW, False): cv.boolean
|
||||
cv.Optional(CONF_PRINT_RAW, False): cv.boolean,
|
||||
cv.Optional(CONF_AVERAGE): cv.Schema({
|
||||
cv.Required(CONF_AVERAGE_MESUREMENTS): cv.int_range(min=1),
|
||||
cv.Optional(CONF_SEND_STATE_EVERY): cv.int_range(min=1),
|
||||
cv.Optional(CONF_ON_READ_VALUE): automation.validate_automation({
|
||||
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(ChlorineValueRead),
|
||||
}),
|
||||
cv.Optional(CONF_AVERAGE_ON_VALUE): automation.validate_automation({
|
||||
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(ChlorineAverageChange),
|
||||
}),
|
||||
cv.Optional(CONF_AVERAGE_NEW): automation.validate_automation({
|
||||
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(ChlorineNewValue),
|
||||
}),
|
||||
}),
|
||||
}
|
||||
).extend(cv.polling_component_schema("60s"))
|
||||
)
|
||||
|
||||
def to_code(config):
|
||||
async def to_code(config):
|
||||
var = cg.new_Pvariable(config[CONF_ID])
|
||||
yield cg.register_component(var, config)
|
||||
yield sensor.register_sensor(var, config)
|
||||
await cg.register_component(var, config)
|
||||
await sensor.register_sensor(var, config)
|
||||
|
||||
sensor_pin = yield cg.gpio_pin_expression(config[CONF_SENSOR_PIN])
|
||||
sensor_pin = await cg.gpio_pin_expression(config[CONF_SENSOR_PIN])
|
||||
cg.add(var.set_pin(sensor_pin))
|
||||
|
||||
cg.add(var.set_inverted(config[CONF_INVERTED]))
|
||||
cg.add(var.set_print_raw(config[CONF_PRINT_RAW]))
|
||||
cg.add(var.set_zero_point(config[CONF_ZERO_POINT]))
|
||||
cg.add(var.set_zero_point(config[CONF_ZERO_POINT]))
|
||||
|
||||
if averager_config := config.get(CONF_AVERAGE):
|
||||
update_each = -1
|
||||
if CONF_SEND_STATE_EVERY in averager_config:
|
||||
update_each = averager_config[CONF_SEND_STATE_EVERY]
|
||||
cg.add(var.init_averager(averager_config[CONF_AVERAGE_MESUREMENTS], update_each))
|
||||
|
||||
for conf in averager_config.get(CONF_ON_READ_VALUE, []):
|
||||
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
||||
await automation.build_automation(trigger, [(float, "value")], conf)
|
||||
|
||||
for conf in averager_config.get(CONF_AVERAGE_ON_VALUE, []):
|
||||
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
||||
await automation.build_automation(trigger, [(float, "value")], conf)
|
||||
|
||||
for conf in averager_config.get(CONF_AVERAGE_NEW, []):
|
||||
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
||||
await automation.build_automation(trigger, [(float, "value")], conf)
|
Reference in New Issue
Block a user