WebSocketで二つの値を利用する方法

Question

websocketで二つのchannelアクセスしてしてその両方の値を利用したいです。

そこでa.pyとb.pyというファイルがあって、その両方のclose_aという値とclose_bという値を使いたいのですが、
これは一個のファイルで2個の値を取得できるものでしょうか?
それとも2個とも動かして、その値を別のファイルで使うのがいいのでしょうか？

もし別ファイルで使うのであればそこについてもお伺いしたいです。

素人なので分からないところも多いのですがご教授宜しくお願い致します。

(a.py)

import websocket
import json

def on_message(ws, message):
    rs = json.loads(message)['params']['message']
    close_a = round(float('{ltp}'.format(**rs)))
    print(close_a)

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

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

def on_open(ws):
    print("### open ###")
    ws.send(json.dumps({
            'method': 'subscribe',
            'params': {'channel': 'lightning_ticker_FX_BTC_JPY'},
        }))
> ここが違う。

if __name__ == "__main__":
    ws = websocket.WebSocketApp("wss://ws.lightstream.bitflyer.com/json-rpc",
        on_message=on_message,
        on_error=on_error,
        on_close=on_close)
    ws.on_open = on_open
    ws.run_forever()   

(b.py)

import websocket
import json

def on_message(ws, message):
    rs = json.loads(message)['params']['message']
    close_b = round(float('{ltp}'.format(**rs)))
    print(close_b)

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

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

def on_open(ws):
    print("### open ###")
    ws.send(json.dumps({
            'method': 'subscribe',
            'params': {'channel': 'lightning_ticker_BTC_JPY'},
        }))
> ここが違う。

if __name__ == "__main__":
    ws = websocket.WebSocketApp("wss://ws.lightstream.bitflyer.com/json-rpc",
        on_message=on_message,
        on_error=on_error,
        on_close=on_close)
    ws.on_open = on_open
    ws.run_forever()   

追記

(c.py)
import websocket
import json


def main(closea, closeb):
    closea = closea + 10
    closeb = closeb + 10
    
    print (closea, closeb)


def on_message(ws, message):
    try:
        rs = json.loads(message)
        channel = rs["params"]["channel"]
        if channel == "lightning_ticker_FX_BTC_JPY":
            closefx    = round(float('{ltp}'.format(**(rs["params"]["message"]))))
        elif channel == "lightning_ticker_BTC_JPY":
            closespot  = round(float('{ltp}'.format(**(rs["params"]["message"]))))
        else:
            print("Got other channel data:", channel, rs)
    except Exception as e:
        print(e)
    
    
    main(closefx, closespot)

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


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


def on_open(ws):
    print("### open ###")
    ws.send(json.dumps([{'method': 'subscribe', 'params': {'channel': 'lightning_ticker_FX_BTC_JPY'}},{'method': 'subscribe', 'params': {'channel': 'lightning_ticker_BTC_JPY'}}]))


if __name__ == "__main__":
    ws = websocket.WebSocketApp("wss://ws.lightstream.bitflyer.com/json-rpc",
                                on_message=on_message,
                                on_error=on_error,
                                on_close=on_close)
    ws.on_open = on_open
    ws.run_forever()

Comments

[yuki23]

close_a と close_b に代入しているところがどこにもありませんが、 close_fx の間違いですか？

[step595]

すみません!その通りです。間違えていたので修正しました。

Answer 1

二つのchannelにアクセスしたいならば、下記のように２つのチャンネル取得用のデータをsendするだけです。

import websocket
import json


def on_message(ws, message):
    try:
        rs = json.loads(message)
        channel = rs["params"]["channel"]
        close_fx = round(float('{ltp}'.format(**(rs["params"]["message"]))))
        print(channel, close_fx)
    except Exception as e:
        print(e)


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


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


def on_open(ws):
    print("### open ###")
    ws.send(json.dumps([{'method': 'subscribe', 'params': {'channel': 'lightning_ticker_FX_BTC_JPY'}},
                         {'method': 'subscribe', 'params': {'channel': 'lightning_ticker_BTC_JPY'}}]))


if __name__ == "__main__":
    ws = websocket.WebSocketApp("wss://ws.lightstream.bitflyer.com/json-rpc",
                                on_message=on_message,
                                on_error=on_error,
                                on_close=on_close)
    ws.on_open = on_open
    ws.run_forever()

Comments

[step595]

このコードだと両方ともおなじchannelにアクセスしているように見えます。
onopenとonmessageのelifのところの片方のchannelを編集してみましたがそうすると、Got other channel dataに飛ばされる気がします。

[退会済みユーザー]

わかりやすくなるように修正しました。（なお、編集前のコードでも同じチャンネルにアクセスしていたわけではありません）

[step595]

そこでなのですが、この取得した値を利用するにあたっていろいろ試行錯誤してみたのですが、（追記(c.py)）別の関数の引数として利用しようとしたのですが、片方のチャンネルしか値を取得できていない場合にエラーというか止まってしまうという現象を起こしてしまいました。
いろいろ試してみたつもりではあるのですがこの場合においてなにかいい対策法はないでしょうか？

[退会済みユーザー]

https://teratail.com/questions/330046にて回答しました。

[step595]

ご回答ありがとうございます。