diff --git a/custom_components/vesync/common.py b/custom_components/vesync/common.py index be0e154..03ccb16 100644 --- 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 index 7980cb1..ceea339 100644 --- 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(