chatroom_server

import asyncio

client_list = {}

async def chat_room(reader, writer):
    addr = writer.get_extra_info('peername')
    print(f'got a connection from client at: {addr}')
    data = await reader.read(100)

    if not data:
        del client_list[addr]

    if addr in client_list:
        # This is a socket we've already connected to
        for client_addr, client_writer in client_list:
            if client_writer is not writer:
                client_writer.write(data)
                await client_writer.drain()
        writer.close()
    else:
        # Add this writer into our client table
        client_list[addr] = writer

async def main():
    server = await asyncio.start_server(
        chat_room, '127.0.0.1', 8888)

    addr = server.sockets[0].getsockname()
    print(f'Serving on {addr}')

    async with server:
        await server.serve_forever()

asyncio.run(main())