feat: enhance weather attribute normalization to support wind gust data across all API endpoints

src/pyatmo/account.py

@@ -235,8 +235,8 @@ async def update_devices(
                         module_data["home_id"] = home_id
                         module_data["id"] = module_data["_id"]
                         module_data["name"] = module_data.get("module_name")
-                        modules_data.append(normalize_weather_attributes(module_data))
-                    modules_data.append(normalize_weather_attributes(device_data))
+                        modules_data.append(module_data)
+                    modules_data.append(device_data)
 
                     self.homes[home_id] = Home(
                         self.auth,
@@ -247,7 +247,7 @@ async def update_devices(
                         },
                     )
                 await self.homes[home_id].update(
-                    {HOME: {"modules": [normalize_weather_attributes(device_data)]}},
+                    {HOME: {"modules": [device_data]}},
                 )
             else:
                 LOG.debug("No home %s (%s) found.", home_id, home_id)
@@ -264,7 +264,6 @@ async def update_devices(
                     "station_name",
                     device_data.get("module_name", "Unknown"),
                 )
-                device_data = normalize_weather_attributes(device_data)
                 if device_data["id"] not in self.modules:
                     self.modules[device_data["id"]] = getattr(
                         modules,
@@ -293,33 +292,3 @@ def find_home_of_device(self, device_data: dict[str, Any]) -> str | None:
             ),
             None,
         )
-
-
-ATTRIBUTES_TO_FIX: dict[str, str] = {
-    "_id": "id",
-    "firmware": "firmware_revision",
-    "wifi_status": "wifi_strength",
-    "rf_status": "rf_strength",
-    "Temperature": "temperature",
-    "Humidity": "humidity",
-    "Pressure": "pressure",
-    "CO2": "co2",
-    "AbsolutePressure": "absolute_pressure",
-    "Noise": "noise",
-    "Rain": "rain",
-    "WindStrength": "wind_strength",
-    "WindAngle": "wind_angle",
-    "GustStrength": "gust_strength",
-    "GustAngle": "gust_angle",
-}
-
-
-def normalize_weather_attributes(raw_data: RawData) -> dict[str, Any]:
-    """Normalize weather attributes."""
-    result: dict[str, Any] = {}
-    for attribute, value in raw_data.items():
-        if attribute == "dashboard_data":
-            result.update(**normalize_weather_attributes(value))
-        else:
-            result[ATTRIBUTES_TO_FIX.get(attribute, attribute)] = value
-    return result

src/pyatmo/helpers.py

@@ -12,6 +12,57 @@
 
 LOG: logging.Logger = logging.getLogger(__name__)
 
+# types for data that can be normalized (recursive structures) - to satisfy mypy
+NormalizableData = dict[str, Any] | list[Any] | str | int | float | bool | None
+
+ATTRIBUTES_TO_FIX: dict[str, str] = {
+    "firmware": "firmware_revision",
+    "firmware_revision": "firmware_revision",
+    "firmware_name": "firmware_name",
+    "wifi_status": "wifi_strength",
+    "rf_status": "rf_strength",
+    "Temperature": "temperature",
+    "Humidity": "humidity",
+    "Pressure": "pressure",
+    "CO2": "co2",
+    "AbsolutePressure": "absolute_pressure",
+    "Noise": "noise",
+    "Rain": "rain",
+    "WindStrength": "wind_strength",
+    "WindAngle": "wind_angle",
+    "GustStrength": "gust_strength",
+    "GustAngle": "gust_angle",
+    "wind_gust": "gust_strength",
+    "wind_gust_angle": "gust_angle",
+}
+
+
+def normalize_weather_attributes(raw_data: NormalizableData) -> NormalizableData:
+    """Normalize weather-related attributes recursively."""
+
+    if isinstance(raw_data, list):
+        return [normalize_weather_attributes(item) for item in raw_data]
+
+    if not isinstance(raw_data, dict):
+        return raw_data
+
+    normalized: dict[str, Any] = {}
+    for key, value in raw_data.items():
+        if key == "_id":
+            normalized["_id"] = value
+            continue
+        if key == "dashboard_data" and isinstance(value, dict):
+            # useless cast here to satisfy mypy
+            normalized |= cast("dict[str, Any]", normalize_weather_attributes(value))
+            continue
+
+        mapped_key = ATTRIBUTES_TO_FIX[key] if key in ATTRIBUTES_TO_FIX else key  # noqa: SIM401
+        normalized[mapped_key] = normalize_weather_attributes(value)
+
+    if "_id" in normalized and "id" not in normalized:
+        normalized["id"] = normalized["_id"]
+    return normalized
+
 
 def fix_id(raw_data: list[RawData | str]) -> list[RawData | str]:
     """Fix known errors in station ids like superfluous spaces."""
@@ -43,20 +94,23 @@ def extract_raw_data(resp: RawData, tag: str) -> RawData:
         msg = "No device found, errors in response"
         raise NoDeviceError(msg)
 
+    # useless cast here again to satisfy mypy
+    body = cast("dict[str, Any]", normalize_weather_attributes(resp["body"]))
+
     if tag == "homes":
-        homes: list[dict[str, Any] | str] = fix_id(resp["body"].get(tag))
+        homes: list[dict[str, Any] | str] = fix_id(body.get(tag, []))
         if not homes:
             LOG.debug("Server response (tag: %s): %s", tag, resp)
             msg = "No homes found"
             raise NoDeviceError(msg)
         return {
             tag: homes,
-            "errors": resp["body"].get("errors", []),
+            "errors": body.get("errors", []),
         }
 
-    if not (raw_data := fix_id(resp["body"].get(tag))):
+    if not (raw_data := fix_id(body.get(tag, []))):
         LOG.debug("Server response (tag: %s): %s", tag, resp)
         msg = "No device data available"
         raise NoDeviceError(msg)
 
-    return {tag: raw_data, "errors": resp["body"].get("errors", [])}
+    return {tag: raw_data, "errors": body.get("errors", [])}