From faffd809d51bff27715a7a833ca1918c7632fb49 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 29 Apr 2022 21:48:55 +0200 Subject: [PATCH] Use LightEntityFeature enum in osramlightify (#71059) --- homeassistant/components/osramlightify/light.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/osramlightify/light.py b/homeassistant/components/osramlightify/light.py index 20e50b8b7fd..a247497550f 100644 --- a/homeassistant/components/osramlightify/light.py +++ b/homeassistant/components/osramlightify/light.py @@ -18,9 +18,8 @@ from homeassistant.components.light import ( SUPPORT_BRIGHTNESS, SUPPORT_COLOR, SUPPORT_COLOR_TEMP, - SUPPORT_EFFECT, - SUPPORT_TRANSITION, LightEntity, + LightEntityFeature, ) from homeassistant.const import CONF_HOST from homeassistant.core import HomeAssistant @@ -210,13 +209,18 @@ class Luminary(LightEntity): """Get list of supported features.""" features = 0 if "lum" in self._luminary.supported_features(): - features = features | SUPPORT_BRIGHTNESS | SUPPORT_TRANSITION + features = features | SUPPORT_BRIGHTNESS | LightEntityFeature.TRANSITION if "temp" in self._luminary.supported_features(): - features = features | SUPPORT_COLOR_TEMP | SUPPORT_TRANSITION + features = features | SUPPORT_COLOR_TEMP | LightEntityFeature.TRANSITION if "rgb" in self._luminary.supported_features(): - features = features | SUPPORT_COLOR | SUPPORT_TRANSITION | SUPPORT_EFFECT + features = ( + features + | SUPPORT_COLOR + | LightEntityFeature.TRANSITION + | LightEntityFeature.EFFECT + ) return features @@ -416,7 +420,7 @@ class OsramLightifyGroup(Luminary): """Get list of supported features.""" features = super()._get_supported_features() if self._luminary.scenes(): - features = features | SUPPORT_EFFECT + features = features | LightEntityFeature.EFFECT return features