WebSocket with Python on Ubuntu 17.10

client.py

import websocket
import thread
import time
import sys


port = sys.argv[1]


def on_message(ws, message):
    print(message)
    pass

def on_error(ws, error):
    print (error)

def on_close(ws):
    print("### closed ###")

def on_open(ws):
    def run(*args):
        while True:
            time.sleep(5)
            ws.send("Hello")
        time.sleep(1)
        ws.close()
    thread.start_new_thread(run, ())


if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://localhost:" + port + "/",
                              on_message = on_message,
                              on_error = on_error,
                              on_close = on_close)
    ws.on_open = on_open
    ws.run_forever()

gistfile1.txt

import websocket
import thread
import time
import sys


port = sys.argv[1]


def on_message(ws, message):
    print(message)
    pass

def on_error(ws, error):
    print (error)

def on_close(ws):
    print("### closed ###")

def on_open(ws):
    def run(*args):
        while True:
            time.sleep(5)
            ws.send("Hello")
        time.sleep(1)
        ws.close()
    thread.start_new_thread(run, ())


if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://localhost:" + port + "/",
                              on_message = on_message,
                              on_error = on_error,
                              on_close = on_close)
    ws.on_open = on_open
    ws.run_forever()

from SimpleWebSocketServer import SimpleWebSocketServer, WebSocket

import sys
import time

port = int(sys.argv[1])

class SimpleEcho(WebSocket):

    def handleMessage(self):
        # echo message back to client
        print(self.data)
        self.sendMessage(self.data)
        self.sendMessage(u"ok")

    def handleConnected(self):
        print(self.address, 'connected')

    def handleClose(self):
        print(self.address, 'closed')

server = SimpleWebSocketServer('', port, SimpleEcho)
print(1)
server.serveforever()
print(2)

run.sh

sudo apt install -y python3-pip
pip3 install websocket
pip3 install git+https://github.com/dpallot/simple-websocket-server.git
python3 server.py 8080 &
python client.py 8080

server.py

from SimpleWebSocketServer import SimpleWebSocketServer, WebSocket

import sys
import time

port = int(sys.argv[1])

class SimpleEcho(WebSocket):

    def handleMessage(self):
        # echo message back to client
        print(self.data)
        self.sendMessage(self.data)
        self.sendMessage(u"ok")

    def handleConnected(self):
        print(self.address, 'connected')

    def handleClose(self):
        print(self.address, 'closed')

server = SimpleWebSocketServer('', port, SimpleEcho)
print(1)
server.serveforever()
print(2)