feat: add more codes for status (#55)

Author: Lash-L

roborock/api.py

@@ -30,6 +30,7 @@
     MultiMapsList,
     NetworkInfo,
     RoomMapping,
+    S7MaxVStatus,
     SmartWashParams,
     Status,
     UserData,
@@ -227,7 +228,9 @@ async def send_command(self, method: RoborockCommand, params: Optional[list | di
     async def get_status(self) -> Status | None:
         status = await self.send_command(RoborockCommand.GET_STATUS)
         if isinstance(status, dict):
-            _cls: Type[Status] = ModelStatus[self.device_info.model]
+            _cls: Type[Status] = ModelStatus.get(
+                self.device_info.model, S7MaxVStatus
+            )  # Default to S7 MAXV if we don't have the data
             return _cls.from_dict(status)
 
         return None

roborock/code_mappings.py

@@ -149,6 +149,7 @@ class RoborockFanSpeedS7MaxV(RoborockFanPowerCode):
     balanced = 102
     turbo = 103
     max = 104
+    custom = 106
     max_plus = 108
 
 

roborock/const.py

@@ -23,6 +23,7 @@
 ROBOROCK_Q7 = "roborock.vacuum.a37"  # CHECK THIS
 ROBOROCK_Q7_MAX = "roborock.vacuum.a38"
 ROBOROCK_Q7PLUS = "roborock.vacuum.a40"
+ROBOROCK_G10S_PRO = "roborock.vacuum.a26"
 ROBOROCK_G10S = "roborock.vacuum.a46"
 ROBOROCK_G10 = "roborock.vacuum.a29"
 ROBOROCK_S7 = "roborock.vacuum.a15"
@@ -37,6 +38,7 @@
 SUPPORTED_VACUUMS = (
     [  # These are the devices that show up when you add a device - more could be supported and just not show up
         ROBOROCK_G10,
+        ROBOROCK_G10S_PRO,
         ROBOROCK_Q5,
         ROBOROCK_Q7,
         ROBOROCK_Q7_MAX,

roborock/containers.py

@@ -31,9 +31,11 @@
 from .const import (
     FILTER_REPLACE_TIME,
     MAIN_BRUSH_REPLACE_TIME,
+    ROBOROCK_G10S_PRO,
     ROBOROCK_Q7_MAX,
     ROBOROCK_S4_MAX,
     ROBOROCK_S5_MAX,
+    ROBOROCK_S6,
     ROBOROCK_S6_MAXV,
     ROBOROCK_S6_PURE,
     ROBOROCK_S7,
@@ -306,11 +308,13 @@ class S8ProUltraStatus(Status):
     ROBOROCK_S4_MAX: S4MaxStatus,
     ROBOROCK_S5_MAX: S5MaxStatus,
     ROBOROCK_Q7_MAX: Q7MaxStatus,
+    ROBOROCK_S6: S6PureStatus,
     ROBOROCK_S6_MAXV: S6MaxVStatus,
     ROBOROCK_S6_PURE: S6PureStatus,
     ROBOROCK_S7_MAXV: S7MaxVStatus,
     ROBOROCK_S7: S7Status,
     ROBOROCK_S8_PRO_ULTRA: S8ProUltraStatus,
+    ROBOROCK_G10S_PRO: S7MaxVStatus,
 }