Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-22 11:13:34 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Add _check_supported

Browse Source
This commit is contained in:
Teemu Rytilahti 2024-12-05 17:59:50 +01:00
parent c0f5d5b594
commit f3ebb02b10
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
kasa/smart/modules/powerprotection.py
View File

@ -96,3 +96,10 @@ class PowerProtection(SmartModule):
"protection_power": threshold,
}
return await self.call("set_protection_power", params)
async def _check_supported(self) -> bool:
"""Return True if module is supported.
This is needed, as strips like P304M report the status only for children.
"""
return "power_protection_status" in self._device.sys_info

2
tests/smart/modules/test_powerprotection.py
View File

@ -22,7 +22,7 @@ def _skip_on_unavailable(dev: SmartDevice):
@pytest.mark.parametrize(
("feature", "prop_name", "type"),
[
("overloaded", "overloaded", bool | None),
("overloaded", "overloaded", bool),
("power_protection_enabled", "enabled", bool),
("power_protection_threshold", "protection_threshold", int),
],