From 9f1e6613f8ce1c3b3c2076c0fb0387520abce665 Mon Sep 17 00:00:00 2001
From: Richard Mitchell <richard.j.mitchell@gmail.com>
Date: Thu, 18 Apr 2019 15:59:50 +0100
Subject: [PATCH] Add completed work status

---
 eufy_robovac/robovac.py | 1 +
 eufy_robovac/vacuum.py  | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

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: