Migrate smarttub light to color_mode ()

This commit is contained in:
epenet 2022-04-07 09:10:40 +02:00 committed by GitHub
parent 766e6f2e9c
commit 3828c36202
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -4,8 +4,8 @@ from smarttub import SpaLight
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_EFFECT,
COLOR_MODE_BRIGHTNESS,
EFFECT_COLORLOOP,
SUPPORT_BRIGHTNESS,
SUPPORT_EFFECT,
LightEntity,
)
@ -43,6 +43,9 @@ async def async_setup_entry(
class SmartTubLight(SmartTubEntity, LightEntity):
"""A light on a spa."""
_attr_color_mode = COLOR_MODE_BRIGHTNESS
_attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS}
def __init__(self, coordinator, light):
"""Initialize the entity."""
super().__init__(coordinator, light.spa, "light")
@ -89,7 +92,7 @@ class SmartTubLight(SmartTubEntity, LightEntity):
@property
def supported_features(self):
"""Flag supported features."""
return SUPPORT_BRIGHTNESS | SUPPORT_EFFECT
return SUPPORT_EFFECT
@property
def effect(self):