diff --git a/devtools/helpers/smartrequests.py b/devtools/helpers/smartrequests.py index 3db1a2c9..3756cb95 100644 --- a/devtools/helpers/smartrequests.py +++ b/devtools/helpers/smartrequests.py @@ -448,7 +448,7 @@ COMPONENT_REQUESTS = { "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 index c477dbed..50650965 100644 --- 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 index 48378a57..a17859e4 100644 --- a/kasa/smart/modules/__init__.py +++ b/kasa/smart/modules/__init__.py @@ -6,6 +6,7 @@ from .autooff import AutoOff 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 @@ __all__ = [ "Energy", "DeviceModule", "ChildDevice", + "ChildLock", "BatterySensor", "HumiditySensor", "TemperatureSensor", diff --git a/kasa/smart/modules/childlock.py b/kasa/smart/modules/childlock.py new file mode 100644 index 00000000..1c5e72d9 --- /dev/null +++ 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 lock.""" + + 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 lock is enabled.""" + return self.data["child_lock_status"] + + async def set_enabled(self, enabled: bool) -> dict: + """Set child lock.""" + 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 index 2f945c94..c978f89c 100644 --- 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 new file mode 100644 index 00000000..2ffa9104 --- /dev/null +++ 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"}, +) + + +@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