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