From 8a9eff2bb1bd76c09a80f3bf9a89a7f852426935 Mon Sep 17 00:00:00 2001
From: Michael Graf <michael@trunner.de>
Date: Wed, 26 May 2021 17:15:49 +0200
Subject: [PATCH] Fix vacuum won't start when docked #4

---
 manifest.json | 2 +-
 vacuum.py     | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/manifest.json b/manifest.json
index 70b5f74..866e863 100644
--- a/manifest.json
+++ b/manifest.json
@@ -13,5 +13,5 @@
     "http"
   ],
   "iot_class": "cloud_polling",
-  "version": "0.9.0"
+  "version": "0.9.2"
 }
\ No newline at end of file
diff --git a/vacuum.py b/vacuum.py
index 113c1f6..1b4c565 100644
--- a/vacuum.py
+++ b/vacuum.py
@@ -11,6 +11,7 @@ import voluptuous as vol
 from homeassistant.components.vacuum import (
     ATTR_STATUS,
     STATE_CLEANING,
+    STATE_DOCKED,
     STATE_IDLE,
     STATE_PAUSED,
     SUPPORT_BATTERY,
@@ -155,7 +156,7 @@ class VorwerkConnectedVacuum(CoordinatorEntity, StateVacuumEntity):
         if not self._state:
             return
         try:
-            if self._state.state == STATE_IDLE:
+            if self._state.state == STATE_IDLE or self._state.state == STATE_DOCKED:
                 self.robot.start_cleaning()
             elif self._state.state == STATE_PAUSED:
                 self.robot.resume_cleaning()