From 230fca0f6ff9b5a7b6b3a29780820e9be1ef5022 Mon Sep 17 00:00:00 2001
From: Michael Graf <michael@trunner.de>
Date: Sat, 15 May 2021 20:34:10 +0200
Subject: [PATCH] Adding RETURNING to ATTR_STATUS #1

---
 __init__.py | 2 ++
 const.py    | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/__init__.py b/__init__.py
index 314c9a1..0c4deb9 100644
--- a/__init__.py
+++ b/__init__.py
@@ -279,6 +279,8 @@ class VorwerkState:
             status = self._cleaning_status()
         elif self.state == STATE_PAUSED:
             status = "Paused"
+        elif self.state == STATE_RETURNING:
+            status = "Returning"
 
         return status
 
diff --git a/const.py b/const.py
index 83eba21..e255fe8 100644
--- a/const.py
+++ b/const.py
@@ -54,7 +54,7 @@ ROBOT_CLEANING_ACTIONS = [
     ROBOT_ACTION_SUSPENDED_CLEANING,
     ROBOT_ACTION_MAP_CLEANING,
     ROBOT_ACTION_EXPLORING_MAP,
-    ROBOT_ACTION_SUSPENDED_EXPLORATION
+    ROBOT_ACTION_SUSPENDED_EXPLORATION,
 ]
 
 ACTION = {