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

Question

下記のようなコードで2個の値を取得することまではできたのですが、このclosea, closebという値を利用しようとしたときに困っています。

例えば下記のようにmainでコードを利用して代入しようとした場合、両方から値が来ていればいいですが、来ていない場合に、値がなくなってしまい、プログラムが動かないということが起きます。

これを値が来ていないときは片方は前回の値を（初回だけは両方揃うまでは片方は0でもいいのですが）したいと考えているのですが、どのようにしたらいいかで詰まってしまいました。。。
（試しにb.pyのようにmainの中にclosea, closebを0と定義してみたりすればコードは動きますが、そうなると次の時に片方が0になってしまって前回の値でなくなってしまうのでどうしたものかと。。。）

対応法があればお伺いできれば幸いです。
（def mainの中はとりあえず適当なコードです。実際にはもっと長いコードが使われるものとご認識いただければと思います）

(a.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()

(b.py)

import websocket
import json


def main(closea, closeb):
　　closea = 0
   closeb = 0

    closea = closea + 10
    closeb = closeb + 10

    print (closea, closeb)

以下省略

Answer 1

websocketにこだわるならば、例として下記のように別スレッドでデータを取得し
使いたい値をTickerオブジェクト経由でアクセスするようなプログラムを考えることはできます。

a.py（データを取得する方）

# a.py

import websocket
import json
import threading


class Ticker:

    def __init__(self):
        self.values = dict(closea=0, closeb=0)

    def start(self):
        def on_message(ws, message):
            try:
                # valuesにそれぞれの値を格納する。
                rs = json.loads(message)
                channel = rs["params"]["channel"]
                if channel == "lightning_ticker_FX_BTC_JPY":
                    self.values["closea"] = round(float('{ltp}'.format(**(rs["params"]["message"]))))
                elif channel == "lightning_ticker_BTC_JPY":
                    self.values["closeb"] = round(float('{ltp}'.format(**(rs["params"]["message"]))))
                else:
                    print("Got other channel data:", channel, rs)
            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'}}
            ]))

        def run(ws):
            ws.run_forever()

        self.ws = websocket.WebSocketApp("wss://ws.lightstream.bitflyer.com/json-rpc",
                                        on_message=on_message,
                                        on_error=on_error,
                                        on_open =on_open,
                                        on_close=on_close)
        self.th = threading.Thread(target=run,args=(self.ws,))
        self.th.start()

    def shutdown(self):
        self.ws.close()

　

b.py （値を使う方のプログラムの例）

# b.py （値を使う方のプログラムの例）


from a import Ticker  # a.pyからTickerクラスをインポート。
import time

# 開始処理
ticker = Ticker()
ticker.start()

#　～～～任意のコード～～～

# プログラム中で、任意のタイミングでtickerのvaluesを呼び出せばよい。
while True:
    try:
      print("closea", ticker.values["closea"])
      print("closeb", ticker.values["closeb"])
      time.sleep(1)
    except KeyboardInterrupt:
      break

#　～～～任意のコード～～～


# 終了前に必ずshutdownを呼び出す。
ticker.shutdown()
        
print("end")

スレッドを使っているため、a.py内でのエラー処理に気を付けないとハマります。
プログラムの目的にもよりますが、API制限の範囲内でアクセス頻度でよいなら、websocketではなく下記のようにHTTP APIを使った方が圧倒的に楽かと思います。

# そもそもwebsocketを使わない例。
import requests
import json


url = "https://api.bitflyer.com/v1/ticker"


def get_fx_btc_jpy():
    res = requests.get(url, params={"product_code": "FX_BTC_JPY"})
    return res.json()["ltp"]


def get_btc_jpy():
    res = requests.get(url, params={"product_code": "BTC_JPY"})
    return res.json()["ltp"]


# 欲しいタイミングでそのときの値を取得できる。
print(get_fx_btc_jpy())
print(get_btc_jpy())

Comments

[step595]

ご回答ありがとうございます。
httpであればできたのですが、察しの通り制限の問題でwebsocketにこだわりがありました。
並列処理についてもっと勉強しないとですね。。。

あとこのコードを起動させてみたのですが、a.pyもb.pyも両方起動させるということであってますでしょうか？
それだとa.pyがうごかず、b.pyだけだとずっと0を取得し続ける状況です。
初歩的なミスであればすみません。

[退会済みユーザー]

動かすのは(pythonコマンドで実行するのは）b.pyだけとなります。
（b.pyの中でaをインポートしているため、aまで実行する必要はないです）
（aとbを同じディレクトリに入れてそのディレクトリからb.pyを実行するという想定です）

b.pyのコードを一部修正しました。（whileで1秒ごとに表示）
最初の数秒は、websocketからデータが来ていないので、0が表示される場合があります。

このコードをそのまま実行した場合でも、うごかないでしょうか？

もし質問者さんがカスタマイズされた特定のコードの場合だと動かなくなるorうまく表示されない、というのであれば、
当該コード（少なくともb.pyはそのまま全部、あと、もしa.pyも回答欄のコードから一部でもカスタマイズされている場合はa.pyも）を質問欄に追記していただくことは可能でしょうか？
こちらで調べてみます。

[step595]

ご丁寧にありがとうございます。
特に編集せずにまるまるコピペで起動してみてるつもりなのですが、ずっと値が0.0の取得なんですよね。。。（編集後のものでもためしました）

ちなみにどこまでコードが届いているかを調べるために別途on_message内に適当にprint("1")とか記入してみてるのですがprintの文章が起動されずそこまで動いている感じがしないのです。。どっかで詰まっているのでしょうか。。。

[退会済みユーザー]

私の手元ではきちんと表示されるので不思議ですね・・・

考えられるのは
（１）websocketで取得できるはずのデータ（Realtime API）が、IPアドレスで制限されている可能性
（２）OS、python、ライブラリ固有の環境
によってうまく動いていない可能性があります。

（１）については、前の質問の回答（https://teratail.com/questions/328811#reply-454944）のプログラムだったらうまく表示されるのであれば、あてはまりません。

（２）については、
pip install --upgrade websocket-client
を実行してwebsocket-clientを最新版にしてみるか、

とりあえずOS環境、pythonバージョン、および
pip show websocket-client
をコマンドプロンプト（ターミナル）で実行したときに表示されるwebsocket-clientのバージョンを教えていただけないでしょうか。

自分の環境は、Python3.8.6、windows10、websocket-client 0.58.0で
a.pyとb.pyを同じディレクトリにいれ、そのディレクトリからコマンドプロンプトで
python b.py
を実行しています。

[退会済みユーザー]

あとは、２つのファイルに分けているのが不具合の原因かもしれません。
1つのファイルにして実行してみるとどうでしょうか。

具体的には、a.pyのコードの後ろに、b.pyのコードをくっつけて（ただし、from a import Tickerの行は削除）、a.pyを実行してみる、ということです。

[step595]

できました！ありがとうございます。
ちなみにちょっとお伺いできたらありがたいのですが、b.py 修正後　while True:を付けた影響でticker.shutdown()まで届いていなくてもうごいているのですが、ticker.shutdown()はいるのでしょうか?

[退会済みユーザー]

ticker.shutdown() はあくまでプログラム終了時にwebsocketからのデータ取得を終了させるためのものです。
なのでwhileループが動いている間は届いていなくても構いません（届いていないのが正常です）

回答の例示コードでは、プログラム停止のためにCtrl+Cを押すとKeyboardInterrupt例外が発生し、whileループを抜けるようになっています。
whileループを抜けた後、ticker.shutdown()が実行されて、websocketからのデータ取得も終了されます。

[step595]

ありがとうございます。解決してたすかりました。