websocketのon_message関数に変数を渡す

Question

今、websocketを使ってメッセージを取得し、その都度加工するプログラムを書いています。

以下のように、on_messageの引数に変数を追加したいのですが、可能でしょうか？

def on_message(ws, message, signal):

＜その他気になっていること＞
そもそも、websocketを使ったプログラミングの基本として、def on_messageの中で複雑な処理をしない方がいいような気がしているのですが、その他の方法を見つけることができませんでした。
もし、良い方法をご存じの方がいれば、ご教示ください。

import sys
import websocket
import _thread

def on_message(ws, message):
    # 今はこの関数の中でmessageの加工を行っている。
    print('--- RECV MSG. --- ')
    print(message)

def on_error(ws, error):
    print('--- ERROR --- ')
    print(error)

def on_close(ws):
    print('--- DISCONNECTED --- ')

def on_open(ws):
    print('--- CONNECTED --- ')
    def run(*args):
        while(True):
            line = sys.stdin.readline()
            if line != '':
                print('closing...')
                ws.close()
    _thread.start_new_thread(run, ())

url = 'ws://*****/****/websocket'
# websocket.enableTrace(True)
ws = websocket.WebSocketApp(url,
                          on_message = on_message,
                          on_error = on_error,
                          on_close = on_close)
ws.on_open = on_open
ws.run_forever()

Answer 1

質問者様が感じられていますように、websocketをつなぎっぱなしにして、来るメッセージに処理をしていく、というプログラムをエレガントに書く方法は悩ましいです。on_messageはコールバックとしてライブラリに規程されているため、おそらく引数を追加することはできないのではないでしょうか。また、on_messageに複雑な処理は書くなとは決まっていませんが、なんとなく複雑な処理を書くのは憚られる場所という雰囲気です。

以前の質問回答で、そういう要望にチャレンジしてみたものがあり、参考になると思います。
https://teratail.com/questions/296228#reply-420312
websocketの通信処理をうまくクラス内に隠蔽し、メッセージだけをmain関数側で処理できるようにしています。ややasync/awaitの記述がうるさいのはご容赦ください。このコードをベースにすると、質問者が書きたいコードが書きやすいように思います。

なお、websocketsという紛らわしい名前のライブラリがあり、そちらを使っていますのでご注意ください。

Comments

[taro_yamada]

asyncioについては、何度かチャレンジしているのですが、平日昼間のみのPUSHデータのみを扱っているので、テストができず、学習が進まずにいます。
時期を見てしっかり勉強したいと思います。
とりあえず、今回の問題は、thread処理とglobal変数（モジュール内変数）でなんとか乗り切ってみようと思っています。
いつもご回答ありがとうございます。