fix: add some new roborock codes (#233)

humbertogontijo · Nov 8, 2024 · 59546dd · 59546dd
1 parent f08619d
commit 59546dd
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git 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
@@ -28,11 +28,11 @@
     RoborockInCleaning,
     RoborockMopIntensityCode,
     RoborockMopIntensityP10,
+    RoborockMopIntensityQ7Max,
     RoborockMopIntensityS5Max,
     RoborockMopIntensityS6MaxV,
     RoborockMopIntensityS7,
     RoborockMopIntensityS8MaxVUltra,
-    RoborockMopIntensityV2,
     RoborockMopModeCode,
     RoborockMopModeS7,
     RoborockMopModeS8MaxVUltra,
@@ -567,7 +567,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