client.py

import asyncio
import sys

class ClientConnection:
    def __init__(self, addr):
        self.addr = addr

    async def run(self):
        try:
            reader, writer = await asyncio.open_connection(self.addr,4000)
            print(f"Connected to {self.addr}")

            send_task = asyncio.create_task(self.send_message(writer))
            recv_task = asyncio.create_task(self.receive_message(reader))

            await asyncio.gather(send_task, recv_task)

        except ConnectionRefusedError:
            print(f"Server {self.addr} unavailable")

    async def send_message(self, writer):
        try:
            writer.write(b"req")
            await writer.drain()
        except KeyboardInterrupt:
            await self.send_disconnect_signal(writer)

    async def receive_message(self, reader):
        while True:
            data = await reader.read(1024)
            if not data:
                print("Server failed")
                break

            if data[:1] == b"\x11":
                peers = data[1:].decode("utf-8")
                print("Got peers:", peers)
            else:
                print(data.decode("utf-8"))

    async def send_disconnect_signal(self, writer):
        writer.write(b"q")
        await writer.drain()
        writer.close()
        await writer.wait_closed()
        sys.exit()