Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2025-10-19 22:08:02 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Allow calling update directly on child devices and skipping updates on the parent

Browse Source
This commit is contained in:
sdb9696
2024-06-28 19:25:39 +01:00
parent 2a62849987
commit ec1082a228
5 changed files with 124 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
kasa/iot/iotstrip.py
View File

@@ -121,7 +121,18 @@ class IotStrip(IotDevice):
"""Return if any of the outlets are on."""
return any(plug.is_on for plug in self.children)
async def update(self, update_children: bool = True):
async def update(self, update_children: bool = True, update_parent: bool = True):
"""Update some of the attributes.
Needed for methods that are decorated with `requires_update`.
"""
await self._update(update_children)
async def _update(
self,
update_children: bool = True,
called_from_child: IotStripPlug | None = None,
):
"""Update some of the attributes.
Needed for methods that are decorated with `requires_update`.
@@ -143,9 +154,11 @@ class IotStrip(IotDevice):
for child in self._children.values():
await child._initialize_modules()
if update_children:
for plug in self.children:
await plug.update()
if called_from_child:
await called_from_child._update()
elif update_children:
for child in self._children.values():
await child._update()
if not self.features:
await self._initialize_features()
@@ -355,7 +368,17 @@ class IotStripPlug(IotPlug):
for module_feat in module._module_features.values():
self._add_feature(module_feat)
async def update(self, update_children: bool = True):
async def update(self, update_children: bool = True, update_parent: bool = True):
"""Query the device to update the data.
Needed for properties that are decorated with `requires_update`.
"""
if update_parent:
await self.parent._update(update_children=False, called_from_child=self)
else:
await self._update()
async def _update(self):
"""Query the device to update the data.
Needed for properties that are decorated with `requires_update`.