mirror of https://github.com/esphome/esphome.git
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
import hashlib
|
|
|
|
from esphome import config_validation as cv, automation
|
|
from esphome import codegen as cg
|
|
from esphome.const import CONF_ID, CONF_INITIAL_VALUE, CONF_RESTORE_VALUE, CONF_TYPE, CONF_VALUE
|
|
from esphome.core import coroutine_with_priority
|
|
|
|
globals_ns = cg.esphome_ns.namespace('globals')
|
|
GlobalsComponent = globals_ns.class_('GlobalsComponent', cg.Component)
|
|
GlobalVarSetAction = globals_ns.class_('GlobalVarSetAction', automation.Action)
|
|
|
|
MULTI_CONF = True
|
|
CONFIG_SCHEMA = cv.Schema({
|
|
cv.Required(CONF_ID): cv.declare_id(GlobalsComponent),
|
|
cv.Required(CONF_TYPE): cv.string_strict,
|
|
cv.Optional(CONF_INITIAL_VALUE): cv.string_strict,
|
|
cv.Optional(CONF_RESTORE_VALUE, default=False): cv.boolean,
|
|
}).extend(cv.COMPONENT_SCHEMA)
|
|
|
|
|
|
# Run with low priority so that namespaces are registered first
|
|
@coroutine_with_priority(-100.0)
|
|
def to_code(config):
|
|
type_ = cg.RawExpression(config[CONF_TYPE])
|
|
template_args = cg.TemplateArguments(type_)
|
|
res_type = GlobalsComponent.template(template_args)
|
|
|
|
initial_value = None
|
|
if CONF_INITIAL_VALUE in config:
|
|
initial_value = cg.RawExpression(config[CONF_INITIAL_VALUE])
|
|
|
|
rhs = GlobalsComponent.new(template_args, initial_value)
|
|
glob = cg.Pvariable(config[CONF_ID], rhs, res_type)
|
|
yield cg.register_component(glob, config)
|
|
|
|
if config[CONF_RESTORE_VALUE]:
|
|
value = config[CONF_ID].id
|
|
if isinstance(value, str):
|
|
value = value.encode()
|
|
hash_ = int(hashlib.md5(value).hexdigest()[:8], 16)
|
|
cg.add(glob.set_restore_value(hash_))
|
|
|
|
|
|
@automation.register_action('globals.set', GlobalVarSetAction, cv.Schema({
|
|
cv.Required(CONF_ID): cv.use_id(GlobalsComponent),
|
|
cv.Required(CONF_VALUE): cv.templatable(cv.string_strict),
|
|
}))
|
|
def globals_set_to_code(config, action_id, template_arg, args):
|
|
full_id, paren = yield cg.get_variable_with_full_id(config[CONF_ID])
|
|
template_arg = cg.TemplateArguments(full_id.type, *template_arg)
|
|
var = cg.new_Pvariable(action_id, template_arg, paren)
|
|
templ = yield cg.templatable(config[CONF_VALUE], args, None,
|
|
to_exp=cg.RawExpression)
|
|
cg.add(var.set_value(templ))
|
|
yield var
|