jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 21:57:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use ColorMode enum in decora_wifi (#70430)

Browse Source
This commit is contained in:
epenet 2022-04-23 07:49:22 +02:00 committed by GitHub
parent 35e395235a
commit 0e1911cd9b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/decora_wifi/light.py
View File

@ -14,9 +14,8 @@ from homeassistant.components import persistent_notification
from homeassistant.components.light import ( from homeassistant.components.light import (
ATTR_BRIGHTNESS, ATTR_BRIGHTNESS,
ATTR_TRANSITION, ATTR_TRANSITION,
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_ONOFF,
PLATFORM_SCHEMA, PLATFORM_SCHEMA,
ColorMode,
LightEntity, LightEntity,
LightEntityFeature, LightEntityFeature,
) )
@ -103,8 +102,8 @@ class DecoraWifiLight(LightEntity):
def color_mode(self) -> str: def color_mode(self) -> str:
"""Return the color mode of the light.""" """Return the color mode of the light."""
if self._switch.canSetLevel: if self._switch.canSetLevel:
return COLOR_MODE_BRIGHTNESS return ColorMode.BRIGHTNESS
return COLOR_MODE_ONOFF return ColorMode.ONOFF
@property @property
def supported_color_modes(self) -> set[str] | None: def supported_color_modes(self) -> set[str] | None: