tcp_socket_server_3_defaultselector.py

#!/usr/bin/env python3
"""
Socket server using Selector.
"""

import socket, selectors

HOST, PORT = ('localhost', 12345)

def handle(sock, addr):
    try:
        data = sock.recv(1024)  # Should be ready
    except ConnectionError:
        print(f"Client suddenly closed while receiving")
        return False
    print(f"Received {data} from: {addr}")
    if not data:
        print("Disconnected by", addr)
        return False
    data = data.upper()
    print(f"Send: {data} to: {addr}")
    try:
        sock.send(data)  # Hope it won't block
    except ConnectionError:
        print(f"Client suddenly closed, cannot send")
        return False
    return True

serv_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
serv_sock.bind((HOST, PORT))
serv_sock.listen(1)
# serv_sock.setblocking(False)
print("Server started")

sel = selectors.DefaultSelector()
sel.register(serv_sock, selectors.EVENT_READ)
while True:
    print("Waiting for connections or data...")
    events = sel.select()

    for key, mask in events:
        sock = key.fileobj
        if sock == serv_sock:
            sock, addr = serv_sock.accept()  # Should be ready
            print("Connected by", addr)
            sel.register(sock, selectors.EVENT_READ)
        else:
            addr = sock.getpeername()
            if not handle(sock, addr): # Disconnected
                sel.unregister(sock)
                sock.close()
                continue
serv_sock.close()