Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Migrate WLED to use EntityFeature enums (#69219)

pull/69263/head
Franck Nijhof 2022-04-04 13:28:19 +02:00 committed by GitHub
parent 76247414bf
commit 614076cdf1
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/wled/light.py
View File

@ -13,9 +13,8 @@ from homeassistant.components.light import (
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_RGB,
COLOR_MODE_RGBW,
SUPPORT_EFFECT,
SUPPORT_TRANSITION,
LightEntity,
LightEntityFeature,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
@ -55,7 +54,7 @@ class WLEDMasterLight(WLEDEntity, LightEntity):
_attr_color_mode = COLOR_MODE_BRIGHTNESS
_attr_icon = "mdi:led-strip-variant"
_attr_supported_features = SUPPORT_TRANSITION
_attr_supported_features = LightEntityFeature.TRANSITION
def __init__(self, coordinator: WLEDDataUpdateCoordinator) -> None:
"""Initialize WLED master light."""
@ -105,7 +104,7 @@ class WLEDMasterLight(WLEDEntity, LightEntity):
class WLEDSegmentLight(WLEDEntity, LightEntity):
"""Defines a WLED light based on a segment."""
_attr_supported_features = SUPPORT_EFFECT | SUPPORT_TRANSITION
_attr_supported_features = LightEntityFeature.EFFECT | LightEntityFeature.TRANSITION
_attr_icon = "mdi:led-strip-variant"
def __init__(