Add feature flags to fan entity (#17)

samuolis · Dec 15, 2024 · 20e8397 · 20e8397
1 parent 41c4491
commit 20e8397
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/custom_components/brink_ventilation/fan.py b/custom_components/brink_ventilation/fan.py
@@ -6,7 +6,7 @@
 from homeassistant.components.fan import (
     DOMAIN,
     FanEntity,
-    SUPPORT_SET_SPEED
+    FanEntityFeature
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -75,9 +75,9 @@ def unique_id(self):
 
     @property
     def supported_features(self):
-        """Return supported features."""
-        return SUPPORT_SET_SPEED
-    
+        """Flag supported features."""
+        return FanEntityFeature.TURN_OFF | FanEntityFeature.TURN_ON | FanEntityFeature.SET_SPEED
+
     @property
     def is_on(self):
         """If the fan currently is on or off."""