mirror of
https://github.com/esphome/esphome.git
synced 2024-11-23 23:48:11 +01:00
96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome import automation
|
|
from esphome.components import key_provider
|
|
from esphome.const import (
|
|
CONF_ID,
|
|
CONF_MAX_LENGTH,
|
|
CONF_MIN_LENGTH,
|
|
CONF_ON_TIMEOUT,
|
|
CONF_SOURCE_ID,
|
|
CONF_TIMEOUT,
|
|
)
|
|
|
|
CODEOWNERS = ["@ssieb"]
|
|
|
|
MULTI_CONF = True
|
|
|
|
AUTO_LOAD = ["key_provider"]
|
|
|
|
CONF_START_KEYS = "start_keys"
|
|
CONF_END_KEYS = "end_keys"
|
|
CONF_END_KEY_REQUIRED = "end_key_required"
|
|
CONF_BACK_KEYS = "back_keys"
|
|
CONF_CLEAR_KEYS = "clear_keys"
|
|
CONF_ALLOWED_KEYS = "allowed_keys"
|
|
CONF_ON_PROGRESS = "on_progress"
|
|
CONF_ON_RESULT = "on_result"
|
|
|
|
key_collector_ns = cg.esphome_ns.namespace("key_collector")
|
|
KeyCollector = key_collector_ns.class_("KeyCollector", cg.Component)
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
cv.COMPONENT_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(KeyCollector),
|
|
cv.Optional(CONF_SOURCE_ID): cv.use_id(key_provider.KeyProvider),
|
|
cv.Optional(CONF_MIN_LENGTH): cv.int_,
|
|
cv.Optional(CONF_MAX_LENGTH): cv.int_,
|
|
cv.Optional(CONF_START_KEYS): cv.string,
|
|
cv.Optional(CONF_END_KEYS): cv.string,
|
|
cv.Optional(CONF_END_KEY_REQUIRED): cv.boolean,
|
|
cv.Optional(CONF_BACK_KEYS): cv.string,
|
|
cv.Optional(CONF_CLEAR_KEYS): cv.string,
|
|
cv.Optional(CONF_ALLOWED_KEYS): cv.string,
|
|
cv.Optional(CONF_ON_PROGRESS): automation.validate_automation(single=True),
|
|
cv.Optional(CONF_ON_RESULT): automation.validate_automation(single=True),
|
|
cv.Optional(CONF_ON_TIMEOUT): automation.validate_automation(single=True),
|
|
cv.Optional(CONF_TIMEOUT): cv.positive_time_period_milliseconds,
|
|
}
|
|
),
|
|
cv.has_at_least_one_key(CONF_END_KEYS, CONF_MAX_LENGTH),
|
|
)
|
|
|
|
|
|
async def to_code(config):
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
if CONF_SOURCE_ID in config:
|
|
source = await cg.get_variable(config[CONF_SOURCE_ID])
|
|
cg.add(var.set_provider(source))
|
|
if CONF_MIN_LENGTH in config:
|
|
cg.add(var.set_min_length(config[CONF_MIN_LENGTH]))
|
|
if CONF_MAX_LENGTH in config:
|
|
cg.add(var.set_max_length(config[CONF_MAX_LENGTH]))
|
|
if CONF_START_KEYS in config:
|
|
cg.add(var.set_start_keys(config[CONF_START_KEYS]))
|
|
if CONF_END_KEYS in config:
|
|
cg.add(var.set_end_keys(config[CONF_END_KEYS]))
|
|
if CONF_END_KEY_REQUIRED in config:
|
|
cg.add(var.set_end_key_required(config[CONF_END_KEY_REQUIRED]))
|
|
if CONF_BACK_KEYS in config:
|
|
cg.add(var.set_back_keys(config[CONF_BACK_KEYS]))
|
|
if CONF_CLEAR_KEYS in config:
|
|
cg.add(var.set_clear_keys(config[CONF_CLEAR_KEYS]))
|
|
if CONF_ALLOWED_KEYS in config:
|
|
cg.add(var.set_allowed_keys(config[CONF_ALLOWED_KEYS]))
|
|
if CONF_ON_PROGRESS in config:
|
|
await automation.build_automation(
|
|
var.get_progress_trigger(),
|
|
[(cg.std_string, "x"), (cg.uint8, "start")],
|
|
config[CONF_ON_PROGRESS],
|
|
)
|
|
if CONF_ON_RESULT in config:
|
|
await automation.build_automation(
|
|
var.get_result_trigger(),
|
|
[(cg.std_string, "x"), (cg.uint8, "start"), (cg.uint8, "end")],
|
|
config[CONF_ON_RESULT],
|
|
)
|
|
if CONF_ON_TIMEOUT in config:
|
|
await automation.build_automation(
|
|
var.get_timeout_trigger(),
|
|
[(cg.std_string, "x"), (cg.uint8, "start")],
|
|
config[CONF_ON_TIMEOUT],
|
|
)
|
|
if CONF_TIMEOUT in config:
|
|
cg.add(var.set_timeout(config[CONF_TIMEOUT]))
|