128 lines
4.8 KiB
Python
128 lines
4.8 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome import automation
|
|
from esphome.components import output
|
|
from esphome.components.analog_orp.sensor import ChlorineSensor
|
|
from esphome.const import (
|
|
CONF_ID,
|
|
CONF_TRIGGER_ID
|
|
)
|
|
|
|
|
|
DEPENDENCIES=["sensor"]
|
|
|
|
chlorine_pump_ns = cg.esphome_ns.namespace('chlorine_pump')
|
|
ChlorinePump = chlorine_pump_ns.class_('ChlorinePump', cg.Component)
|
|
|
|
CONF_SENSOR = "sensor"
|
|
CONF_PUMP_OUT = "pump"
|
|
CONF_PUMP_CYCLE_TIME = "cycle_time"
|
|
CONF_PUMP_PROPORTIONAL_BAND = "proportional_band"
|
|
CONF_PUMP_VALUE = "on_pump_value"
|
|
CONF_CYCLE_START = "on_cycle_start"
|
|
CONF_DISABLE_CLOCK="disable_clock"
|
|
CONF_TARGET="target"
|
|
|
|
def to_proportional_band(value):
|
|
try:
|
|
value = int(value)
|
|
except (TypeError, ValueError):
|
|
# pylint: disable=raise-missing-from
|
|
raise cv.Invalid(f"")
|
|
return cv.one_of(20, 50, 100, 150, 200, 250)(value)
|
|
|
|
ChlorSensorOutputTrigger = chlorine_pump_ns.class_(
|
|
"ChlorSensorOutputTrigger", automation.Trigger.template(cg.bool_, cg.int_)
|
|
)
|
|
ChlorSensorCycleTrigger = chlorine_pump_ns.class_(
|
|
"ChlorSensorCycleTrigger", automation.Trigger.template(cg.int_, cg.int_)
|
|
)
|
|
|
|
PrimeAction = chlorine_pump_ns.class_("ChlorinePrime", automation.Action)
|
|
StartAction = chlorine_pump_ns.class_("ChlorineStart", automation.Action)
|
|
StopAction = chlorine_pump_ns.class_("ChlorineStop", automation.Action)
|
|
SetTargetAction = chlorine_pump_ns.class_("ChlorineSetTarget", automation.Action)
|
|
|
|
CONFIG_SCHEMA = cv.COMPONENT_SCHEMA.extend({
|
|
cv.GenerateID(): cv.declare_id(ChlorinePump),
|
|
cv.Required(CONF_PUMP_OUT): cv.use_id(output.BinaryOutput),
|
|
cv.Optional(CONF_SENSOR): cv.use_id(ChlorineSensor),
|
|
cv.Optional(CONF_PUMP_CYCLE_TIME, default=360): cv.int_range(min=30, max=1400),
|
|
cv.Optional(CONF_PUMP_PROPORTIONAL_BAND, default=200): to_proportional_band,
|
|
cv.Optional(CONF_DISABLE_CLOCK, default=False): cv.boolean,
|
|
cv.Optional(CONF_TARGET, 700): cv.int_range(300, 1400),
|
|
cv.Optional(CONF_PUMP_VALUE): automation.validate_automation({
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(ChlorSensorOutputTrigger),
|
|
}
|
|
),
|
|
cv.Optional(CONF_CYCLE_START): automation.validate_automation({
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(ChlorSensorCycleTrigger),
|
|
}
|
|
),
|
|
})
|
|
|
|
|
|
|
|
ACTION_SCHEMA = cv.Schema({
|
|
cv.GenerateID(): cv.use_id(ChlorinePump)
|
|
})
|
|
|
|
@automation.register_action("chlorine_pump.prime", PrimeAction, ACTION_SCHEMA)
|
|
async def prime_action_to_code(config, action_id, template_arg, args):
|
|
paren = await cg.get_variable(config[CONF_ID])
|
|
return cg.new_Pvariable(action_id, template_arg, paren)
|
|
|
|
@automation.register_action("chlorine_pump.start", StartAction, ACTION_SCHEMA)
|
|
async def start_action_to_code(config, action_id, template_arg, args):
|
|
paren = await cg.get_variable(config[CONF_ID])
|
|
return cg.new_Pvariable(action_id, template_arg, paren)
|
|
|
|
@automation.register_action("chlorine_pump.stop", StopAction, ACTION_SCHEMA)
|
|
async def stop_action_to_code(config, action_id, template_arg, args):
|
|
paren = await cg.get_variable(config[CONF_ID])
|
|
return cg.new_Pvariable(action_id, template_arg, paren)
|
|
|
|
@automation.register_action("chlorine_pump.set_target", SetTargetAction, ACTION_SCHEMA.extend({
|
|
cv.Required(CONF_TARGET): cv.templatable(cv.float_),
|
|
}))
|
|
async def number_set_to_code(config, action_id, template_arg, args):
|
|
paren = await cg.get_variable(config[CONF_ID])
|
|
var = cg.new_Pvariable(action_id, template_arg, paren)
|
|
template_ = await cg.templatable(config[CONF_TARGET], args, float)
|
|
cg.add(var.set_value(template_))
|
|
return var
|
|
|
|
async def to_code(config):
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
#yield mqtt.register_mqtt_component(var, config)
|
|
|
|
#btnReset = yield cg.gpio_pin_expression(config[CONF_SENSOR_PIN])
|
|
#cg.add(var.set_input_pin(btnReset))
|
|
|
|
|
|
sensor = await cg.get_variable(config[CONF_SENSOR])
|
|
cg.add(var.set_sensor(sensor))
|
|
|
|
pump_out = await cg.get_variable(config[CONF_PUMP_OUT])
|
|
cg.add(var.set_pump_out(pump_out))
|
|
|
|
cg.add(var.set_cycle_time(config[CONF_PUMP_CYCLE_TIME])) # Needs be configured before sensor
|
|
|
|
cg.add(var.disable_clock(config[CONF_DISABLE_CLOCK]))
|
|
#cycle_time = yield cg.get_variable(config[CONF_PUMP_CYCLE_TIME])
|
|
|
|
#prop_band = yield cg.get_variable(config[CONF_PUMP_PROPORTIONAL_BAND])
|
|
cg.add(var.set_prop_band(config[CONF_PUMP_PROPORTIONAL_BAND]))
|
|
|
|
#cg.add(var.set_state(config[CONF_STATE]))
|
|
|
|
|
|
if CONF_PUMP_VALUE in config:
|
|
for conf in config.get(CONF_PUMP_VALUE, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [(bool, "pState"), (int, "pTime")], conf)
|
|
|
|
for conf in config.get(CONF_CYCLE_START, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [(int, "tOn"), (int, "tOff")], conf) |