Use ColorMode enum in home_connect (#70502)

This commit is contained in:
epenet 2022-04-23 21:14:49 +02:00 committed by GitHub
parent 648c973785
commit 994905d1ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -7,8 +7,7 @@ from homeconnect.api import HomeConnectError
from homeassistant.components.light import ( from homeassistant.components.light import (
ATTR_BRIGHTNESS, ATTR_BRIGHTNESS,
ATTR_HS_COLOR, ATTR_HS_COLOR,
COLOR_MODE_BRIGHTNESS, ColorMode,
COLOR_MODE_HS,
LightEntity, LightEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@ -68,15 +67,15 @@ class HomeConnectLight(HomeConnectEntity, LightEntity):
self._key = BSH_AMBIENT_LIGHT_ENABLED self._key = BSH_AMBIENT_LIGHT_ENABLED
self._custom_color_key = BSH_AMBIENT_LIGHT_CUSTOM_COLOR self._custom_color_key = BSH_AMBIENT_LIGHT_CUSTOM_COLOR
self._color_key = BSH_AMBIENT_LIGHT_COLOR self._color_key = BSH_AMBIENT_LIGHT_COLOR
self._attr_color_mode = COLOR_MODE_HS self._attr_color_mode = ColorMode.HS
self._attr_supported_color_modes = {COLOR_MODE_HS} self._attr_supported_color_modes = {ColorMode.HS}
else: else:
self._brightness_key = COOKING_LIGHTING_BRIGHTNESS self._brightness_key = COOKING_LIGHTING_BRIGHTNESS
self._key = COOKING_LIGHTING self._key = COOKING_LIGHTING
self._custom_color_key = None self._custom_color_key = None
self._color_key = None self._color_key = None
self._attr_color_mode = COLOR_MODE_BRIGHTNESS self._attr_color_mode = ColorMode.BRIGHTNESS
self._attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS} self._attr_supported_color_modes = {ColorMode.BRIGHTNESS}
@property @property
def is_on(self): def is_on(self):