evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use ColorMode enum in homekit_controller (#70503)

Browse source
This commit is contained in:
epenet 2022-04-23 21:09:25 +02:00 • committed by GitHub
parent 50823045e8
commit a0c7fca003
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
homeassistant/components/homekit_controller/light.py
View file

@ -10,10 +10,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,
)
from homeassistant.config_entries import ConfigEntry
@ -87,34 +84,34 @@ class HomeKitLight(HomeKitEntity, LightEntity):
if self.service.has(CharacteristicsTypes.HUE) or self.service.has(
CharacteristicsTypes.SATURATION
):
return COLOR_MODE_HS
return ColorMode.HS
if self.service.has(CharacteristicsTypes.COLOR_TEMPERATURE):
return COLOR_MODE_COLOR_TEMP
return ColorMode.COLOR_TEMP
if self.service.has(CharacteristicsTypes.BRIGHTNESS):
return COLOR_MODE_BRIGHTNESS
return ColorMode.BRIGHTNESS
return COLOR_MODE_ONOFF
return ColorMode.ONOFF
@property
def supported_color_modes(self) -> set[str] | None:
def supported_color_modes(self) -> set[ColorMode | str] | None:
"""Flag supported color modes."""
color_modes = set()
color_modes: set[ColorMode | str] = set()
if self.service.has(CharacteristicsTypes.HUE) or self.service.has(
CharacteristicsTypes.SATURATION
):
color_modes.add(COLOR_MODE_HS)
color_modes.add(ColorMode.HS)
if self.service.has(CharacteristicsTypes.COLOR_TEMPERATURE):
color_modes.add(COLOR_MODE_COLOR_TEMP)
color_modes.add(ColorMode.COLOR_TEMP)
if not color_modes and self.service.has(CharacteristicsTypes.BRIGHTNESS):
color_modes.add(COLOR_MODE_BRIGHTNESS)
color_modes.add(ColorMode.BRIGHTNESS)
if not color_modes:
color_modes.add(COLOR_MODE_ONOFF)
color_modes.add(ColorMode.ONOFF)
return color_modes