chattle/server.py

__author__ = "Kieran Osborne"
__version__ = "0.0.1"
__status__ = "Development"

if (__name__ == "__main__"):
    import threading
    import config
    import socket

    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
        # Avoid bind() exception: OSError: [Errno 48] Address already in use
        client_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        client_socket.bind((config.host, config.port))
        client_socket.listen()

        clients = []

        def spawn_client(client_connection, client_address):
            client_connection.send("Welcome to this chatroom!".encode("utf-8"))

            try:
                data = client_connection.recv(4096)

                print("test", data if data else "OOOPS")

                while data:
                    message = "<" + client_address[0] + "> " + data

                    print(message)

                    for client in clients:
                        if client != client_connection:
                            try:
                                client.send(message)

                            except:
                                client.close()
                                # if the link is broken, we remove the client
                                if client_connection in clients:
                                    clients.remove(client_connection)

                    data = client_connection.recv(4096)

                if client_connection in clients:
                    print("dead")
                    clients.remove(client_connection)

            except:
                return

        while True:
            connection, address = client_socket.accept()

            clients.append(connection)

            # prints the address of the user that just connected
            print(address[0] + " connected")
            threading.Thread(target=spawn_client, args=(connection, address)).start()