diff --git a/eufy_robovac/robovac.py b/eufy_robovac/robovac.py index a627af4..535243a 100644 --- a/eufy_robovac/robovac.py +++ b/eufy_robovac/robovac.py @@ -47,8 +47,8 @@ class WorkStatus(StringEnum): STAND_BY = 'standby' # Not in the dock - goes into this state after being paused for a while SLEEPING = 'Sleeping' - # ? Not sure how this differs from charging - RECHARGING = 'Recharge' + # Going home because battery is depleted + RECHARGE_NEEDED = 'Recharge' # In the dock, full charged COMPLETED = 'completed' diff --git a/eufy_robovac/vacuum.py b/eufy_robovac/vacuum.py index 443cffa..2e56964 100644 --- a/eufy_robovac/vacuum.py +++ b/eufy_robovac/vacuum.py @@ -121,8 +121,9 @@ class EufyVacuum(VacuumDevice): return STATE_CLEANING elif self.robovac.work_status == robovac.WorkStatus.CHARGING: return STATE_DOCKED - elif self.robovac.work_status == robovac.WorkStatus.RECHARGING: - return STATE_DOCKED + elif self.robovac.work_status == robovac.WorkStatus.RECHARGE_NEEDED: + # Should be captured by `go_home` above, but just in case + return STATE_RETURNING elif self.robovac.work_status == robovac.WorkStatus.SLEEPING: return STATE_IDLE elif self.robovac.work_status == robovac.WorkStatus.STAND_BY: