Add display switch to fan devices. (#32)

* Improve night_light feature discovery * add display switch to fans * revert night_light removal from humidifier * Improve switch detection * Update light.py * fix style
vlebourl · Jul 11, 2022 · 3498625 · 3498625
1 parent cdbb0b8
commit 3498625
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/custom_components/vesync/common.py b/custom_components/vesync/common.py
@@ -59,6 +59,7 @@ async def async_process_devices(hass, manager):
             else:
                 if hasattr(fan, "config_dict"):
                     devices[VS_NUMBERS].append(fan)
+                devices[VS_SWITCHES].append(fan)  # for automatic stop and display
                 devices[VS_FANS].append(fan)
         _LOGGER.info("%d VeSync fans found", len(manager.fans))
 

diff --git a/custom_components/vesync/switch.py b/custom_components/vesync/switch.py
@@ -52,6 +52,8 @@ def _setup_entities(devices, async_add_entities):
                     VeSyncHumidifierAutoOnHA(dev),
                 )
             )
+        elif getattr(dev, "turn_on_display", None):
+            entities.append(VeSyncHumidifierDisplayHA(dev))
 
         else:
             _LOGGER.warning(