Use ColorMode enum in elkm1 (#70488)

This commit is contained in:
epenet 2022-04-23 20:40:58 +02:00 committed by GitHub
parent 6c391ecc59
commit fddc0e14c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -7,11 +7,7 @@ from elkm1_lib.elements import Element
from elkm1_lib.elk import Elk
from elkm1_lib.lights import Light
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
COLOR_MODE_BRIGHTNESS,
LightEntity,
)
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -36,8 +32,8 @@ async def async_setup_entry(
class ElkLight(ElkEntity, LightEntity):
"""Representation of an Elk lighting device."""
_attr_color_mode = COLOR_MODE_BRIGHTNESS
_attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS}
_attr_color_mode = ColorMode.BRIGHTNESS
_attr_supported_color_modes = {ColorMode.BRIGHTNESS}
_element: Light
def __init__(self, element: Element, elk: Elk, elk_data: dict[str, Any]) -> None: