mirror of
https://github.com/esphome/esphome.git
synced 2024-09-21 10:01:19 +02:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import light
|
|
from esphome.const import CONF_FROM, CONF_ID, CONF_SEGMENTS, CONF_TO, CONF_OUTPUT_ID
|
|
|
|
partitions_ns = cg.esphome_ns.namespace('partition')
|
|
AddressableSegment = partitions_ns.class_('AddressableSegment')
|
|
PartitionLightOutput = partitions_ns.class_('PartitionLightOutput', light.AddressableLight)
|
|
|
|
|
|
def validate_from_to(value):
|
|
if value[CONF_FROM] > value[CONF_TO]:
|
|
raise cv.Invalid(u"From ({}) must not be larger than to ({})"
|
|
u"".format(value[CONF_FROM], value[CONF_TO]))
|
|
return value
|
|
|
|
|
|
CONFIG_SCHEMA = light.ADDRESSABLE_LIGHT_SCHEMA.extend({
|
|
cv.GenerateID(CONF_OUTPUT_ID): cv.declare_id(PartitionLightOutput),
|
|
cv.Required(CONF_SEGMENTS): cv.All(cv.ensure_list({
|
|
cv.Required(CONF_ID): cv.use_id(light.AddressableLightState),
|
|
cv.Required(CONF_FROM): cv.positive_int,
|
|
cv.Required(CONF_TO): cv.positive_int,
|
|
}, validate_from_to), cv.Length(min=1)),
|
|
})
|
|
|
|
|
|
def to_code(config):
|
|
segments = []
|
|
for conf in config[CONF_SEGMENTS]:
|
|
var = yield cg.get_variable(conf[CONF_ID])
|
|
segments.append(AddressableSegment(var, conf[CONF_FROM],
|
|
conf[CONF_TO] - conf[CONF_FROM] + 1))
|
|
|
|
var = cg.new_Pvariable(config[CONF_OUTPUT_ID], segments)
|
|
yield cg.register_component(var, config)
|
|
yield light.register_light(var, config)
|