Removes duplicate sensors

JohNan · Jan 14, 2025 · c777728 · c777728
1 parent 1ae6099
commit c777728
Showing 1 changed file with 26 additions and 31 deletions.
diff --git 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",