Simplify availability property in WLED (home-assistant#157800)

Co-authored-by: mik-laj <[email protected]>

Author: mik-laj

homeassistant/components/wled/light.py

@@ -150,12 +150,9 @@ def __init__(
     @property
     def available(self) -> bool:
         """Return True if entity is available."""
-        try:
-            self.coordinator.data.state.segments[self._segment]
-        except KeyError:
-            return False
-
-        return super().available
+        return (
+            super().available and self._segment in self.coordinator.data.state.segments
+        )
 
     @property
     def rgb_color(self) -> tuple[int, int, int] | None:

homeassistant/components/wled/number.py

@@ -97,12 +97,9 @@ def __init__(
     @property
     def available(self) -> bool:
         """Return True if entity is available."""
-        try:
-            self.coordinator.data.state.segments[self._segment]
-        except KeyError:
-            return False
-
-        return super().available
+        return (
+            super().available and self._segment in self.coordinator.data.state.segments
+        )
 
     @property
     def native_value(self) -> float | None:

homeassistant/components/wled/quality_scale.yaml

@@ -31,10 +31,6 @@ rules:
   config-entry-unloading: done
   docs-configuration-parameters: done
   docs-installation-parameters: todo
-  entity-unavailable:
-    status: todo
-    comment: |
-      The WLEDSegmentLight.available property can just be an if .. in .. check
   integration-owner: done
   log-when-unavailable: done
   parallel-updates: done

homeassistant/components/wled/select.py

@@ -173,12 +173,9 @@ def __init__(self, coordinator: WLEDDataUpdateCoordinator, segment: int) -> None
     @property
     def available(self) -> bool:
         """Return True if entity is available."""
-        try:
-            self.coordinator.data.state.segments[self._segment]
-        except KeyError:
-            return False
-
-        return super().available
+        return (
+            super().available and self._segment in self.coordinator.data.state.segments
+        )
 
     @property
     def current_option(self) -> str | None:

homeassistant/components/wled/switch.py

@@ -167,12 +167,9 @@ def __init__(self, coordinator: WLEDDataUpdateCoordinator, segment: int) -> None
     @property
     def available(self) -> bool:
         """Return True if entity is available."""
-        try:
-            self.coordinator.data.state.segments[self._segment]
-        except KeyError:
-            return False
-
-        return super().available
+        return (
+            super().available and self._segment in self.coordinator.data.state.segments
+        )
 
     @property
     def is_on(self) -> bool: