Python websockets，订阅多个频道

Question

我正在尝试同时连接到多个通道，并通过python websocket库从推送API接收消息。

考虑到下面的代码，您将如何连接到多个通道？ 此代码是从这里获得并略微修改的： https://pypi.python.org/pypi/websocket-client

让我感到困惑的是倒数第二行：ws.on_open = on_open. on_open被定义为上面的函数，并接受 1 个参数，但在调用函数时没有传递任何参数，我不记得以前在 python 代码中遇到过这种情况，所以我不确定这一行到底发生了什么。

如何修改此代码，以便可以将包含字符串的变量传递给函数on_open，以便指定要订阅的香奈儿的名称？我的主要目标是能够使用多处理库传递多个通道以同时订阅。

我会通过创建多个 ws 对象或一个 ws 对象并使用不同的通道作为参数多次调用on_open来实现这一点吗？

import websocket
import thread
import time
import json
def on_message(ws, message):
print(message)
def on_error(ws, error):
print(error)
def on_close(ws):
print("### closed ###")
def on_open(ws):
def run(*args):   
ws.send(json.dumps({'channel':'channel1'}))  
while True:
time.sleep(1)
ws.close()
print("thread terminating...")
thread.start_new_thread(run, ())
if __name__ == "__main__":
websocket.enableTrace(True)
ws = websocket.WebSocketApp("wss://random.example.com",
on_message = on_message,
on_error = on_error,
on_close = on_close)
ws.on_open = on_open
ws.run_forever()

Answer 1

使用partial传入其他参数

from functools import partial
def on_open(ws, channel_name):
"""
Notice the channel_name parameter
"""
# create a new function with the predefined variable
chan1 = partial(on_open, channel_name='channel 1')
# set the new function as the on_open callback
ws1.on_open = chan1
# do the same for the rest
chan2 = partial(on_open, channel_name='channel 2')
ws2.on_open = chan2

作为旁注，请考虑使用龙卷风或 Crossbar.io(又名autobahn(。这些是适当的异步框架，使 websocket 开发更简单，而不是线程和多处理。