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

src/pyatmo/account.py

@@ -18,7 +18,7 @@
     SETSTATE_ENDPOINT,
     RawData,
 )
-from pyatmo.helpers import extract_raw_data
+from pyatmo.helpers import extract_raw_data, normalize_weather_attributes
 from pyatmo.home import Home
 from pyatmo.modules.module import Energy, MeasureInterval, Module
 
@@ -293,33 +293,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,55 @@
 
 LOG: logging.Logger = logging.getLogger(__name__)
 
+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: RawData) -> RawData:
+    """Normalize weather-related attributes recursively."""
+
+    if not isinstance(raw_data, dict):
+        return (
+            [normalize_weather_attributes(item) for item in raw_data]
+            if isinstance(raw_data, list)
+            else raw_data
+        )
+    normalized: dict[str, Any] = {}
+    has_internal_id = "_id" in raw_data
+    for key, value in raw_data.items():
+        if key == "_id":
+            normalized["_id"] = value
+            if "id" not in raw_data:
+                normalized.setdefault("id", value)
+            continue
+        if key == "dashboard_data" and isinstance(value, dict):
+            normalized |= normalize_weather_attributes(value)
+            continue
+        normalized[ATTRIBUTES_TO_FIX.get(key, key)] = normalize_weather_attributes(
+            value
+        )
+    if has_internal_id and "id" not in normalized:
+        normalized["id"] = raw_data["_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 +92,22 @@ def extract_raw_data(resp: RawData, tag: str) -> RawData:
         msg = "No device found, errors in response"
         raise NoDeviceError(msg)
 
+    body = 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", [])}

src/pyatmo/home.py

@@ -25,6 +25,7 @@
     InvalidStateError,
     NoScheduleError,
 )
+from pyatmo.helpers import normalize_weather_attributes
 from pyatmo.person import Person
 from pyatmo.room import Room
 from pyatmo.schedule import Schedule
@@ -174,6 +175,7 @@ async def update(
 
         has_an_update = False
         for module in data.get("modules", []):
+            module = normalize_weather_attributes(module)
             has_an_update = True
             if module["id"] not in self.modules:
                 self.update_topology({"modules": [module]})