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