1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
from __future__ import annotations
from .code_mappings import RoborockModeEnum
from .device_features import DeviceFeatures
class VacuumModes(RoborockModeEnum):
GENTLE = ("gentle", 105)
OFF = ("off", 105)
QUIET = ("quiet", 101)
BALANCED = ("balanced", 102)
TURBO = ("turbo", 103)
MAX = ("max", 104)
MAX_PLUS = ("max_plus", 108)
CUSTOMIZED = ("custom", 106)
SMART_MODE = ("smart_mode", 110)
class CleanRoutes(RoborockModeEnum):
STANDARD = ("standard", 300)
DEEP = ("deep", 301)
DEEP_PLUS = ("deep_plus", 303)
FAST = ("fast", 304)
DEEP_PLUS_CN = ("deep_plus", 305)
SMART_MODE = ("smart_mode", 306)
CUSTOMIZED = ("custom", 302)
class VacuumModesOld(RoborockModeEnum):
QUIET = ("quiet", 38)
BALANCED = ("balanced", 60)
TURBO = ("turbo", 75)
MAX = ("max", 100)
class WaterModes(RoborockModeEnum):
OFF = ("off", 200)
LOW = ("low", 201)
MILD = ("mild", 201)
MEDIUM = ("medium", 202)
STANDARD = ("standard", 202)
HIGH = ("high", 203)
INTENSE = ("intense", 203)
CUSTOMIZED = ("custom", 204)
CUSTOM = ("custom_water_flow", 207)
EXTREME = ("extreme", 208)
SMART_MODE = ("smart_mode", 209)
class WashTowelModes(RoborockModeEnum):
SMART = ("smart", 10)
LIGHT = ("light", 0)
BALANCED = ("balanced", 1)
DEEP = ("deep", 2)
SUPER_DEEP = ("super_deep", 8)
def get_wash_towel_modes(features: DeviceFeatures) -> list[WashTowelModes]:
"""Get the valid wash towel modes for the device"""
modes = [WashTowelModes.LIGHT, WashTowelModes.BALANCED, WashTowelModes.DEEP]
if features.is_super_deep_wash_supported and not features.is_dirty_replenish_clean_supported:
modes.append(WashTowelModes.SUPER_DEEP)
elif features.is_dirty_replenish_clean_supported:
modes.append(WashTowelModes.SMART)
return modes
def get_clean_modes(features: DeviceFeatures) -> list[VacuumModes]:
"""Get the valid clean modes for the device - also known as 'fan power' or 'suction mode'"""
modes = [VacuumModes.QUIET, VacuumModes.BALANCED, VacuumModes.TURBO, VacuumModes.MAX]
if features.is_max_plus_mode_supported or features.is_none_pure_clean_mop_with_max_plus:
# If the vacuum has max plus mode supported
modes.append(VacuumModes.MAX_PLUS)
if features.is_pure_clean_mop_supported:
# If the vacuum is capable of 'pure mop clean' aka no vacuum
modes.append(VacuumModes.OFF)
else:
# If not, we can add gentle
modes.append(VacuumModes.GENTLE)
if features.is_smart_clean_mode_set_supported:
modes.append(VacuumModes.SMART_MODE)
if features.is_customized_clean_supported:
modes.append(VacuumModes.CUSTOMIZED)
return modes
def get_clean_routes(features: DeviceFeatures, region: str) -> list[CleanRoutes]:
"""The routes that the vacuum will take while mopping"""
if features.is_none_pure_clean_mop_with_max_plus:
return [CleanRoutes.FAST, CleanRoutes.STANDARD]
supported = [CleanRoutes.STANDARD, CleanRoutes.DEEP]
if features.is_careful_slow_mop_supported:
if not (
features.is_corner_clean_mode_supported
and features.is_clean_route_deep_slow_plus_supported
and region == "cn"
):
# for some reason there is a china specific deep plus mode
supported.append(CleanRoutes.DEEP_PLUS_CN)
else:
supported.append(CleanRoutes.DEEP_PLUS)
if features.is_clean_route_fast_mode_supported:
supported.append(CleanRoutes.FAST)
if features.is_smart_clean_mode_set_supported:
supported.append(CleanRoutes.SMART_MODE)
if features.is_customized_clean_supported:
supported.append(CleanRoutes.CUSTOMIZED)
return supported
def get_water_modes(features: DeviceFeatures) -> list[WaterModes]:
"""Get the valid water modes for the device - also known as 'water flow' or 'water level'"""
supported_modes = [WaterModes.OFF]
if features.is_mop_shake_module_supported:
# For mops that have the vibrating mop pad, they do mild standard intense
supported_modes.extend([WaterModes.MILD, WaterModes.STANDARD, WaterModes.INTENSE])
else:
supported_modes.extend([WaterModes.LOW, WaterModes.MEDIUM, WaterModes.HIGH])
if features.is_custom_water_box_distance_supported:
# This is for devices that allow you to set a custom water flow from 0-100
supported_modes.append(WaterModes.CUSTOM)
if features.is_mop_shake_module_supported and features.is_mop_shake_water_max_supported:
supported_modes.append(WaterModes.EXTREME)
if features.is_smart_clean_mode_set_supported:
supported_modes.append(WaterModes.SMART_MODE)
if features.is_customized_clean_supported:
supported_modes.append(WaterModes.CUSTOMIZED)
return supported_modes
def is_smart_mode_set(water_mode: WaterModes, clean_mode: VacuumModes, mop_mode: CleanRoutes) -> bool:
"""Check if the smart mode is set for the given water mode and clean mode"""
return (
water_mode == WaterModes.SMART_MODE
or clean_mode == VacuumModes.SMART_MODE
or mop_mode == CleanRoutes.SMART_MODE
)
|
