Merge pull request #73 from andrey-git/pure-pm25-value

Fix pure pm25 value
andrey-git · Aug 18, 2024 · 24b09b2 · 24b09b2
2 parents 587159d + 143a06a
commit 24b09b2
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/pysensibo/__init__.py b/pysensibo/__init__.py
@@ -240,6 +240,12 @@ async def async_get_devices_data(self) -> SensiboData:  # noqa: C901
             pure_geo_integration: bool | None = None
             pure_measure_integration: bool | None = None
             pure_prime_integration: bool | None = None
+
+            # Pure devices has special handling for PM2.5 value
+            # as it represents an AQI value
+            pm25 = None
+            pm25_pure = None
+
             if dev["productModel"] == "pure":
                 pure_boost_enabled = pure_conf.get("enabled", False)
                 pure_sensitivity = pure_conf.get("sensitivity", "n").lower()
@@ -249,12 +255,10 @@ async def async_get_devices_data(self) -> SensiboData:  # noqa: C901
                     "measurements_integration", False
                 )
                 pure_prime_integration = pure_conf.get("prime_integration", False)
-            if dev["productModel"] == "pure" and (pm25 := measure.get("pm25")):
-                pm25 = None
-                pm25_pure = PureAQI(pm25)
+                if (_pm25 := measure.get("pm25")) is not None:
+                    pm25_pure = PureAQI(_pm25)
             else:
                 pm25 = measure.get("pm25")
-                pm25_pure = None
 
             # Binary sensors for main device
             room_occupied = dev["roomIsOccupied"]