From 1ba0a58b6d0dc2d81f7bfe75750a0f3a0bae55ee Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Sat, 23 Apr 2022 21:21:33 +0200 Subject: [PATCH] Use ColorMode enum in lcn (#70515) --- homeassistant/components/lcn/light.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/lcn/light.py b/homeassistant/components/lcn/light.py index 255619c14cc..2f8ce94db2a 100644 --- a/homeassistant/components/lcn/light.py +++ b/homeassistant/components/lcn/light.py @@ -8,10 +8,9 @@ import pypck from homeassistant.components.light import ( ATTR_BRIGHTNESS, ATTR_TRANSITION, - COLOR_MODE_BRIGHTNESS, - COLOR_MODE_ONOFF, DOMAIN as DOMAIN_LIGHT, SUPPORT_TRANSITION, + ColorMode, LightEntity, ) from homeassistant.config_entries import ConfigEntry @@ -85,9 +84,9 @@ class LcnOutputLight(LcnEntity, LightEntity): self._is_dimming_to_zero = False if self.dimmable: - self._attr_color_mode = COLOR_MODE_BRIGHTNESS + self._attr_color_mode = ColorMode.BRIGHTNESS else: - self._attr_color_mode = COLOR_MODE_ONOFF + self._attr_color_mode = ColorMode.ONOFF self._attr_supported_color_modes = {self._attr_color_mode} async def async_added_to_hass(self) -> None: @@ -169,8 +168,8 @@ class LcnOutputLight(LcnEntity, LightEntity): class LcnRelayLight(LcnEntity, LightEntity): """Representation of a LCN light for relay ports.""" - _attr_color_mode = COLOR_MODE_ONOFF - _attr_supported_color_modes = {COLOR_MODE_ONOFF} + _attr_color_mode = ColorMode.ONOFF + _attr_supported_color_modes = {ColorMode.ONOFF} def __init__( self, config: ConfigType, entry_id: str, device_connection: DeviceConnectionType