Put modules back on children for wall switches (#881)

Puts modules back on the children for `WallSwitches` (i.e. ks240) and
makes them accessible from the `modules` property on the parent.
This commit is contained in:
Steven B
2024-04-29 17:34:20 +01:00
committed by GitHub
parent 6724506fab
commit cb11b36511
8 changed files with 80 additions and 47 deletions

View File

@@ -39,6 +39,7 @@ from kasa.iot import (
IotStrip,
IotWallSwitch,
)
from kasa.iot.modules import Usage
from kasa.smart import SmartBulb, SmartDevice
try:
@@ -829,7 +830,7 @@ async def usage(dev: Device, year, month, erase):
Daily and monthly data provided in CSV format.
"""
echo("[bold]== Usage ==[/bold]")
usage = dev.modules["usage"]
usage = cast(Usage, dev.modules["usage"])
if erase:
echo("Erasing usage statistics..")