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

Handle paging of partial responses of lists like child_device_info (#862)

Browse Source 
When devices have lists greater than 10 for child devices only the first
10 are returned. This retrieves the rest of the items (currently with
single requests rather than multiple requests)
This commit is contained in:
Steven B
2024-04-24 19:32:30 +01:00
committed by GitHub
parent eff8db450d
commit 53b84b7683
3 changed files with 131 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
kasa/tests/fakeprotocol_smart.py
View File

@@ -21,7 +21,14 @@ class FakeSmartProtocol(SmartProtocol):
class FakeSmartTransport(BaseTransport):
def __init__(self, info, fixture_name):
def __init__(
self,
info,
fixture_name,
*,
list_return_size=10,
component_nego_not_included=False,
):
super().__init__(
config=DeviceConfig(
"127.0.0.123",
@@ -33,10 +40,12 @@ class FakeSmartTransport(BaseTransport):
)
self.fixture_name = fixture_name
self.info = copy.deepcopy(info)
self.components = {
comp["id"]: comp["ver_code"]
for comp in self.info["component_nego"]["component_list"]
}
if not component_nego_not_included:
self.components = {
comp["id"]: comp["ver_code"]
for comp in self.info["component_nego"]["component_list"]
}
self.list_return_size = list_return_size
@property
def default_port(self):
@@ -177,7 +186,20 @@ class FakeSmartTransport(BaseTransport):
elif method == "component_nego" or method[:4] == "get_":
if method in info:
result = copy.deepcopy(info[method])
if "start_index" in result and "sum" in result:
list_key = next(
iter([key for key in result if isinstance(result[key], list)])
)
start_index = (
start_index
if (params and (start_index := params.get("start_index")))
else 0
)
result[list_key] = result[list_key][
start_index : start_index + self.list_return_size
]
return {"result": result, "error_code": 0}
if (
# FIXTURE_MISSING is for service calls not in place when
# SMART fixtures started to be generated