Skip to content
7 changes: 6 additions & 1 deletion roborock/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -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(),
Expand Down
2 changes: 1 addition & 1 deletion roborock/code_mappings.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down