diff --git a/eufy_robovac/robovac.py b/eufy_robovac/robovac.py index 59f0503..38df677 100644 --- a/eufy_robovac/robovac.py +++ b/eufy_robovac/robovac.py @@ -44,6 +44,7 @@ class WorkStatus(StringEnum): STAND_BY = 'standby' SLEEPING = 'Sleeping' RECHARGING = 'Recharge' + COMPLETED = 'completed' class CleanSpeed(StringEnum): diff --git a/eufy_robovac/vacuum.py b/eufy_robovac/vacuum.py index 9de082e..c858129 100644 --- a/eufy_robovac/vacuum.py +++ b/eufy_robovac/vacuum.py @@ -71,12 +71,12 @@ class EufyVacuum(VacuumDevice): self._name = device_config['name'] async def async_update(self): - """Synchronise state from the bulb.""" + """Synchronise state from the vacuum.""" await self.robovac.async_get() @property def unique_id(self): - """Return the ID of this light.""" + """Return the ID of this vacuum.""" return self._device_id @property @@ -127,6 +127,8 @@ class EufyVacuum(VacuumDevice): return STATE_DOCKED elif self.robovac.work_status == robovac.WorkStatus.STAND_BY: return STATE_IDLE + elif self.robovac.work_status == robovac.WorkStatus.COMPLETED: + return STATE_IDLE @property def available(self) -> bool: