python-kasa · rytilahti · Jan 17, 2025 · Jan 16, 2025 · Jan 17, 2025 · Jan 17, 2025
diff --git a/devtools/helpers/smartrequests.py b/devtools/helpers/smartrequests.py
@@ -448,7 +448,7 @@ def get_component_requests(component_id, ver_code):
     "battery": [SmartRequest.get_raw_request("getBatteryInfo")],
     "consumables": [SmartRequest.get_raw_request("getConsumablesInfo")],
     "direction_control": [],
-    "button_and_led": [],
+    "button_and_led": [SmartRequest.get_raw_request("getChildLockInfo")],
     "speaker": [
         SmartRequest.get_raw_request("getSupportVoiceLanguage"),
         SmartRequest.get_raw_request("getCurrentVoiceLanguage"),

diff --git a/kasa/module.py b/kasa/module.py
@@ -152,6 +152,7 @@ class Module(ABC):
     ChildProtection: Final[ModuleName[smart.ChildProtection]] = ModuleName(
         "ChildProtection"
     )
+    ChildLock: Final[ModuleName[smart.ChildLock]] = ModuleName("ChildLock")
     TriggerLogs: Final[ModuleName[smart.TriggerLogs]] = ModuleName("TriggerLogs")
 
     HomeKit: Final[ModuleName[smart.HomeKit]] = ModuleName("HomeKit")

diff --git a/kasa/smart/modules/__init__.py b/kasa/smart/modules/__init__.py
@@ -6,6 +6,7 @@
 from .batterysensor import BatterySensor
 from .brightness import Brightness
 from .childdevice import ChildDevice
+from .childlock import ChildLock
 from .childprotection import ChildProtection
 from .clean import Clean
 from .cloud import Cloud
@@ -45,6 +46,7 @@
     "Energy",
     "DeviceModule",
     "ChildDevice",
+    "ChildLock",
     "BatterySensor",
     "HumiditySensor",
     "TemperatureSensor",

diff --git a/kasa/smart/modules/childlock.py b/kasa/smart/modules/childlock.py
@@ -0,0 +1,37 @@
+"""Child lock module."""
+
+from __future__ import annotations
+
+from ...feature import Feature
+from ..smartmodule import SmartModule
+
+
+class ChildLock(SmartModule):
+    """Implementation for child_protection."""
+
+    REQUIRED_COMPONENT = "button_and_led"
+    QUERY_GETTER_NAME = "getChildLockInfo"
+
+    def _initialize_features(self) -> None:
+        """Initialize features after the initial update."""
+        self._add_feature(
+            Feature(
+                device=self._device,
+                id="child_lock",
+                name="Child lock",
+                container=self,
+                attribute_getter="enabled",
+                attribute_setter="set_enabled",
+                type=Feature.Type.Switch,
+                category=Feature.Category.Config,
+            )
+        )
+
+    @property
+    def enabled(self) -> bool:
+        """Return True if child protection is enabled."""
+        return self.data["child_lock_status"]
+
+    async def set_enabled(self, enabled: bool) -> dict:
+        """Set child protection."""
+        return await self.call("setChildLockInfo", {"child_lock_status": enabled})
diff --git a/tests/fixtures/smart/RV20 Max Plus(EU)_1.0_1.0.7.json b/tests/fixtures/smart/RV20 Max Plus(EU)_1.0_1.0.7.json
@@ -165,6 +165,9 @@
     "getCarpetClean": {
         "carpet_clean_prefer": "boost"
     },
+    "getChildLockInfo": {
+        "child_lock_status": false
+    },
     "getCleanAttr": {
         "cistern": 2,
         "clean_number": 1,

diff --git a/tests/smart/modules/test_childlock.py b/tests/smart/modules/test_childlock.py
@@ -0,0 +1,44 @@
+import pytest
+
+from kasa import Module
+from kasa.smart.modules import ChildLock
+
+from ...device_fixtures import parametrize
+
+childlock = parametrize(
+    "has child lock",
+    component_filter="button_and_led",
+    protocol_filter={"SMART.CHILD"},
+)
+
+
+@childlock
+@pytest.mark.parametrize(
+    ("feature", "prop_name", "type"),
+    [
+        ("child_lock", "enabled", bool),
+    ],
+)
+async def test_features(dev, feature, prop_name, type):
+    """Test that features are registered and work as expected."""
+    protect: ChildLock = dev.modules[Module.ChildLock]
+    assert protect is not None
+
+    prop = getattr(protect, prop_name)
+    assert isinstance(prop, type)
+
+    feat = protect._device.features[feature]
+    assert feat.value == prop
+    assert isinstance(feat.value, type)
+
+
+@childlock
+async def test_enabled(dev):
+    """Test the API."""
+    protect: ChildLock = dev.modules[Module.ChildLock]
+    assert protect is not None
+
+    assert isinstance(protect.enabled, bool)
+    await protect.set_enabled(False)
+    await dev.update()
+    assert protect.enabled is False