From 4620142ed1e782ebfcac0e9a8c8d31daa08f8fef Mon Sep 17 00:00:00 2001 From: mystical-prog Date: Fri, 18 Oct 2024 22:35:06 +0530 Subject: [PATCH] ran make lint --- examples/ping/ping.py | 37 ++++++++++++++++++++++++------------- 1 file changed, 24 insertions(+), 13 deletions(-) diff --git a/examples/ping/ping.py b/examples/ping/ping.py index 7ec68092..3b99f78a 100644 --- a/examples/ping/ping.py +++ b/examples/ping/ping.py @@ -1,18 +1,26 @@ import argparse -import sys import multiaddr import trio -from libp2p import new_host -from libp2p.network.stream.net_stream_interface import INetStream -from libp2p.peer.peerinfo import info_from_p2p_addr -from libp2p.typing import TProtocol +from libp2p import ( + new_host, +) +from libp2p.network.stream.net_stream_interface import ( + INetStream, +) +from libp2p.peer.peerinfo import ( + info_from_p2p_addr, +) +from libp2p.typing import ( + TProtocol, +) PING_PROTOCOL_ID = TProtocol("/ipfs/ping/1.0.0") PING_LENGTH = 32 RESP_TIMEOUT = 60 + async def handle_ping(stream: INetStream) -> None: while True: try: @@ -27,11 +35,12 @@ async def handle_ping(stream: INetStream) -> None: except: await stream.reset() + async def send_ping(stream: INetStream) -> None: try: payload = b"\x01" * PING_LENGTH print(f"sending ping to {stream.muxed_conn.peer_id}") - + await stream.write(payload) with trio.fail_after(RESP_TIMEOUT): @@ -39,10 +48,11 @@ async def send_ping(stream: INetStream) -> None: if response == payload: print(f"received pong from {stream.muxed_conn.peer_id}") - + except Exception as e: print(f"error occurred : {e}") + async def run(port: int, destination: str) -> None: localhost_ip = "127.0.0.1" listen_addr = multiaddr.Multiaddr(f"/ip4/0.0.0.0/tcp/{port}") @@ -58,7 +68,7 @@ async def run(port: int, destination: str) -> None: f"-d /ip4/{localhost_ip}/tcp/{port}/p2p/{host.get_id().pretty()}\n" ) print("Waiting for incoming connection...") - + else: maddr = multiaddr.Multiaddr(destination) info = info_from_p2p_addr(maddr) @@ -66,20 +76,20 @@ async def run(port: int, destination: str) -> None: stream = await host.new_stream(info.peer_id, [PING_PROTOCOL_ID]) nursery.start_soon(send_ping, stream) - + return await trio.sleep_forever() + def main() -> None: - description = """ This program demonstrates a simple p2p ping application using libp2p. To use it, first run 'python ping.py -p ', where is the port number. Then, run another instance with 'python ping.py -p -d ', where is the multiaddress of the previous listener host. """ - + example_maddr = ( "/ip4/127.0.0.1/tcp/8000/p2p/QmQn4SwGkDZKkUEpBRBvTmheQycxAHJUNmVEnjA2v1qe8Q" ) @@ -99,11 +109,12 @@ def main() -> None: if not args.port: raise RuntimeError("failed to determine local port") - + try: trio.run(run, *(args.port, args.destination)) except KeyboardInterrupt: pass + if __name__ == "__main__": - main() \ No newline at end of file + main()