Fallback to unknown filter type if not defined for A7 (#71)

JohNan · Jan 14, 2024 · 48c704b · 48c704b
1 parent 2cff817
commit 48c704b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/wellbeing/api.py b/custom_components/wellbeing/api.py
@@ -124,12 +124,12 @@ def _create_entities(data):
                 device_class=SensorDeviceClass.CO2
             ),
             ApplianceSensor(
-                name=f"{FILTER_TYPE[data.get('FilterType_1', 0)]} Life",
+                name=f"{FILTER_TYPE.get(data.get('FilterType_1', 0), 'Unknown filter')} Life",
                 attr='FilterLife_1',
                 unit=PERCENTAGE
             ),
             ApplianceSensor(
-                name=f"{FILTER_TYPE[data.get('FilterType_2', 0)]} Life",
+                name=f"{FILTER_TYPE.get(data.get('FilterType_2', 0), 'Unknown filter')} Life",
                 attr='FilterLife_2',
                 unit=PERCENTAGE
             ),
@@ -145,7 +145,7 @@ def _create_entities(data):
 
         a9_entities = [
             ApplianceSensor(
-                name=f"{FILTER_TYPE.get(data.get('FilterType', 0), 'Filter')} Life",
+                name=f"{FILTER_TYPE.get(data.get('FilterType', 0), 'Unknown filter')} Life",
                 attr='FilterLife',
                 unit=PERCENTAGE
             ),