Check device type before setting an potentially unsupported model

JohNan · Aug 26, 2024 · 6cb52ad · 6cb52ad
1 parent dad4512
commit 6cb52ad
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/custom_components/wellbeing/api.py b/custom_components/wellbeing/api.py
@@ -354,17 +354,17 @@ async def async_get_appliances(self) -> Appliances:
             appliance_id = appliance.id
             appliance_name = appliance.name
 
-            app = Appliance(appliance_name, appliance_id, model_name)
             _LOGGER.debug(f"Appliance initial: {appliance.initial_data}")
             _LOGGER.debug(f"Appliance state: {appliance.state}")
 
+            if appliance.device_type != "AIR_PURIFIER":
+                continue
+
+            app = Appliance(appliance_name, appliance_id, model_name)
             app.brand = appliance.brand
             app.serialNumber = appliance.serial_number
             app.device = appliance.device_type
 
-            if app.device != "AIR_PURIFIER":
-                continue
-
             data = appliance.state
             data["status"] = appliance.state_data.get("status", "unknown")
             data["connectionState"] = appliance.state_data.get("connectionState", "unknown")