WebSocket小叙

import websocket
import threading
import time

class Status:
    INIT = 'init'
    OPENED = 'opened'
    CLOSED = 'closed'

class WSTest:
    def __init__(self):
        self.thread = None
        self.ping_thread = None
        self.ws = None
        self.status = Status.INIT

    def send_message(self, message) -> bool:
        if self.status != Status.OPENED:
            return False
        self.ws.send(message)
        return True
    def close(self):
        self.ws.close()
        # 等待关闭
        for i in range(200):
            time.sleep(0.02)
            if self.status == Status.CLOSED:
                break
        if self.status != Status.CLOSED:
            return
        if self.ping_thread and self.ping_thread.is_alive():
            self.ping_thread.join()
    def run(self):
        def on_message(ws, message):
            print(message)
        def on_error(ws, error):
            print(error)
        def on_close(ws):
            print('关闭连接')
            self.status = Status.CLOSED
        def on_open(ws):
            self.status = Status.OPENED
        self.ws = websocket.WebSocketApp("ws://echo.websocket.org",
                                         on_open=on_open,
                                         on_message=on_message,
                                         on_close=on_close,
                                         on_error=on_error)
        self.ping_thread = threading.Thread(target=self.ws.run_forever, args=(None, None, 2, 5))
        self.ping_thread.start()

if __name__ == "__main__":
    ws = WSTest()
    ws.run()
    while True:
        s = input()
        if s == 'close':
            break
        ws.send_message(s)
    ws.close()