From c777728b45cd55ae15c714fd698491a30a28e86c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Johan=20Nenz=C3=A9n?= <johan.nanzen@gmail.com>
Date: Tue, 14 Jan 2025 19:18:44 +0000
Subject: [PATCH] Removes duplicate sensors

---
 custom_components/wellbeing/api.py | 57 ++++++++++++++----------------
 1 file changed, 26 insertions(+), 31 deletions(-)

diff --git a/custom_components/wellbeing/api.py b/custom_components/wellbeing/api.py
index e4291f0..4382cf8 100644
--- a/custom_components/wellbeing/api.py
+++ b/custom_components/wellbeing/api.py
@@ -171,22 +171,6 @@ def _create_entities(data):
         ]
 
         pm700_entities = [
-            ApplianceSensor(
-                name=f"{FILTER_TYPE.get(data.get('FilterType_1', 0), 'Unknown filter')} Life",
-                attr="FilterLife_1",
-                unit=PERCENTAGE,
-            ),
-            ApplianceSensor(
-                name=f"{FILTER_TYPE.get(data.get('FilterType_2', 0), 'Unknown filter')} Life",
-                attr="FilterLife_2",
-                unit=PERCENTAGE,
-            ),
-            
-            ApplianceBinary(
-                name="Ionizer",
-                attr="Ionizer",
-                device_class=BinarySensorDeviceClass.RUNNING,
-            ),
             ApplianceBinary(
                 name="AQI Light",
                 attr="AQILight",
@@ -202,26 +186,19 @@ def _create_entities(data):
                 attr="HumidityTarget",
                 unit=PERCENTAGE,
             ),
-            
-            ApplianceSensor(name="Louver Swing", attr="LouverSwing", device_class=SensorDeviceClass.ENUM),
+            ApplianceSensor(
+                name="Louver Swing",
+                attr="LouverSwing",
+                device_class=SensorDeviceClass.ENUM
+            ),
             ApplianceBinary(
                 name="Empty Water Tray",
                 attr="WaterTrayLevelLow",
                 device_class=BinarySensorDeviceClass.PROBLEM,
             ),
-            
         ]
+
         a7_entities = [
-            ApplianceSensor(
-                name=f"{FILTER_TYPE.get(data.get('FilterType_1', 0), 'Unknown filter')} Life",
-                attr="FilterLife_1",
-                unit=PERCENTAGE,
-            ),
-            ApplianceSensor(
-                name=f"{FILTER_TYPE.get(data.get('FilterType_2', 0), 'Unknown filter')} Life",
-                attr="FilterLife_2",
-                unit=PERCENTAGE,
-            ),
             ApplianceSensor(
                 name="State",
                 attr="State",
@@ -291,6 +268,16 @@ def _create_entities(data):
         ]
 
         common_entities = [
+            ApplianceSensor(
+                name=f"{FILTER_TYPE.get(data.get('FilterType_1', 0), 'Unknown filter')} Life",
+                attr="FilterLife_1",
+                unit=PERCENTAGE,
+            ),
+            ApplianceSensor(
+                name=f"{FILTER_TYPE.get(data.get('FilterType_2', 0), 'Unknown filter')} Life",
+                attr="FilterLife_2",
+                unit=PERCENTAGE,
+            ),
             ApplianceFan(
                 name="Fan Speed",
                 attr="Fanspeed",
@@ -347,8 +334,16 @@ def _create_entities(data):
                 device_class=SensorDeviceClass.ENUM,
                 entity_category=EntityCategory.DIAGNOSTIC,
             ),
-            ApplianceBinary(name="Ionizer", attr="Ionizer"),
-            ApplianceBinary(name="UI Light", attr="UILight"),
+            ApplianceBinary(
+                name="Ionizer",
+                attr="Ionizer",
+                device_class=BinarySensorDeviceClass.RUNNING,
+            ),
+            ApplianceBinary(
+                name="UI Light",
+                attr="UILight",
+                device_class=BinarySensorDeviceClass.LIGHT,
+            ),
             ApplianceBinary(
                 name="Door Open",
                 attr="DoorOpen",