Add optional disabling Fan presets #882

custom_components/sonoff/fan.py

@@ -5,7 +5,7 @@
 
 from .core.const import DOMAIN
 from .core.entity import XEntity
-from .core.ewelink import SIGNAL_ADD_ENTITIES, XRegistry
+from .core.ewelink import SIGNAL_ADD_ENTITIES, XRegistry, XDevice
 
 PARALLEL_UPDATES = 0  # fix entity_platform parallel_updates Semaphore
 
@@ -28,9 +28,15 @@ async def async_setup_entry(hass, config_entry, add_entities):
 class XFan(XEntity, FanEntity):
     params = {"switches", "fan"}
     _attr_speed_count = 3
-    _attr_supported_features = FanEntityFeature.SET_SPEED | FanEntityFeature.PRESET_MODE
+    _attr_supported_features = FanEntityFeature.SET_SPEED
     _attr_preset_modes = [SPEED_OFF, SPEED_LOW, SPEED_MEDIUM, SPEED_HIGH]
 
+    def __init__(self, ewelink: XRegistry, device: XDevice) -> None:
+        super().__init__(ewelink, device)
+
+        if device.get("preset_mode", True):
+            self._attr_supported_features |= FanEntityFeature.PRESET_MODE
+
     def set_state(self, params: dict):
         mode = None
         # Cloud sends switches, LAN sends fan/speed