jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-14 16:57:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use LightEntityFeature enum in zha (#71060)

Browse Source
This commit is contained in:
epenet 2022-04-29 21:45:25 +02:00 committed by GitHub
parent e63aef79fe
commit f08615fc76
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
homeassistant/components/zha/light.py
View File

@ -26,9 +26,7 @@ from homeassistant.components.light import (
SUPPORT_BRIGHTNESS, SUPPORT_BRIGHTNESS,
SUPPORT_COLOR, SUPPORT_COLOR,
SUPPORT_COLOR_TEMP, SUPPORT_COLOR_TEMP,
SUPPORT_EFFECT, LightEntityFeature,
SUPPORT_FLASH,
SUPPORT_TRANSITION,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
@ -91,10 +89,10 @@ SIGNAL_LIGHT_GROUP_STATE_CHANGED = "zha_light_group_state_changed"
SUPPORT_GROUP_LIGHT = ( SUPPORT_GROUP_LIGHT = (
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
) )
@ -290,7 +288,7 @@ class BaseLight(LogMixin, light.LightEntity):
if ( if (
effect == light.EFFECT_COLORLOOP effect == light.EFFECT_COLORLOOP
and self.supported_features & light.SUPPORT_EFFECT and self.supported_features & light.LightEntityFeature.EFFECT
): ):
result = await self._color_channel.color_loop_set( result = await self._color_channel.color_loop_set(
UPDATE_COLORLOOP_ACTION UPDATE_COLORLOOP_ACTION
@ -306,7 +304,7 @@ class BaseLight(LogMixin, light.LightEntity):
elif ( elif (
self._effect == light.EFFECT_COLORLOOP self._effect == light.EFFECT_COLORLOOP
and effect != light.EFFECT_COLORLOOP and effect != light.EFFECT_COLORLOOP
and self.supported_features & light.SUPPORT_EFFECT and self.supported_features & light.LightEntityFeature.EFFECT
): ):
result = await self._color_channel.color_loop_set( result = await self._color_channel.color_loop_set(
UPDATE_COLORLOOP_ACTION, UPDATE_COLORLOOP_ACTION,
@ -318,7 +316,10 @@ class BaseLight(LogMixin, light.LightEntity):
t_log["color_loop_set"] = result t_log["color_loop_set"] = result
self._effect = None self._effect = None
if flash is not None and self._supported_features & light.SUPPORT_FLASH: if (
flash is not None
and self._supported_features & light.LightEntityFeature.FLASH
):
result = await self._identify_channel.trigger_effect( result = await self._identify_channel.trigger_effect(
FLASH_EFFECTS[flash], EFFECT_DEFAULT_VARIANT FLASH_EFFECTS[flash], EFFECT_DEFAULT_VARIANT
) )
@ -373,7 +374,7 @@ class Light(BaseLight, ZhaEntity):
if self._level_channel: if self._level_channel:
self._supported_features |= light.SUPPORT_BRIGHTNESS self._supported_features |= light.SUPPORT_BRIGHTNESS
self._supported_features |= light.SUPPORT_TRANSITION self._supported_features |= light.LightEntityFeature.TRANSITION
self._brightness = self._level_channel.current_level self._brightness = self._level_channel.current_level
if self._color_channel: if self._color_channel:
@ -394,13 +395,13 @@ class Light(BaseLight, ZhaEntity):
self._hs_color = (0, 0) self._hs_color = (0, 0)
if color_capabilities & CAPABILITIES_COLOR_LOOP: if color_capabilities & CAPABILITIES_COLOR_LOOP:
self._supported_features |= light.SUPPORT_EFFECT self._supported_features |= light.LightEntityFeature.EFFECT
effect_list.append(light.EFFECT_COLORLOOP) effect_list.append(light.EFFECT_COLORLOOP)
if self._color_channel.color_loop_active == 1: if self._color_channel.color_loop_active == 1:
self._effect = light.EFFECT_COLORLOOP self._effect = light.EFFECT_COLORLOOP
if self._identify_channel: if self._identify_channel:
self._supported_features |= light.SUPPORT_FLASH self._supported_features |= light.LightEntityFeature.FLASH
if effect_list: if effect_list:
self._effect_list = effect_list self._effect_list = effect_list