diff --git a/homeassistant/components/hue/v1/light.py b/homeassistant/components/hue/v1/light.py index eed8a35f705..420256f421b 100644 --- a/homeassistant/components/hue/v1/light.py +++ b/homeassistant/components/hue/v1/light.py @@ -20,11 +20,9 @@ from homeassistant.components.light import ( EFFECT_RANDOM, FLASH_LONG, FLASH_SHORT, - SUPPORT_EFFECT, - SUPPORT_FLASH, - SUPPORT_TRANSITION, ColorMode, LightEntity, + LightEntityFeature, filter_supported_color_modes, ) from homeassistant.core import callback @@ -73,10 +71,10 @@ COLOR_MODES_HUE = { "Color temperature light": COLOR_MODES_HUE_COLOR_TEMP, } -SUPPORT_HUE_ON_OFF = SUPPORT_FLASH | SUPPORT_TRANSITION +SUPPORT_HUE_ON_OFF = LightEntityFeature.FLASH | LightEntityFeature.TRANSITION SUPPORT_HUE_DIMMABLE = SUPPORT_HUE_ON_OFF SUPPORT_HUE_COLOR_TEMP = SUPPORT_HUE_DIMMABLE -SUPPORT_HUE_COLOR = SUPPORT_HUE_DIMMABLE | SUPPORT_EFFECT +SUPPORT_HUE_COLOR = SUPPORT_HUE_DIMMABLE | LightEntityFeature.EFFECT SUPPORT_HUE_EXTENDED = SUPPORT_HUE_COLOR_TEMP | SUPPORT_HUE_COLOR SUPPORT_HUE = { diff --git a/homeassistant/components/hue/v2/group.py b/homeassistant/components/hue/v2/group.py index 8a6752307cc..cbf974325e0 100644 --- a/homeassistant/components/hue/v2/group.py +++ b/homeassistant/components/hue/v2/group.py @@ -15,10 +15,9 @@ from homeassistant.components.light import ( ATTR_TRANSITION, ATTR_XY_COLOR, FLASH_SHORT, - SUPPORT_FLASH, - SUPPORT_TRANSITION, ColorMode, LightEntity, + LightEntityFeature, ) from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant, callback @@ -81,8 +80,8 @@ class GroupedHueLight(HueBaseEntity, LightEntity): self.group = group self.controller = controller self.api: HueBridgeV2 = bridge.api - self._attr_supported_features |= SUPPORT_FLASH - self._attr_supported_features |= SUPPORT_TRANSITION + self._attr_supported_features |= LightEntityFeature.FLASH + self._attr_supported_features |= LightEntityFeature.TRANSITION self._dynamic_mode_active = False self._update_values() diff --git a/homeassistant/components/hue/v2/light.py b/homeassistant/components/hue/v2/light.py index aaf96a3ca17..0f7cc6cdbab 100644 --- a/homeassistant/components/hue/v2/light.py +++ b/homeassistant/components/hue/v2/light.py @@ -17,11 +17,9 @@ from homeassistant.components.light import ( ATTR_TRANSITION, ATTR_XY_COLOR, FLASH_SHORT, - SUPPORT_EFFECT, - SUPPORT_FLASH, - SUPPORT_TRANSITION, ColorMode, LightEntity, + LightEntityFeature, ) from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant, callback @@ -74,7 +72,7 @@ class HueLight(HueBaseEntity, LightEntity): """Initialize the light.""" super().__init__(bridge, controller, resource) if self.resource.alert and self.resource.alert.action_values: - self._attr_supported_features |= SUPPORT_FLASH + self._attr_supported_features |= LightEntityFeature.FLASH self.resource = resource self.controller = controller self._supported_color_modes: set[ColorMode | str] = set() @@ -87,7 +85,7 @@ class HueLight(HueBaseEntity, LightEntity): # only add color mode brightness if no color variants self._supported_color_modes.add(ColorMode.BRIGHTNESS) # support transition if brightness control - self._attr_supported_features |= SUPPORT_TRANSITION + self._attr_supported_features |= LightEntityFeature.TRANSITION # get list of supported effects (combine effects and timed_effects) self._attr_effect_list = [] if effects := resource.effects: @@ -102,7 +100,7 @@ class HueLight(HueBaseEntity, LightEntity): ] if len(self._attr_effect_list) > 0: self._attr_effect_list.insert(0, EFFECT_NONE) - self._attr_supported_features |= SUPPORT_EFFECT + self._attr_supported_features |= LightEntityFeature.EFFECT @property def brightness(self) -> int | None: