fix: add missing s5 codes (#116)

* fix: add missing codes for s5 max

* chore: lint

Author: Lash-L

roborock/code_mappings.py

@@ -163,6 +163,7 @@ class RoborockFanSpeedS6Pure(RoborockFanPowerCode):
     balanced = 102
     turbo = 103
     max = 104
+    custom = 106
 
 
 class RoborockFanSpeedQ7Max(RoborockFanPowerCode):
@@ -226,6 +227,16 @@ class RoborockMopIntensityV2(RoborockMopIntensityCode):
     custom = 207
 
 
+class RoborockMopIntensityS5Max(RoborockMopIntensityCode):
+    """Describes the mop intensity of the vacuum cleaner."""
+
+    off = 200
+    low = 201
+    medium = 202
+    high = 203
+    custom = 204
+
+
 class RoborockDockErrorCode(RoborockEnum):
     """Describes the error code of the dock."""
 

roborock/containers.py

@@ -22,6 +22,7 @@
     RoborockFanSpeedS7,
     RoborockFanSpeedS7MaxV,
     RoborockMopIntensityCode,
+    RoborockMopIntensityS5Max,
     RoborockMopIntensityS7,
     RoborockMopIntensityV2,
     RoborockMopModeCode,
@@ -309,7 +310,7 @@ class S4MaxStatus(Status):
 @dataclass
 class S5MaxStatus(Status):
     fan_power: Optional[RoborockFanSpeedS6Pure] = None
-    water_box_mode: Optional[RoborockMopIntensityV2] = None
+    water_box_mode: Optional[RoborockMopIntensityS5Max] = None
 
 
 @dataclass