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

Use ColorMode enum in knx (#70513)

Browse Source
This commit is contained in:
epenet 2022-04-23 21:17:18 +02:00 committed by GitHub
parent 5811e43319
commit 7e7519fe05
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
homeassistant/components/knx/light.py
View File

@ -14,13 +14,7 @@ from homeassistant.components.light import (
ATTR_RGB_COLOR, ATTR_RGB_COLOR,
ATTR_RGBW_COLOR, ATTR_RGBW_COLOR,
ATTR_XY_COLOR, ATTR_XY_COLOR,
COLOR_MODE_BRIGHTNESS, ColorMode,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_HS,
COLOR_MODE_ONOFF,
COLOR_MODE_RGB,
COLOR_MODE_RGBW,
COLOR_MODE_XY,
LightEntity, LightEntity,
) )
from homeassistant.const import CONF_ENTITY_CATEGORY, CONF_NAME, Platform from homeassistant.const import CONF_ENTITY_CATEGORY, CONF_NAME, Platform
@ -267,24 +261,24 @@ class KNXLight(KnxEntity, LightEntity):
return None return None
@property @property
def color_mode(self) -> str | None: def color_mode(self) -> ColorMode | None:
"""Return the color mode of the light.""" """Return the color mode of the light."""
if self._device.supports_xyy_color: if self._device.supports_xyy_color:
return COLOR_MODE_XY return ColorMode.XY
if self._device.supports_hs_color: if self._device.supports_hs_color:
return COLOR_MODE_HS return ColorMode.HS
if self._device.supports_rgbw: if self._device.supports_rgbw:
return COLOR_MODE_RGBW return ColorMode.RGBW
if self._device.supports_color: if self._device.supports_color:
return COLOR_MODE_RGB return ColorMode.RGB
if ( if (
self._device.supports_color_temperature self._device.supports_color_temperature
or self._device.supports_tunable_white or self._device.supports_tunable_white
): ):
return COLOR_MODE_COLOR_TEMP return ColorMode.COLOR_TEMP
if self._device.supports_brightness: if self._device.supports_brightness:
return COLOR_MODE_BRIGHTNESS return ColorMode.BRIGHTNESS
return COLOR_MODE_ONOFF return ColorMode.ONOFF
@property @property
def supported_color_modes(self) -> set | None: def supported_color_modes(self) -> set | None:
@ -373,17 +367,17 @@ class KNXLight(KnxEntity, LightEntity):
await self._device.set_brightness(brightness) await self._device.set_brightness(brightness)
return return
# brightness without color in kwargs; set via color # brightness without color in kwargs; set via color
if self.color_mode == COLOR_MODE_XY: if self.color_mode == ColorMode.XY:
await self._device.set_xyy_color(XYYColor(brightness=brightness)) await self._device.set_xyy_color(XYYColor(brightness=brightness))
return return
# default to white if color not known for RGB(W) # default to white if color not known for RGB(W)
if self.color_mode == COLOR_MODE_RGBW: if self.color_mode == ColorMode.RGBW:
_rgbw = self.rgbw_color _rgbw = self.rgbw_color
if not _rgbw or not any(_rgbw): if not _rgbw or not any(_rgbw):
_rgbw = (0, 0, 0, 255) _rgbw = (0, 0, 0, 255)
await set_color(_rgbw[:3], _rgbw[3], brightness) await set_color(_rgbw[:3], _rgbw[3], brightness)
return return
if self.color_mode == COLOR_MODE_RGB: if self.color_mode == ColorMode.RGB:
_rgb = self.rgb_color _rgb = self.rgb_color
if not _rgb or not any(_rgb): if not _rgb or not any(_rgb):
_rgb = (255, 255, 255) _rgb = (255, 255, 255)