Use ColorMode enum in zengge (#70557)

This commit is contained in:
epenet 2022-04-23 22:10:35 +02:00 committed by GitHub
parent da2312765c
commit f9dbb92e60
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -10,9 +10,8 @@ from homeassistant.components.light import (
ATTR_BRIGHTNESS, ATTR_BRIGHTNESS,
ATTR_HS_COLOR, ATTR_HS_COLOR,
ATTR_WHITE, ATTR_WHITE,
COLOR_MODE_HS,
COLOR_MODE_WHITE,
PLATFORM_SCHEMA, PLATFORM_SCHEMA,
ColorMode,
LightEntity, LightEntity,
) )
from homeassistant.const import CONF_DEVICES, CONF_NAME from homeassistant.const import CONF_DEVICES, CONF_NAME
@ -100,16 +99,16 @@ class ZenggeLight(LightEntity):
return self._white return self._white
@property @property
def color_mode(self): def color_mode(self) -> ColorMode:
"""Return the current color mode.""" """Return the current color mode."""
if self._white != 0: if self._white != 0:
return COLOR_MODE_WHITE return ColorMode.WHITE
return COLOR_MODE_HS return ColorMode.HS
@property @property
def supported_color_modes(self): def supported_color_modes(self) -> set[ColorMode | str]:
"""Flag supported color modes.""" """Flag supported color modes."""
return {COLOR_MODE_HS, COLOR_MODE_WHITE} return {ColorMode.HS, ColorMode.WHITE}
@property @property
def assumed_state(self): def assumed_state(self):