From 7a7a3035bd148fcd9ef22faa28fa2eef857a066a Mon Sep 17 00:00:00 2001
From: Luke <conway220@gmail.com>
Date: Thu, 7 Nov 2024 18:13:37 -0500
Subject: [PATCH] fix: add some new roborock codes

---
 roborock/code_mappings.py | 11 +++++++++++
 roborock/containers.py    |  4 ++--
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/roborock/code_mappings.py b/roborock/code_mappings.py
index 321def2..f4b5e03 100644
--- a/roborock/code_mappings.py
+++ b/roborock/code_mappings.py
@@ -288,6 +288,7 @@ class RoborockMopModeS8ProUltra(RoborockMopModeCode):
 class RoborockMopModeS8MaxVUltra(RoborockMopModeCode):
     standard = 300
     deep = 301
+    custom = 302
     deep_plus = 303
     fast = 304
     deep_plus_pearl = 305
@@ -362,6 +363,16 @@ class RoborockMopIntensityS6MaxV(RoborockMopIntensityCode):
     custom_water_flow = 207
 
 
+class RoborockMopIntensityQ7Max(RoborockMopIntensityCode):
+    """Describes the mop intensity of the vacuum cleaner."""
+
+    off = 200
+    low = 201
+    medium = 202
+    high = 203
+    custom_water_flow = 207
+
+
 class RoborockDockErrorCode(RoborockEnum):
     """Describes the error code of the dock."""
 
diff --git a/roborock/containers.py b/roborock/containers.py
index 862bb4a..194f01b 100644
--- a/roborock/containers.py
+++ b/roborock/containers.py
@@ -30,11 +30,11 @@
     RoborockInCleaning,
     RoborockMopIntensityCode,
     RoborockMopIntensityP10,
+    RoborockMopIntensityQ7Max,
     RoborockMopIntensityS5Max,
     RoborockMopIntensityS6MaxV,
     RoborockMopIntensityS7,
     RoborockMopIntensityS8MaxVUltra,
-    RoborockMopIntensityV2,
     RoborockMopModeCode,
     RoborockMopModeS7,
     RoborockMopModeS8MaxVUltra,
@@ -512,7 +512,7 @@ class S5MaxStatus(Status):
 @dataclass
 class Q7MaxStatus(Status):
     fan_power: RoborockFanSpeedQ7Max | None = None
-    water_box_mode: RoborockMopIntensityV2 | None = None
+    water_box_mode: RoborockMopIntensityQ7Max | None = None
 
 
 @dataclass