fix: update mop intensity, fan speed, and dock mappings for the QRevo…

… Master (#260) * fix: update mop intensity, fan speed, and dock mappings for QRevo Master * Fix sorting of imports * Rerun precommit
humbertogontijo · Dec 20, 2024 · 77f6d6f · 77f6d6f
1 parent 71e8a9e
commit 77f6d6f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/roborock/code_mappings.py b/roborock/code_mappings.py
@@ -244,12 +244,14 @@ class RoborockFanSpeedQ7Max(RoborockFanPowerCode):
 
 
 class RoborockFanSpeedQRevoMaster(RoborockFanPowerCode):
+    off = 105
     quiet = 101
     balanced = 102
     turbo = 103
     max = 104
-    max_plus = 105
-    custom = 110  # Smartplan
+    custom = 106
+    max_plus = 108
+    smart_mode = 110
 
 
 class RoborockFanSpeedP10(RoborockFanPowerCode):
@@ -344,8 +346,9 @@ class RoborockMopIntensityQRevoMaster(RoborockMopIntensityCode):
     low = 201
     medium = 202
     high = 203
+    custom = 204
     custom_water_flow = 207
-    custom = 209  # SmartPlan
+    smart_mode = 209
 
 
 class RoborockMopIntensityP10(RoborockMopIntensityCode):
@@ -426,6 +429,7 @@ class RoborockDockTypeCode(RoborockEnum):
     p10_dock = 8
     p10_pro_dock = 9
     s8_maxv_ultra_dock = 10
+    qrevo_master_dock = 14
     qrevo_s_dock = 15
 
 

diff --git a/roborock/containers.py b/roborock/containers.py
@@ -36,10 +36,10 @@
     RoborockMopIntensityS7,
     RoborockMopIntensityS8MaxVUltra,
     RoborockMopModeCode,
+    RoborockMopModeQRevoMaster,
     RoborockMopModeS7,
     RoborockMopModeS8MaxVUltra,
     RoborockMopModeS8ProUltra,
-    RoborockMopModeQRevoMaster,
     RoborockStartType,
     RoborockStateCode,
 )