feat: Enhnaced the wind direciton function agian

Author: jeeftor

homeassistant/components/weatherflow_cloud/sensor.py

@@ -47,26 +47,24 @@
 from .const import DOMAIN
 from .entity import WeatherFlowCloudEntity
 
-
-def _get_wind_direction_icon(degree: int) -> str:
-    """Get the wind direction icon based on the degree."""
-
-    if degree is None or not isinstance(degree, (int, float)):
-        raise ValueError("Degree must be a number")
-
-    # Use integer division to divide into correct sector
-    sector = int(((degree % 360) + 22.5) / 45) % 8
-
-    return [
-        "mdi:arrow-up",  # N   (337.5 - 22.5)
-        "mdi:arrow-top-right",  # NE  (22.5 - 67.5)
-        "mdi:arrow-right",  # E   (67.5 - 112.5)
-        "mdi:arrow-bottom-right",  # SE  (112.5 - 157.5)
-        "mdi:arrow-down",  # S   (157.5 - 202.5)
-        "mdi:arrow-bottom-left",  # SW  (202.5 - 247.5)
-        "mdi:arrow-left",  # W   (247.5 - 292.5)
-        "mdi:arrow-top-left",  # NW  (292.5 - 337.5)
-    ][sector]
+DIRECTION_MAP = {
+    0: "mdi:arrow-up",  # N   (337.5° - 22.5°)
+    1: "mdi:arrow-top-right",  # NE  (22.5° - 67.5°)
+    2: "mdi:arrow-right",  # E   (67.5° - 112.5°)
+    3: "mdi:arrow-bottom-right",  # SE  (112.5° - 157.5°)
+    4: "mdi:arrow-down",  # S   (157.5° - 202.5°)
+    5: "mdi:arrow-bottom-left",  # SW  (202.5° - 247.5°)
+    6: "mdi:arrow-left",  # W   (247.5° - 292.5°)
+    7: "mdi:arrow-top-left",  # NW  (292.5° - 337.5°)
+}
+
+
+def _get_wind_direction_icon(wind_angle: float | None = None) -> str:
+    """Get the wind direction icon based on the wind angle."""
+    if wind_angle is None:
+        return "mdi:compass"
+    sector = int(((wind_angle % 360) + 22.5) / 45) % 8
+    return DIRECTION_MAP[sector]
 
 
 @dataclass(frozen=True, kw_only=True)