Skip to content

Commit a82de33

Browse files
authored
Improve options handling in vacuum wrapper (#57)
1 parent 94e01e6 commit a82de33

File tree

1 file changed

+5
-5
lines changed
  • src/tuya_device_handlers/device_wrapper

1 file changed

+5
-5
lines changed

src/tuya_device_handlers/device_wrapper/vacuum.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -102,14 +102,14 @@ def __init__(
102102
if charge_wrapper or (
103103
mode_wrapper and self._TUYA_MODE_RETURN_HOME in mode_wrapper.options
104104
):
105-
self.options.append("return_to_base")
105+
self.options.append(TuyaVacuumAction.RETURN_TO_BASE)
106106
if locate_wrapper:
107-
self.options.append("locate")
107+
self.options.append(TuyaVacuumAction.LOCATE)
108108
if pause_wrapper:
109-
self.options.append("pause")
109+
self.options.append(TuyaVacuumAction.PAUSE)
110110
if switch_wrapper:
111-
self.options.append("start")
112-
self.options.append("stop")
111+
self.options.append(TuyaVacuumAction.START)
112+
self.options.append(TuyaVacuumAction.STOP)
113113

114114
@classmethod
115115
def find_dpcode(cls, device: CustomerDevice) -> Self:

0 commit comments

Comments
 (0)