python3 websocket客户端

需要安装websocket-client-py3模块

try:
    import websocket
except ImportError:
    print("请先安装websocket-client-py3扩展")
    sys.exit(1)

class socket:
    """
    socket通讯
    """
    address = "ws://127.0.0.1:2345"
    ws = None

    def __init__(self):
        websocket.enableTrace(False)
        self.ws = websocket.WebSocketApp(self.address,
                                         on_message=self.on_message,
                                         on_error=self.on_error,
                                         on_close=self.on_close,
                                         on_open=self.on_open)
        self.ws.run_forever()

    def on_message(self, ws, message):
        """
        服务端消息
        :param ws:
        :param message:
        :return:
        """
        print(message)

    def on_error(self, ws, error):
        print("错误：" + error)

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

    def on_open(self, ws):
        def run(*args):
            ws.send('hello')
            print("thread terminating...")

        thread.start_new_thread(run, ())

socket_client = socket()