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