Add Support for 527E (#9)

* Added support for 527E * Added support for 527E * Add support for 527E
shenxn · May 18, 2021 · 714073b · 714073b
1 parent 5a3ccbf
commit 714073b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/libdyson/__init__.py b/libdyson/__init__.py
@@ -11,6 +11,7 @@
     DEVICE_TYPE_PURE_COOL_LINK,
     DEVICE_TYPE_PURE_COOL_LINK_DESK,
     DEVICE_TYPE_PURE_HOT_COOL,
+    DEVICE_TYPE_PURE_HOT_COOL_NEW,
     DEVICE_TYPE_PURE_HOT_COOL_LINK,
     DEVICE_TYPE_PURE_HOT_COOL_LINK_NEW,
     DEVICE_TYPE_PURE_HUMIDIFY_COOL,
@@ -58,7 +59,10 @@ def get_device(serial: str, credential: str, device_type: str) -> Optional[Dyson
         DEVICE_TYPE_PURE_HOT_COOL_LINK_NEW,
     ]:
         return DysonPureHotCoolLink(serial, credential, device_type)
-    if device_type == DEVICE_TYPE_PURE_HOT_COOL:
+    if device_type in [                                    
+        DEVICE_TYPE_PURE_HOT_COOL,                                                       
+        DEVICE_TYPE_PURE_HOT_COOL_NEW,                            
+    ]:
         return DysonPureHotCool(serial, credential, device_type)
     if device_type == DEVICE_TYPE_PURE_HUMIDIFY_COOL:
         return DysonPureHumidifyCool(serial, credential, device_type)

diff --git a/libdyson/const.py b/libdyson/const.py
@@ -12,6 +12,7 @@
 DEVICE_TYPE_PURE_HOT_COOL_LINK = "455"
 DEVICE_TYPE_PURE_HOT_COOL_LINK_NEW = "455A"
 DEVICE_TYPE_PURE_HOT_COOL = "527"
+DEVICE_TYPE_PURE_HOT_COOL_NEW = "527E"
 
 DEVICE_TYPE_NAMES = {
     DEVICE_TYPE_360_EYE: "360 Eye robot vacuum",
@@ -22,6 +23,7 @@
     DEVICE_TYPE_PURE_COOL_LINK: "Pure Cool Link",
     DEVICE_TYPE_PURE_COOL_LINK_DESK: "Pure Cool Link Desk",
     DEVICE_TYPE_PURE_HOT_COOL: "Pure Hot+Cool",
+    DEVICE_TYPE_PURE_HOT_COOL_NEW: "Pure Hot+Cool (New)",
     DEVICE_TYPE_PURE_HOT_COOL_LINK: "Pure Hot+Cool Link",
     DEVICE_TYPE_PURE_HOT_COOL_LINK_NEW: "Pure Hot+Cool Link (New)",
     DEVICE_TYPE_PURE_HUMIDIFY_COOL: "Pure Humidify+Cool",

diff --git a/tests/test_init.py b/tests/test_init.py
@@ -12,6 +12,7 @@
     DEVICE_TYPE_PURE_COOL_LINK,
     DEVICE_TYPE_PURE_COOL_LINK_DESK,
     DEVICE_TYPE_PURE_HOT_COOL,
+    DEVICE_TYPE_PURE_HOT_COOL_NEW,
     DEVICE_TYPE_PURE_HOT_COOL_LINK,
     DEVICE_TYPE_PURE_HOT_COOL_LINK_NEW,
     DEVICE_TYPE_PURE_HUMIDIFY_COOL,
@@ -42,6 +43,7 @@
         (DEVICE_TYPE_PURE_HOT_COOL_LINK, DysonPureHotCoolLink),
         (DEVICE_TYPE_PURE_HOT_COOL_LINK_NEW, DysonPureHotCoolLink),
         (DEVICE_TYPE_PURE_HOT_COOL, DysonPureHotCool),
+        (DEVICE_TYPE_PURE_HOT_COOL_NEW, DysonPureHotCool),
         (DEVICE_TYPE_PURE_HUMIDIFY_COOL, DysonPureHumidifyCool),
     ],
 )