python實現建立websocket通信

實現代碼如下：

#websocket協議通信
import threading
import time
import websocket

def when_message(ws, message):
    print('/n接收到的消息：' + message)
# 當建立連接后，死循環不斷輸入消息發送給服務器
# 這里需要另起一個線程
def when_open(ws):
    print('連接成功')
    def run():
        while True:
            a = input('請輸入：')
            ws.send(a)
            time.sleep(0.5)
            if a == 'close':
                ws.close()
                break
    t = threading.Thread(target=run)
    # t.setDaemon(True)
    t.start()
def when_close(ws):
    print('連接關閉')

if __name__ == '__main__':
    ws = websocket.WebSocketApp('ws://localhost:9999/', on_message=when_message, on_open=when_open, on_close=when_close)
    ws.run_forever()