fix: add custom code for p10

humbertogontijo · Sep 24, 2023 · 49ad8f3 · 49ad8f3
1 parent 36932c1
commit 49ad8f3
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
@@ -178,6 +178,7 @@ class RoborockFanSpeedP10(RoborockFanPowerCode):
     balanced = 102
     turbo = 103
     max = 104
+    custom = 106
     max_plus = 108
 
 
@@ -225,6 +226,16 @@ class RoborockMopIntensityV2(RoborockMopIntensityCode):
     high = 203
     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
@@ -29,7 +29,7 @@
     RoborockMopModeCode,
     RoborockMopModeS7,
     RoborockMopModeS8ProUltra,
-    RoborockStateCode,
+    RoborockStateCode, RoborockMopIntensityP10,
 )
 from .const import (
     FILTER_REPLACE_TIME,
@@ -362,7 +362,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