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

Make timeout adjustable (#494)

Browse Source
This commit is contained in:
J. Nick Koston
2023-10-07 08:58:00 -10:00
committed by GitHub
parent 20b3f7a771
commit 0ec0826cc7
10 changed files with 39 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
kasa/discover.py
View File

@@ -47,7 +47,8 @@ class _DiscoverProtocol(asyncio.DatagramProtocol):
port: Optional[int] = None,
discovered_event: Optional[asyncio.Event] = None,
credentials: Optional[Credentials] = None,
):
timeout: Optional[int] = None,
) -> None:
self.transport = None
self.discovery_packets = discovery_packets
self.interface = interface
@@ -61,6 +62,7 @@ class _DiscoverProtocol(asyncio.DatagramProtocol):
self.on_unsupported = on_unsupported
self.discovered_event = discovered_event
self.credentials = credentials
self.timeout = timeout
def connection_made(self, transport) -> None:
"""Set socket options for broadcasting."""
@@ -124,7 +126,9 @@ class _DiscoverProtocol(asyncio.DatagramProtocol):
self.discovered_event.set()
return
device = device_class(ip, port=port, credentials=self.credentials)
device = device_class(
ip, port=port, credentials=self.credentials, timeout=self.timeout
)
device.update_from_discover_info(info)
self.discovered_devices[ip] = device
@@ -226,6 +230,7 @@ class Discover:
interface=interface,
on_unsupported=on_unsupported,
credentials=credentials,
timeout=timeout,
),
local_addr=("0.0.0.0", 0),
)
@@ -259,7 +264,11 @@ class Discover:
event = asyncio.Event()
transport, protocol = await loop.create_datagram_endpoint(
lambda: _DiscoverProtocol(
target=host, port=port, discovered_event=event, credentials=credentials
target=host,
port=port,
discovered_event=event,
credentials=credentials,
timeout=timeout,
),
local_addr=("0.0.0.0", 0),
)