diff --git a/roborock/api.py b/roborock/api.py index 59bf24c..de5aa1c 100644 --- a/roborock/api.py +++ b/roborock/api.py @@ -75,6 +75,11 @@ RoborockCommand.GET_MULTI_MAP, ] RT = TypeVar("RT", bound=RoborockBase) +WASH_N_FILL_DOCK = [ + RoborockDockTypeCode.empty_wash_fill_dock, + RoborockDockTypeCode.s8_dock, + RoborockDockTypeCode.p10_dock, +] def md5hex(message: str) -> str: @@ -454,7 +459,7 @@ async def get_dock_summary(self, dock_type: RoborockDockTypeCode) -> DockSummary DustCollectionMode | WashTowelMode | SmartWashParams | None, ] ] = [self.get_dust_collection_mode()] - if dock_type == RoborockDockTypeCode.empty_wash_fill_dock or dock_type == RoborockDockTypeCode.s8_dock: + if dock_type in WASH_N_FILL_DOCK: commands += [ self.get_wash_towel_mode(), self.get_smart_wash_params(), diff --git a/roborock/code_mappings.py b/roborock/code_mappings.py index 3d3ab46..fb5ed51 100644 --- a/roborock/code_mappings.py +++ b/roborock/code_mappings.py @@ -245,7 +245,7 @@ class RoborockDockTypeCode(RoborockEnum): auto_empty_dock_pure = 5 s7_max_ultra_dock = 6 s8_dock = 7 - q_revo = 8 + p10_dock = 8 class RoborockDockDustCollectionModeCode(RoborockEnum):