1
0
mirror of https://github.com/esphome/esphome.git synced 2025-09-12 08:12:22 +01:00
* Add black

Update pre commit

Update pre commit

add empty line

* Format with black
This commit is contained in:
Guillermo Ruffino
2021-03-07 16:03:16 -03:00
committed by GitHub
parent 2b60b0f1fa
commit 69879920eb
398 changed files with 21624 additions and 12644 deletions

View File

@@ -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)