jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 13:47:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use ColorMode enum in tuya (#70545)

Browse Source
This commit is contained in:
epenet 2022-04-23 23:00:34 +02:00 committed by GitHub
parent 9906fd649e
commit 89200b27f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
homeassistant/components/tuya/light.py
View File

@ -11,10 +11,7 @@ from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_COLOR_TEMP,
ATTR_HS_COLOR,
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_HS,
COLOR_MODE_ONOFF,
ColorMode,
LightEntity,
LightEntityDescription,
)
@ -401,7 +398,7 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
super().__init__(device, device_manager)
self.entity_description = description
self._attr_unique_id = f"{super().unique_id}{description.key}"
self._attr_supported_color_modes = {COLOR_MODE_ONOFF}
self._attr_supported_color_modes = set()
# Determine DPCodes
self._color_mode_dpcode = self.find_dpcode(
@ -412,7 +409,7 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
description.brightness, dptype=DPType.INTEGER, prefer_function=True
):
self._brightness = int_type
self._attr_supported_color_modes.add(COLOR_MODE_BRIGHTNESS)
self._attr_supported_color_modes.add(ColorMode.BRIGHTNESS)
self._brightness_max = self.find_dpcode(
description.brightness_max, dptype=DPType.INTEGER
)
@ -424,13 +421,13 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
description.color_temp, dptype=DPType.INTEGER, prefer_function=True
):
self._color_temp = int_type
self._attr_supported_color_modes.add(COLOR_MODE_COLOR_TEMP)
self._attr_supported_color_modes.add(ColorMode.COLOR_TEMP)
if (
dpcode := self.find_dpcode(description.color_data, prefer_function=True)
) and self.get_dptype(dpcode) == DPType.JSON:
self._color_data_dpcode = dpcode
self._attr_supported_color_modes.add(COLOR_MODE_HS)
self._attr_supported_color_modes.add(ColorMode.HS)
if dpcode in self.device.function:
values = cast(str, self.device.function[dpcode].values)
else:
@ -451,6 +448,9 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
):
self._color_data_type = DEFAULT_COLOR_TYPE_DATA_V2
if not self._attr_supported_color_modes:
self._attr_supported_color_modes = {ColorMode.ONOFF}
@property
def is_on(self) -> bool:
"""Return true if light is on."""
@ -484,7 +484,7 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
]
elif self._color_data_type and (
ATTR_HS_COLOR in kwargs
or (ATTR_BRIGHTNESS in kwargs and self.color_mode == COLOR_MODE_HS)
or (ATTR_BRIGHTNESS in kwargs and self.color_mode == ColorMode.HS)
):
if self._color_mode_dpcode:
commands += [
@ -527,7 +527,7 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
if (
ATTR_BRIGHTNESS in kwargs
and self.color_mode != COLOR_MODE_HS
and self.color_mode != ColorMode.HS
and self._brightness
):
brightness = kwargs[ATTR_BRIGHTNESS]
@ -578,7 +578,7 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
def brightness(self) -> int | None:
"""Return the brightness of the light."""
# If the light is currently in color mode, extract the brightness from the color data
if self.color_mode == COLOR_MODE_HS and (color_data := self._get_color_data()):
if self.color_mode == ColorMode.HS and (color_data := self._get_color_data()):
return color_data.brightness
if not self._brightness:
@ -640,7 +640,7 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
return color_data.hs_color
@property
def color_mode(self) -> str:
def color_mode(self) -> ColorMode:
"""Return the color_mode of the light."""
# We consider it to be in HS color mode, when work mode is anything
# else than "white".
@ -648,12 +648,12 @@ class TuyaLightEntity(TuyaEntity, LightEntity):
self._color_mode_dpcode
and self.device.status.get(self._color_mode_dpcode) != WorkMode.WHITE
):
return COLOR_MODE_HS
return ColorMode.HS
if self._color_temp:
return COLOR_MODE_COLOR_TEMP
return ColorMode.COLOR_TEMP
if self._brightness:
return COLOR_MODE_BRIGHTNESS
return COLOR_MODE_ONOFF
return ColorMode.BRIGHTNESS
return ColorMode.ONOFF
def _get_color_data(self) -> ColorData | None:
"""Get current color data from device."""