fix: add custom code for p10 (#123)

humbertogontijo · Sep 24, 2023 · 8b57d50 · 8b57d50
1 parent d154e15
commit 8b57d50
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/roborock/code_mappings.py b/roborock/code_mappings.py
@@ -178,6 +178,7 @@ class RoborockFanSpeedP10(RoborockFanPowerCode):
     balanced = 102
     turbo = 103
     max = 104
+    custom = 106
     max_plus = 108
 
 
@@ -226,6 +227,16 @@ class RoborockMopIntensityV2(RoborockMopIntensityCode):
     custom = 207
 
 
+class RoborockMopIntensityP10(RoborockMopIntensityCode):
+    """Describes the mop intensity of the vacuum cleaner."""
+
+    off = 200
+    low = 201
+    medium = 202
+    high = 203
+    custom = 204
+
+
 class RoborockMopIntensityS5Max(RoborockMopIntensityCode):
     """Describes the mop intensity of the vacuum cleaner."""
 

diff --git a/roborock/containers.py b/roborock/containers.py
@@ -23,6 +23,7 @@
     RoborockFanSpeedS7,
     RoborockFanSpeedS7MaxV,
     RoborockMopIntensityCode,
+    RoborockMopIntensityP10,
     RoborockMopIntensityS5Max,
     RoborockMopIntensityS7,
     RoborockMopIntensityV2,
@@ -362,7 +363,7 @@ class S8Status(Status):
 @dataclass
 class P10Status(Status):
     fan_power: RoborockFanSpeedP10 | None = None
-    water_box_mode: RoborockMopIntensityV2 | None = None
+    water_box_mode: RoborockMopIntensityP10 | None = None
     mop_mode: RoborockMopModeS8ProUltra | None = None