mirror of
https://github.com/esphome/esphome.git
synced 2025-09-12 08:12:22 +01:00
add-black (#1593)
* Add black Update pre commit Update pre commit add empty line * Format with black
This commit is contained in:
committed by
GitHub
parent
2b60b0f1fa
commit
69879920eb
@@ -3,34 +3,49 @@ 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)
|
||||
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("From ({}) must not be larger than to ({})"
|
||||
"".format(value[CONF_FROM], value[CONF_TO]))
|
||||
raise cv.Invalid(
|
||||
"From ({}) must not be larger than to ({})"
|
||||
"".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)),
|
||||
})
|
||||
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))
|
||||
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)
|
||||
|
Reference in New Issue
Block a user