Fix supported features for fan and deprecated constants

schmittx · Oct 8, 2024 · 4c126bd · 4c126bd
1 parent bdddd58
commit 4c126bd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/custom_components/leviton_decora_smart_wifi/fan.py b/custom_components/leviton_decora_smart_wifi/fan.py
@@ -85,9 +85,10 @@ def speed_count(self) -> int:
     @property
     def supported_features(self) -> FanEntityFeature:
         """Flag supported features."""
+        supported_features = FanEntityFeature.TURN_OFF | FanEntityFeature.TURN_ON
         if self.device.can_set_level:
-            return FanEntityFeature.SET_SPEED
-        return FanEntityFeature(0)
+            supported_features = supported_features | FanEntityFeature.SET_SPEED
+        return supported_features
 
     def turn_on(
         self,

diff --git a/custom_components/leviton_decora_smart_wifi/manifest.json b/custom_components/leviton_decora_smart_wifi/manifest.json
@@ -12,5 +12,5 @@
   "loggers": [],
   "quality_scale": "gold",
   "requirements": ["pypng==0.20220715.0", "PyQRCode==1.2.1"],
-  "version": "1.5.0"
+  "version": "1.5.1"
 }
diff --git a/custom_components/leviton_decora_smart_wifi/number.py b/custom_components/leviton_decora_smart_wifi/number.py
@@ -11,7 +11,7 @@
     NumberEntityDescription,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import PERCENTAGE, TIME_MINUTES
+from homeassistant.const import PERCENTAGE
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback