Add support for waterleak sensor (T300) (#876)

kasa/smart/modules/__init__.py

@@ -19,6 +19,7 @@ from .reportmodule import ReportModule
 from .temperature import TemperatureSensor
 from .temperaturecontrol import TemperatureControl
 from .timemodule import TimeModule
+from .waterleak import WaterleakSensor
 
 __all__ = [
     "AlarmModule",
@@ -40,4 +41,5 @@ __all__ = [
     "LightTransitionModule",
     "ColorTemperatureModule",
     "ColorModule",
+    "WaterleakSensor",
 ]

kasa/smart/modules/waterleak.py

@@ -0,0 +1,62 @@
+"""Implementation of waterleak module."""
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import TYPE_CHECKING
+
+from ...feature import Feature
+from ..smartmodule import SmartModule
+
+if TYPE_CHECKING:
+    from ..smartdevice import SmartDevice
+
+
+class WaterleakStatus(Enum):
+    """Waterleawk status."""
+
+    Normal = "normal"
+    LeakDetected = "water_leak"
+    Drying = "water_dry"
+
+
+class WaterleakSensor(SmartModule):
+    """Implementation of waterleak module."""
+
+    REQUIRED_COMPONENT = "sensor_alarm"
+
+    def __init__(self, device: SmartDevice, module: str):
+        super().__init__(device, module)
+        self._add_feature(
+            Feature(
+                device,
+                "Water leak",
+                container=self,
+                attribute_getter="status",
+                icon="mdi:water",
+            )
+        )
+        self._add_feature(
+            Feature(
+                device,
+                "Water alert",
+                container=self,
+                attribute_getter="alert",
+                icon="mdi:water-alert",
+            )
+        )
+
+    def query(self) -> dict:
+        """Query to execute during the update cycle."""
+        # Water leak information is contained in the main device info response.
+        return {}
+
+    @property
+    def status(self) -> WaterleakStatus:
+        """Return current humidity in percentage."""
+        return WaterleakStatus(self._device.sys_info["water_leak_status"])
+
+    @property
+    def alert(self) -> bool:
+        """Return true if alarm is active."""
+        return self._device.sys_info["in_alarm"]

kasa/smart/smartchilddevice.py

@@ -50,6 +50,7 @@ class SmartChildDevice(SmartDevice):
         child_device_map = {
             "plug.powerstrip.sub-plug": DeviceType.Plug,
             "subg.trigger.temp-hmdt-sensor": DeviceType.Sensor,
+            "subg.trigger.water-leak-sensor": DeviceType.Sensor,
             "kasa.switch.outlet.sub-fan": DeviceType.Fan,
             "kasa.switch.outlet.sub-dimmer": DeviceType.Dimmer,
             "subg.trv": DeviceType.Thermostat,