From c56b3b79dcf186d02b7693268f6eae07416b6ced Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Jald=C3=A9n?= <jalden@kth.se>
Date: Sat, 5 Oct 2024 20:38:18 +0200
Subject: [PATCH] Added separate sensor to pass along raw robotStatus to HA
 (#149)

---
 custom_components/wellbeing/api.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/custom_components/wellbeing/api.py b/custom_components/wellbeing/api.py
index f3ec819..2c9f877 100644
--- a/custom_components/wellbeing/api.py
+++ b/custom_components/wellbeing/api.py
@@ -201,14 +201,19 @@ def _create_entities(data):
         ]
 
         purei9_entities = [
+            ApplianceVacuum(
+                name=data.get("applianceName","Vacuum"),
+                attr="robotStatus",
+            ),
             ApplianceSensor( 
                 name="Dustbin Status",
                 attr="dustbinStatus",
                 device_class=SensorDeviceClass.ENUM,
             ),
-            ApplianceVacuum(
-                name=data.get("applianceName","Vacuum"),
+            ApplianceSensor(
+                name="Robot Status",
                 attr="robotStatus",
+                device_class=SensorDeviceClass.ENUM,
             ),
         ]