Use LightEntityFeature enum in limitlessled (#71061)

This commit is contained in:
epenet 2022-04-30 00:38:49 +02:00 committed by GitHub
parent f184f9e142
commit 6635fc4e31
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -27,10 +27,8 @@ from homeassistant.components.light import (
SUPPORT_BRIGHTNESS, SUPPORT_BRIGHTNESS,
SUPPORT_COLOR, SUPPORT_COLOR,
SUPPORT_COLOR_TEMP, SUPPORT_COLOR_TEMP,
SUPPORT_EFFECT,
SUPPORT_FLASH,
SUPPORT_TRANSITION,
LightEntity, LightEntity,
LightEntityFeature,
) )
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PORT, CONF_TYPE, STATE_ON from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PORT, CONF_TYPE, STATE_ON
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -63,23 +61,26 @@ MIN_SATURATION = 10
WHITE = [0, 0] WHITE = [0, 0]
SUPPORT_LIMITLESSLED_WHITE = ( SUPPORT_LIMITLESSLED_WHITE = (
SUPPORT_BRIGHTNESS | SUPPORT_COLOR_TEMP | SUPPORT_EFFECT | SUPPORT_TRANSITION SUPPORT_BRIGHTNESS
| SUPPORT_COLOR_TEMP
| LightEntityFeature.EFFECT
| LightEntityFeature.TRANSITION
) )
SUPPORT_LIMITLESSLED_DIMMER = SUPPORT_BRIGHTNESS | SUPPORT_TRANSITION SUPPORT_LIMITLESSLED_DIMMER = SUPPORT_BRIGHTNESS | LightEntityFeature.TRANSITION
SUPPORT_LIMITLESSLED_RGB = ( SUPPORT_LIMITLESSLED_RGB = (
SUPPORT_BRIGHTNESS SUPPORT_BRIGHTNESS
| SUPPORT_EFFECT | LightEntityFeature.EFFECT
| SUPPORT_FLASH | LightEntityFeature.FLASH
| SUPPORT_COLOR | SUPPORT_COLOR
| SUPPORT_TRANSITION | LightEntityFeature.TRANSITION
) )
SUPPORT_LIMITLESSLED_RGBWW = ( SUPPORT_LIMITLESSLED_RGBWW = (
SUPPORT_BRIGHTNESS SUPPORT_BRIGHTNESS
| SUPPORT_COLOR_TEMP | SUPPORT_COLOR_TEMP
| SUPPORT_EFFECT | LightEntityFeature.EFFECT
| SUPPORT_FLASH | LightEntityFeature.FLASH
| SUPPORT_COLOR | SUPPORT_COLOR
| SUPPORT_TRANSITION | LightEntityFeature.TRANSITION
) )
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
@ -369,7 +370,7 @@ class LimitlessLEDGroup(LightEntity, RestoreEntity):
pipeline.transition(transition_time, **args) pipeline.transition(transition_time, **args)
# Flash. # Flash.
if ATTR_FLASH in kwargs and self._supported & SUPPORT_FLASH: if ATTR_FLASH in kwargs and self._supported & LightEntityFeature.FLASH:
duration = 0 duration = 0
if kwargs[ATTR_FLASH] == FLASH_LONG: if kwargs[ATTR_FLASH] == FLASH_LONG:
duration = 1 duration = 1