From 77f6d6fc917831f1966d2138bc7355292fa1e5e2 Mon Sep 17 00:00:00 2001
From: Rob Coleman <rccoleman@gmail.com>
Date: Fri, 20 Dec 2024 07:04:49 -0800
Subject: [PATCH] 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
---
 roborock/code_mappings.py | 10 +++++++---
 roborock/containers.py    |  2 +-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/roborock/code_mappings.py b/roborock/code_mappings.py
index 115c9ac..50579e1 100644
--- 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
index 44da6f9..3bda790 100644
--- a/roborock/containers.py
+++ b/roborock/containers.py
@@ -36,10 +36,10 @@
     RoborockMopIntensityS7,
     RoborockMopIntensityS8MaxVUltra,
     RoborockMopModeCode,
+    RoborockMopModeQRevoMaster,
     RoborockMopModeS7,
     RoborockMopModeS8MaxVUltra,
     RoborockMopModeS8ProUltra,
-    RoborockMopModeQRevoMaster,
     RoborockStartType,
     RoborockStateCode,
 )