Add entities for DysonAirQualitySelect

shenxn · Nov 25, 2021 · d6d88e5 · d6d88e5
1 parent 94ec838
commit d6d88e5
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/custom_components/dyson_local/select.py b/custom_components/dyson_local/select.py
@@ -2,7 +2,13 @@
 
 from typing import Callable
 
-from libdyson import DysonPureHumidifyCool, HumidifyOscillationMode, WaterHardness
+from libdyson import (
+    DysonPureHumidifyCool,
+    DysonPureHotCoolLink,
+    DysonPureCoolLink,
+    HumidifyOscillationMode,
+    WaterHardness,
+)
 from libdyson.const import AirQualityTarget
 
 from homeassistant.components.select import SelectEntity
@@ -54,6 +60,10 @@ async def async_setup_entry(
     device = hass.data[DOMAIN][DATA_DEVICES][config_entry.entry_id]
     name = config_entry.data[CONF_NAME]
     entities = []
+    if isinstance(device, DysonPureHotCoolLink) or isinstance(
+        device, DysonPureCoolLink
+    ):
+        entities.append(DysonAirQualitySelect(device, name))
     if isinstance(device, DysonPureHumidifyCool):
         entities.extend(
             [