72 lines
3.0 KiB
Python
72 lines
3.0 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_SOURCE_ID, CONF_TIMEOUT
|
|
|
|
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'
|
|
|
|
AUTO_LOAD = ['key_provider']
|
|
|
|
MULTI_CONF = True
|
|
|
|
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.GenerateID(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_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)
|
|
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_TIMEOUT in config:
|
|
cg.add(var.set_timeout(config[CONF_TIMEOUT]))
|
|
|