Fix vacuum won't start when docked #4

This commit is contained in:
Michael Graf 2021-12-28 12:35:55 +01:00
parent 3a9297bc81
commit d00e44ca5d

View File

@ -11,6 +11,7 @@ import voluptuous as vol
from homeassistant.components.vacuum import ( from homeassistant.components.vacuum import (
ATTR_STATUS, ATTR_STATUS,
STATE_CLEANING, STATE_CLEANING,
STATE_DOCKED,
STATE_IDLE, STATE_IDLE,
STATE_PAUSED, STATE_PAUSED,
SUPPORT_BATTERY, SUPPORT_BATTERY,
@ -176,7 +177,7 @@ class VorwerkConnectedVacuum(CoordinatorEntity, StateVacuumEntity):
def return_to_base(self, **kwargs: Any) -> None: def return_to_base(self, **kwargs: Any) -> None:
"""Set the vacuum cleaner to return to the dock.""" """Set the vacuum cleaner to return to the dock."""
try: try:
if self._state.state == STATE_CLEANING: if self._state.state == STATE_CLEANING or self._state.state == STATE_DOCKED:
self.robot.pause_cleaning() self.robot.pause_cleaning()
self.robot.send_to_base() self.robot.send_to_base()
except NeatoRobotException as ex: except NeatoRobotException as ex: