Dreame F9 Vacuum (dreame.vacuum.p2008) support (#1290)

* Add Dreame F9 support * Update Dreame F9 properties * Dreame F9 add tests * Add play_sound * Dreame integration use base class * Cleaning stats * Fix Dreame 1C name * Use one class for all Dreame implementations * Fixed some conflicts * Add waterflow and fanspeed setters for Dreame F9/D9 * Add Dreame D9 in list of supported models * Fixes * Dreame integrame get rid of Unknown state * Dreame check if speed value set correctly * Dreame Add docstring, fix tests * Dreame merge tests * Dreame increase code coverage * Add Dreame Z10 Pro to the list of supported devices * Dreame vacuum add documnetation for status data container
rytilahti · Feb 7, 2022 · ea51f5f · ea51f5f
1 parent a55149c
commit ea51f5f
Show file tree

Hide file tree

Showing 6 changed files with 616 additions and 117 deletions.
diff --git a/README.rst b/README.rst
@@ -111,6 +111,7 @@ Supported devices
 -  Xiaomi Mijia 360 1080p
 -  Xiaomi Mijia STYJ02YM (Viomi)
 -  Xiaomi Mijia 1C STYTJ01ZHM (Dreame)
+-  Dreame F9, D9, Z10 Pro
 -  Xiaomi Mi Home (Mijia) G1 Robot Vacuum Mop MJSTG1
 -  Xiaomi Roidmi Eve
 -  Xiaomi Mi Smart WiFi Socket

diff --git a/miio/__init__.py b/miio/__init__.py
@@ -55,7 +55,10 @@
     PhilipsWhiteBulb,
 )
 from miio.integrations.petwaterdispenser import PetWaterDispenser
-from miio.integrations.vacuum.dreame.dreamevacuum_miot import DreameVacuumMiot
+from miio.integrations.vacuum.dreame.dreamevacuum_miot import (
+    DreameVacuum,
+    DreameVacuumMiot,
+)
 from miio.integrations.vacuum.mijia import G1Vacuum
 from miio.integrations.vacuum.roborock import RoborockVacuum, Vacuum, VacuumException
 from miio.integrations.vacuum.roborock.vacuumcontainers import (

diff --git a/miio/discovery.py b/miio/discovery.py
@@ -33,6 +33,7 @@
     ChuangmiPlug,
     Cooker,
     Device,
+    DreameVacuum,
     FanLeshow,
     Gateway,
     Heater,
@@ -194,6 +195,10 @@
     "viomi-vacuum-v8": ViomiVacuum,
     "zhimi.heater.za1": partial(Heater, model=MODEL_HEATER_ZA1),
     "zhimi.elecheater.ma1": partial(Heater, model=MODEL_HEATER_MA1),
+    "dreame-vacuum-mc1808": DreameVacuum,
+    "dreame-vacuum-p2008": DreameVacuum,
+    "dreame-vacuum-p2028": DreameVacuum,
+    "dreame-vacuum-p2009": DreameVacuum,
 }