pythonでwebsocketを複数稼働したい

Question

python初学者です、ネットで調べてみたのですがわからなかったので質問させていただきました。
python3,開発環境はJupiter Notebookです。
以下のようなwebSocketに接続し情報を取得するコードを書きましたが

import json
import ast
import websocket

#web_socket OrderBook約定履歴
def gmo_socket_board():

    websocket.enableTrace(True)
    ws = websocket.WebSocketApp('wss://api.coin.z.com/ws/public/v1')

    def on_open(self):
        message = {
            "command": "subscribe",
            "channel": "orderbooks",
            "symbol": "BTC_JPY"
        }
        ws.send(json.dumps(message))

    def on_message(self, message):
        dict_message = ast.literal_eval(message)
        ask_price = dict_message["asks"][0]["price"] #best_ask
        bid_price = dict_message["bids"][0]["price"] #best_bid
        print(ask_price,bid_price)

    ws.on_open = on_open
    ws.on_message = on_message
    ws.run_forever()
    
#web_socket Excution約定履歴
def gmo_socket_trade():

    websocket.enableTrace(True)
    ws2 = websocket.WebSocketApp('wss://api.coin.z.com/ws/public/v1')

    def on_open2(self):
        message2 = {
            "command": "subscribe",
            "channel": "trades",
            "symbol": "BTC_JPY",
            "option": "TAKER_ONLY"
        }
        ws2.send(json.dumps(message2))

    def on_message2(self, message2):
    
        dict_message2 = ast.literal_eval(message2)
        side = dict_message2["side"]
        price = dict_message2["price"]
        size = dict_message2["size"]
        timestamp = dict_message2["timestamp"]
    
        print(side)
        print(size)
        print(price)
        print(timestamp)
        print("=======================")
    
    ws2.on_open = on_open2
    ws2.on_message = on_message2
    ws2.run_forever()

として

gmo_socket_board()
gmo_socket_trade()

を実行してもgmo_socket_board()しか接続されません。
タイムラグはあってもいいので両方とも接続できるような方法はございますでしょうか。
よろしくお願いします。

追記
オリジナルのサンプルコードを見ながら下記のようにコードを書き換えてみましたが、やはり同じように初めのsocketのみが接続され、これを停止すると次のsocketが接続されるような動作をします。run_forever()で処理が一時停止しているような感じがします。thread.start_new_thread(run, ())で並列処理させているつもりですがうまく動作しません。よろしくお願いします。

import websocket
import ast
import json
import time
try:
    import thread
except ImportError:
    import _thread as thread
import time


def on_message(ws, message):
    dict_message = ast.literal_eval(message)
    ask_price = dict_message["asks"][0]["price"] #best_ask
    bid_price = dict_message["bids"][0]["price"] #best_bid
    print(ask_price,bid_price)


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

def on_close(ws):
    print("### closed ###")
    
def on_message2(ws2, message3):
    dict_message2 = ast.literal_eval(message3)
    side = dict_message2["side"]
    price = dict_message2["price"]
    size = dict_message2["size"]
    timestamp = dict_message2["timestamp"]
    
    print(side)
    print(size)
    print(price)
    print(timestamp)

def on_error2(ws2, error):
    print(error)

def on_close2(ws2):
    print("### closed ###")
    
    
def on_open(ws):
    def run(*args):
        
        message2 = {
        "command": "subscribe",
        "channel": "orderbooks",
        "symbol": "BTC_JPY"
        }
        
        ws.send(json.dumps(message2))    
        
        
    print("thread terminating...")
    thread.start_new_thread(run, ())
    
def on_open2(ws2):
    def run2(*args):
        
        message3 = {
         "command": "subscribe",
         "channel": "trades",
         "symbol": "BTC_JPY",
         "option": "TAKER_ONLY"
        }
        
        ws2.send(json.dumps(message3))    
        
    print("thread2 terminating...")
    thread.start_new_thread(run2, ())
    
if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("wss://api.coin.z.com/ws/public/v1",
                              on_message = on_message,
                              on_error = on_error,
                              on_close = on_close)
    ws2 = websocket.WebSocketApp("wss://api.coin.z.com/ws/public/v1",
                              on_message = on_message2,
                              on_error = on_error2,
                              on_close = on_close2)
    
    ws.on_open = on_open
    ws.run_forever()
    
    time.sleep(5)
    
    ws2.on_open = on_open2
    ws2.run_forever()

Comments

[t_obara]

websocket2はどこから出てきたのでしょう？エラーになりませんか？

[hazata]

申し訳ありません、エラーになりました。コードを修正いたします

Answer 1

両方とも接続する、ということは、接続し続けるWebSocketクライアントを作る必要があります。

オリジナルのGit
https://github.com/websocket-client/websocket-client

に記載されているLong-lived connectionの形式、特にon_openにおいてスレッド実行が必要です。
質問コードは、この部分がかなり違った内容になっています。まずは、オリジナルのサンプルコードを起点に、カスタマイズするべきと思います。

Answer 2

修正依頼にも出ているように、websocket2をwebsocketに修正しないと、gmo_socket_tradeは動作しない。

-    websocket2.enableTrace(True)
+    websocket.enableTrace(True)
-    ws2 = websocket2.WebSocketApp('wss://api.coin.z.com/ws/public/v1')
+    ws2 = websocket.WebSocketApp('wss://api.coin.z.com/ws/public/v1')

この修正を行なっても、最初に呼び出したgmo_socket_boardを終了するまでgmo_socket_tradeは呼び出されない。

gmo_socket_boardは、関数末尾に書かれたrun_foreverによって(Ctrl-Cキーで中断するまで)戻ってこないからだ。実際、ソースの修正後に、

gmo_socket_board()
gmo_socket_trade()

を実行して、Ctrl-Cキーでgmo_socket_boardを終了すると、gmo_socket_tradeが実行された。

備考: 利用しているAPIの制限として、

Public WebSocket APIの制限

同一IPからのリクエスト(subscribe, unsubscribe)は、1秒間1回を上限とします。

制限 – APIドキュメント| GMOコイン

とあるので、1秒以内に2つの関数でsubscribeを実行した場合には、後から実行した関数で接続に失敗するだろう。