From c4b286c66a2211a5c8402dc2515219785677dc4c Mon Sep 17 00:00:00 2001 From: David Rapan Date: Mon, 10 Feb 2025 15:36:55 +0100 Subject: [PATCH] fix: Discovery tool --- tools/discovery.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tools/discovery.py b/tools/discovery.py index baf5af72..fa705df6 100644 --- a/tools/discovery.py +++ b/tools/discovery.py @@ -35,7 +35,7 @@ async def main(): wait = True try: - transport, protocol = await loop.create_datagram_endpoint(lambda: DiscoveryProtocol(addresses), family = socket.AF_INET, allow_broadcast = True), + transport, protocol = await loop.create_datagram_endpoint(lambda: DiscoveryProtocol(addresses), family = socket.AF_INET, allow_broadcast = True) r = None while r is None or wait: r = await asyncio.wait_for(protocol.responses.get(), DISCOVERY_TIMEOUT) @@ -46,4 +46,5 @@ async def main(): finally: transport.close() -asyncio.run(main()) +if __name__ == "__main__": + asyncio.run(main())