Update __init__.py

This commit is contained in:
Jesse Hills 2024-06-06 18:37:35 +12:00 committed by GitHub
parent 460ce5991c
commit 385d5c018b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -23,6 +23,7 @@ CHANNELS = {
CONF_UNIVERSE = "universe" CONF_UNIVERSE = "universe"
CONF_E131_ID = "e131_id" CONF_E131_ID = "e131_id"
CONF_CHANNEL_OFFSET = "channel_offset"
CONFIG_SCHEMA = cv.Schema( CONFIG_SCHEMA = cv.Schema(
{ {
@ -46,6 +47,7 @@ async def to_code(config):
cv.GenerateID(CONF_E131_ID): cv.use_id(E131Component), cv.GenerateID(CONF_E131_ID): cv.use_id(E131Component),
cv.Required(CONF_UNIVERSE): cv.int_range(min=1, max=512), cv.Required(CONF_UNIVERSE): cv.int_range(min=1, max=512),
cv.Optional(CONF_CHANNELS, default="RGB"): cv.one_of(*CHANNELS, upper=True), cv.Optional(CONF_CHANNELS, default="RGB"): cv.one_of(*CHANNELS, upper=True),
cv.Optional(CONF_CHANNEL_OFFSET, default=0): cv.int_range(min=0, max=512),
}, },
) )
async def e131_light_effect_to_code(config, effect_id): async def e131_light_effect_to_code(config, effect_id):
@ -54,5 +56,6 @@ async def e131_light_effect_to_code(config, effect_id):
effect = cg.new_Pvariable(effect_id, config[CONF_NAME]) effect = cg.new_Pvariable(effect_id, config[CONF_NAME])
cg.add(effect.set_first_universe(config[CONF_UNIVERSE])) cg.add(effect.set_first_universe(config[CONF_UNIVERSE]))
cg.add(effect.set_channels(CHANNELS[config[CONF_CHANNELS]])) cg.add(effect.set_channels(CHANNELS[config[CONF_CHANNELS]]))
cg.add(effect.set_channel_offset(config[CONF_CHANNEL_OFFSET]))
cg.add(effect.set_e131(parent)) cg.add(effect.set_e131(parent))
return effect return effect