Add supported check to light transition module (#971)

sdb9696 · web-flow · commit f0be672cf51f · 2024-06-11T15:46:36.000+01:00
Adds an implementation of `_check_supported` to the light transition module so it is
not added to a parent device that reports it but doesn't support it, i.e. ks240.
diff --git a/kasa/smart/modules/lighttransition.py b/kasa/smart/modules/lighttransition.py
@@ -181,3 +181,13 @@ def query(self) -> dict:
             return {}
         else:
             return {self.QUERY_GETTER_NAME: None}
+
+    async def _check_supported(self):
+        """Additional check to see if the module is supported by the device.
+
+        Parent devices that report components of children such as ks240 will not have
+        the brightness value is sysinfo.
+        """
+        # Look in _device.sys_info here because self.data is either sys_info or
+        # get_preset_rules depending on whether it's a child device or not.
+        return "brightness" in self._device.sys_info
