Use ColorMode enum in futurenow (#70498)

This commit is contained in:
epenet 2022-04-23 21:21:05 +02:00 committed by GitHub
parent 562927cc17
commit 33c1501f35
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -6,9 +6,8 @@ import voluptuous as vol
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_ONOFF,
PLATFORM_SCHEMA,
ColorMode,
LightEntity,
)
from homeassistant.const import CONF_DEVICES, CONF_HOST, CONF_NAME, CONF_PORT
@ -110,8 +109,8 @@ class FutureNowLight(LightEntity):
def color_mode(self) -> str:
"""Return the color mode of the light."""
if self._dimmable:
return COLOR_MODE_BRIGHTNESS
return COLOR_MODE_ONOFF
return ColorMode.BRIGHTNESS
return ColorMode.ONOFF
@property
def supported_color_modes(self) -> set[str] | None: