fix base class for new models

kongo09 · Jan 10, 2024 · 9024a3f · 9024a3f
1 parent 84148c6
commit 9024a3f
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/custom_components/philips_airpurifier_coap/philips.py b/custom_components/philips_airpurifier_coap/philips.py
@@ -183,7 +183,10 @@ def __init__(self, coordinator: Coordinator) -> None:  # noqa: D107
         self._name = list(
             filter(
                 None,
-                map(coordinator.status.get, [PhilipsApi.NAME, PhilipsApi.NEW_NAME]),
+                map(
+                    coordinator.status.get,
+                    [PhilipsApi.NAME, PhilipsApi.NEW_NAME, PhilipsApi.NEW2_NAME],
+                ),
             )
         )[0]
         # self._modelName = coordinator.status["modelid"]
@@ -192,7 +195,11 @@ def __init__(self, coordinator: Coordinator) -> None:  # noqa: D107
                 None,
                 map(
                     coordinator.status.get,
-                    [PhilipsApi.MODEL_ID, PhilipsApi.NEW_MODEL_ID],
+                    [
+                        PhilipsApi.MODEL_ID,
+                        PhilipsApi.NEW_MODEL_ID,
+                        PhilipsApi.NEW2_MODEL_ID,
+                    ],
                 ),
             )
         )[0]