mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 03:33:35 +00:00
1ad2a05b65
* Add ChildDevice and ChildProtocolWrapper * Initialize & update children * Fix circular imports * Add dummy_protocol fixture and tests for unwrapping responseData * Use dummy_protocol for existing smartprotocol tests * Move _ChildProtocolWrapper to smartprotocol.py * Use dummy_protocol for test multiple requests * Use device_id instead of position for selecting the child * Fix wrapping for regular requests * Remove unused imports * tweak * rename child_device to childdevice * Fix import
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
"""Child device implementation."""
|
|
from typing import Dict, Optional
|
|
|
|
from ..deviceconfig import DeviceConfig
|
|
from ..exceptions import SmartDeviceException
|
|
from ..smartprotocol import SmartProtocol, _ChildProtocolWrapper
|
|
from .tapodevice import TapoDevice
|
|
|
|
|
|
class ChildDevice(TapoDevice):
|
|
"""Presentation of a child device.
|
|
|
|
This wraps the protocol communications and sets internal data for the child.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
parent: TapoDevice,
|
|
child_id: str,
|
|
config: Optional[DeviceConfig] = None,
|
|
protocol: Optional[SmartProtocol] = None,
|
|
) -> None:
|
|
super().__init__(parent.host, config=parent.config, protocol=parent.protocol)
|
|
self._parent = parent
|
|
self._id = child_id
|
|
self.protocol = _ChildProtocolWrapper(child_id, parent.protocol)
|
|
|
|
async def update(self, update_children: bool = True):
|
|
"""We just set the info here accordingly."""
|
|
|
|
def _get_child_info() -> Dict:
|
|
"""Return the subdevice information for this device."""
|
|
for child in self._parent._last_update["child_info"]["child_device_list"]:
|
|
if child["device_id"] == self._id:
|
|
return child
|
|
|
|
raise SmartDeviceException(
|
|
f"Unable to find child device with position {self._id}"
|
|
)
|
|
|
|
self._last_update = self._sys_info = self._info = _get_child_info()
|
|
|
|
def __repr__(self):
|
|
return f"<ChildDevice {self.alias} of {self._parent}>"
|