jeans/esphome
jeans/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2025-08-07 10:57:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

[light] Allow light effect schema to be a schema object already (#10091)

Browse Source
This commit is contained in:
Jesse Hills 2025-08-06 17:05:48 +12:00 committed by GitHub
parent 99125c045f
commit d872c8a999
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 34 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
esphome/components/light/effects.py
View File

@ -29,6 +29,7 @@ from esphome.const import (
CONF_WHITE, CONF_WHITE,
CONF_WIDTH, CONF_WIDTH,
) )
from esphome.cpp_generator import MockObjClass
from esphome.schema_extractors import SCHEMA_EXTRACT, schema_extractor from esphome.schema_extractors import SCHEMA_EXTRACT, schema_extractor
from esphome.util import Registry from esphome.util import Registry
@ -88,8 +89,15 @@ ADDRESSABLE_EFFECTS = []
EFFECTS_REGISTRY = Registry() EFFECTS_REGISTRY = Registry()
def register_effect(name, effect_type, default_name, schema, *extra_validators): def register_effect(
schema = cv.Schema(schema).extend( name: str,
effect_type: MockObjClass,
default_name: str,
schema: cv.Schema | dict,
*extra_validators,
):
schema = schema if isinstance(schema, cv.Schema) else cv.Schema(schema)
schema = schema.extend(
{ {
cv.Optional(CONF_NAME, default=default_name): cv.string_strict, cv.Optional(CONF_NAME, default=default_name): cv.string_strict,
} }
@ -98,7 +106,13 @@ def register_effect(name, effect_type, default_name, schema, *extra_validators):
return EFFECTS_REGISTRY.register(name, effect_type, validator) return EFFECTS_REGISTRY.register(name, effect_type, validator)
def register_binary_effect(name, effect_type, default_name, schema, *extra_validators): def register_binary_effect(
name: str,
effect_type: MockObjClass,
default_name: str,
schema: cv.Schema | dict,
*extra_validators,
):
# binary effect can be used for all lights # binary effect can be used for all lights
BINARY_EFFECTS.append(name) BINARY_EFFECTS.append(name)
MONOCHROMATIC_EFFECTS.append(name) MONOCHROMATIC_EFFECTS.append(name)
@ -109,7 +123,11 @@ def register_binary_effect(name, effect_type, default_name, schema, *extra_valid
def register_monochromatic_effect( def register_monochromatic_effect(
name, effect_type, default_name, schema, *extra_validators name: str,
effect_type: MockObjClass,
default_name: str,
schema: cv.Schema | dict,
*extra_validators,
): ):
# monochromatic effect can be used for all lights expect binary # monochromatic effect can be used for all lights expect binary
MONOCHROMATIC_EFFECTS.append(name) MONOCHROMATIC_EFFECTS.append(name)
@ -119,7 +137,13 @@ def register_monochromatic_effect(
return register_effect(name, effect_type, default_name, schema, *extra_validators) return register_effect(name, effect_type, default_name, schema, *extra_validators)
def register_rgb_effect(name, effect_type, default_name, schema, *extra_validators): def register_rgb_effect(
name: str,
effect_type: MockObjClass,
default_name: str,
schema: cv.Schema | dict,
*extra_validators,
):
# RGB effect can be used for RGB and addressable lights # RGB effect can be used for RGB and addressable lights
RGB_EFFECTS.append(name) RGB_EFFECTS.append(name)
ADDRESSABLE_EFFECTS.append(name) ADDRESSABLE_EFFECTS.append(name)
@ -128,7 +152,11 @@ def register_rgb_effect(name, effect_type, default_name, schema, *extra_validato
def register_addressable_effect( def register_addressable_effect(
name, effect_type, default_name, schema, *extra_validators name: str,
effect_type: MockObjClass,
default_name: str,
schema: cv.Schema | dict,
*extra_validators,
): ):
# addressable effect can be used only in addressable # addressable effect can be used only in addressable
ADDRESSABLE_EFFECTS.append(name) ADDRESSABLE_EFFECTS.append(name)