Use ColorMode enum in control4 (#70428)

This commit is contained in:
epenet 2022-04-23 07:49:08 +02:00 committed by GitHub
parent bd2639af65
commit 7ffe399a08
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -11,8 +11,7 @@ from pyControl4.light import C4Light
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_TRANSITION,
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_ONOFF,
ColorMode,
LightEntity,
LightEntityFeature,
)
@ -166,11 +165,11 @@ class Control4Light(Control4Entity, LightEntity):
)
self._is_dimmer = is_dimmer
if is_dimmer:
self._attr_color_mode = COLOR_MODE_BRIGHTNESS
self._attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS}
self._attr_color_mode = ColorMode.BRIGHTNESS
self._attr_supported_color_modes = {ColorMode.BRIGHTNESS}
else:
self._attr_color_mode = COLOR_MODE_ONOFF
self._attr_supported_color_modes = {COLOR_MODE_ONOFF}
self._attr_color_mode = ColorMode.ONOFF
self._attr_supported_color_modes = {ColorMode.ONOFF}
def create_api_object(self):
"""Create a pyControl4 device object.