Add 360 heurist support

shenxn · Apr 9, 2021 · b18fe76 · b18fe76
1 parent c4a8b83
commit b18fe76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/dyson_cloud/camera.py b/custom_components/dyson_cloud/camera.py
@@ -4,7 +4,7 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.components.camera import Camera
-from libdyson.const import DEVICE_TYPE_360_EYE
+from libdyson.const import DEVICE_TYPE_360_EYE, DEVICE_TYPE_360_HEURIST
 from libdyson.cloud.cloud_360_eye import DysonCloud360Eye
 from libdyson.cloud import DysonDeviceInfo
 from datetime import timedelta
@@ -25,7 +25,7 @@ async def async_setup_entry(
     devices = data[DATA_DEVICES]
     entities = []
     for device in devices:
-        if device.product_type != DEVICE_TYPE_360_EYE:
+        if device.product_type not in [DEVICE_TYPE_360_EYE, DEVICE_TYPE_360_HEURIST]:
             continue
         entities.append(DysonCleaningMapEntity(
             DysonCloud360Eye(account, device.serial),