Learning Python Network Programming

Client and Server Applications

for msg in msgs:

msg = '{}: {}'.format(addr, msg)

print(msg)

broadcast_msg(msg)

def handle_client_send(sock, q, addr):

""" Monitor queue for new messages, send them to client as

they arrive """

while True:

msg = q.get()

if msg == None: break

try:

tincanchat.send_msg(sock, msg)

except (ConnectionError, BrokenPipe):

handle_disconnect(sock, addr)

break

def broadcast_msg(msg):

""" Add message to each connected client's send queue """

with lock:

for q in send_queues.values():

q.put(msg)

def handle_disconnect(sock, addr):

""" Ensure queue is cleaned up and socket closed when a client

disconnects """

fd = sock.fileno()

with lock:

# Get send queue for this client

q = send_queues.get(fd, None)

# If we find a queue then this disconnect has not yet

# been handled

if q:

q.put(None)

del send_queues[fd]

addr = sock.getpeername()

print('Client {} disconnected'.format(addr))

sock.close()

if __name__ == '__main__':

listen_sock = tincanchat.create_listen_socket(HOST, PORT)

addr = listen_sock.getsockname()

print('Listening on {}'.format(addr))