148 lines
6.0 KiB
Python
148 lines
6.0 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
import voluptuous as vol
|
|
from esphome import automation, pins
|
|
from esphome.const import CONF_ID, CONF_TRIGGER_ID
|
|
from esphome.components import binary_sensor
|
|
|
|
DEPENDENCIES = ["preferences"]
|
|
MULTI_CONF = True
|
|
|
|
|
|
ratgdo_ns = cg.esphome_ns.namespace("ratgdo")
|
|
RATGDO = ratgdo_ns.class_("RATGDOComponent", cg.Component)
|
|
|
|
|
|
SyncFailed = ratgdo_ns.class_("SyncFailed", automation.Trigger.template())
|
|
|
|
CONF_OUTPUT_GDO = "output_gdo_pin"
|
|
DEFAULT_OUTPUT_GDO = (
|
|
"D4" # D4 red control terminal / GarageDoorOpener (UART1 TX) pin is D4 on D1 Mini
|
|
)
|
|
CONF_INPUT_GDO = "input_gdo_pin"
|
|
DEFAULT_INPUT_GDO = (
|
|
"D2" # D2 red control terminal / GarageDoorOpener (UART1 RX) pin is D2 on D1 Mini
|
|
)
|
|
CONF_INPUT_OBST = "input_obst_pin"
|
|
DEFAULT_INPUT_OBST = "D7" # D7 black obstruction sensor terminal
|
|
|
|
CONF_DISCRETE_OPEN_PIN = "discrete_open_pin"
|
|
CONF_DISCRETE_CLOSE_PIN = "discrete_close_pin"
|
|
|
|
CONF_RATGDO_ID = "ratgdo_id"
|
|
|
|
CONF_ON_SYNC_FAILED = "on_sync_failed"
|
|
|
|
CONF_PROTOCOL = "protocol"
|
|
|
|
PROTOCOL_SECPLUSV1 = "secplusv1"
|
|
PROTOCOL_SECPLUSV2 = "secplusv2"
|
|
PROTOCOL_DRYCONTACT = "drycontact"
|
|
SUPPORTED_PROTOCOLS = [PROTOCOL_SECPLUSV1, PROTOCOL_SECPLUSV2, PROTOCOL_DRYCONTACT]
|
|
|
|
CONF_DRY_CONTACT_OPEN_SENSOR = "dry_contact_open_sensor"
|
|
CONF_DRY_CONTACT_CLOSE_SENSOR = "dry_contact_close_sensor"
|
|
CONF_DRY_CONTACT_SENSOR_GROUP = "dry_contact_sensor_group"
|
|
|
|
def validate_protocol(config):
|
|
print("Validation")
|
|
if config.get(CONF_PROTOCOL, None) == PROTOCOL_DRYCONTACT and (CONF_DRY_CONTACT_CLOSE_SENSOR not in config or CONF_DRY_CONTACT_OPEN_SENSOR not in config):
|
|
raise cv.Invalid("dry_contact_close_sensor and dry_contact_open_sensor are required when using protocol drycontact")
|
|
if config.get(CONF_PROTOCOL, None) != PROTOCOL_DRYCONTACT and (CONF_DRY_CONTACT_CLOSE_SENSOR in config or CONF_DRY_CONTACT_OPEN_SENSOR in config):
|
|
raise cv.Invalid("dry_contact_close_sensor and dry_contact_open_sensor are only valid when using protocol drycontact")
|
|
# if config.get(CONF_PROTOCOL, None) == PROTOCOL_DRYCONTACT and CONF_DRY_CONTACT_OPEN_SENSOR not in config:
|
|
# raise cv.Invalid("dry_contact_open_sensor is required when using protocol drycontact")
|
|
return config
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(RATGDO),
|
|
cv.Optional(
|
|
CONF_OUTPUT_GDO, default=DEFAULT_OUTPUT_GDO
|
|
): pins.gpio_output_pin_schema,
|
|
cv.Optional(
|
|
CONF_INPUT_GDO, default=DEFAULT_INPUT_GDO
|
|
): pins.gpio_input_pin_schema,
|
|
cv.Optional(CONF_INPUT_OBST, default=DEFAULT_INPUT_OBST): cv.Any(
|
|
cv.none, pins.gpio_input_pin_schema
|
|
),
|
|
cv.Optional(CONF_DISCRETE_OPEN_PIN): pins.gpio_output_pin_schema,
|
|
cv.Optional(CONF_DISCRETE_CLOSE_PIN): pins.gpio_output_pin_schema,
|
|
cv.Optional(CONF_ON_SYNC_FAILED): automation.validate_automation(
|
|
{
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(SyncFailed),
|
|
}
|
|
),
|
|
cv.Optional(CONF_PROTOCOL, default=PROTOCOL_SECPLUSV2): cv.All(vol.In(
|
|
SUPPORTED_PROTOCOLS
|
|
)),
|
|
# cv.Inclusive(CONF_DRY_CONTACT_OPEN_SENSOR,CONF_DRY_CONTACT_SENSOR_GROUP): cv.use_id(binary_sensor.BinarySensor),
|
|
# cv.Inclusive(CONF_DRY_CONTACT_CLOSE_SENSOR,CONF_DRY_CONTACT_SENSOR_GROUP): cv.use_id(binary_sensor.BinarySensor),
|
|
cv.Optional(CONF_DRY_CONTACT_OPEN_SENSOR): cv.use_id(binary_sensor.BinarySensor),
|
|
cv.Optional(CONF_DRY_CONTACT_CLOSE_SENSOR): cv.use_id(binary_sensor.BinarySensor),
|
|
}
|
|
).extend(cv.COMPONENT_SCHEMA),
|
|
validate_protocol,
|
|
)
|
|
|
|
RATGDO_CLIENT_SCHMEA = cv.Schema(
|
|
{
|
|
cv.Required(CONF_RATGDO_ID): cv.use_id(RATGDO),
|
|
}
|
|
)
|
|
|
|
|
|
async def register_ratgdo_child(var, config):
|
|
parent = await cg.get_variable(config[CONF_RATGDO_ID])
|
|
cg.add(var.set_parent(parent))
|
|
|
|
|
|
async def to_code(config):
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
pin = await cg.gpio_pin_expression(config[CONF_OUTPUT_GDO])
|
|
cg.add(var.set_output_gdo_pin(pin))
|
|
pin = await cg.gpio_pin_expression(config[CONF_INPUT_GDO])
|
|
cg.add(var.set_input_gdo_pin(pin))
|
|
if CONF_INPUT_OBST in config and config[CONF_INPUT_OBST]:
|
|
pin = await cg.gpio_pin_expression(config[CONF_INPUT_OBST])
|
|
cg.add(var.set_input_obst_pin(pin))
|
|
|
|
if CONF_DRY_CONTACT_OPEN_SENSOR in config and config[CONF_DRY_CONTACT_OPEN_SENSOR]:
|
|
dry_contact_open_sensor = await cg.get_variable(config[CONF_DRY_CONTACT_OPEN_SENSOR])
|
|
cg.add(var.set_dry_contact_open_sensor(dry_contact_open_sensor))
|
|
|
|
if CONF_DRY_CONTACT_CLOSE_SENSOR in config and config[CONF_DRY_CONTACT_CLOSE_SENSOR]:
|
|
dry_contact_close_sensor = await cg.get_variable(config[CONF_DRY_CONTACT_CLOSE_SENSOR])
|
|
cg.add(var.set_dry_contact_close_sensor(dry_contact_close_sensor))
|
|
|
|
for conf in config.get(CONF_ON_SYNC_FAILED, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [], conf)
|
|
|
|
cg.add_library(
|
|
name="secplus",
|
|
repository="https://github.com/ratgdo/secplus#f98c3220356c27717a25102c0b35815ebbd26ccc",
|
|
version=None,
|
|
)
|
|
cg.add_library(
|
|
name="espsoftwareserial",
|
|
repository="https://github.com/ratgdo/espsoftwareserial#autobaud",
|
|
version=None,
|
|
)
|
|
|
|
if config[CONF_PROTOCOL] == PROTOCOL_SECPLUSV1:
|
|
cg.add_define("PROTOCOL_SECPLUSV1")
|
|
elif config[CONF_PROTOCOL] == PROTOCOL_SECPLUSV2:
|
|
cg.add_define("PROTOCOL_SECPLUSV2")
|
|
elif config[CONF_PROTOCOL] == PROTOCOL_DRYCONTACT:
|
|
cg.add_define("PROTOCOL_DRYCONTACT")
|
|
cg.add(var.init_protocol())
|
|
|
|
if CONF_DISCRETE_OPEN_PIN in config and config[CONF_DISCRETE_OPEN_PIN]:
|
|
pin = await cg.gpio_pin_expression(config[CONF_DISCRETE_OPEN_PIN])
|
|
cg.add(var.set_discrete_open_pin(pin))
|
|
if CONF_DISCRETE_CLOSE_PIN in config and config[CONF_DISCRETE_CLOSE_PIN]:
|
|
pin = await cg.gpio_pin_expression(config[CONF_DISCRETE_CLOSE_PIN])
|
|
cg.add(var.set_discrete_close_pin(pin)) |