如何在 class 中使用 websocket 获取数据并实例化变量

Question

我试图在 class 中建立一个 websocket 连接，因为我需要访问 class 之外的变量 self.close。问题是连接没有建立，我也无法访问on_message

里面的变量

我怎样才能让它工作？

import websocket
import json


class WS:

    def __init__(self):

        self.socket = 'wss://ftx.com/ws/'

    def stream(self):
        self.ws = websocket.WebSocketApp(self.socket,
            on_message=self.on_message,
            on_error=self.on_error,
            on_close=self.on_close)
        self.ws.run_forever()

    def on_open(self):

        print('connected')

        data = {'op': 'subscribe', 'channel': 'ticker', 'market': 'ETH-PERP'}
        self.ws.send(json.dumps(data))

    def on_close(ws):
        print('disconnected')

    def on_message(self,message):

        json_msg = json.loads(message)
        self.close = json_msg['data']['last']

    def get_data_out(self):

        return self.close
   
    def on_error(ws,error):
        print(error)

Answer 1

我试过运行你的代码

WS().stream()

但它对我不起作用（即使我添加缺失 on_open=self.on_open）

我发现甚至 ftx documentation and its example 也使用模块 websocket 但我没有测试它。

但是如果我使用不同的模块 websockets（使用 char s）而不是 websocket（不使用 char s），那么我可以订阅频道并获取值。

最小工作示例

import asyncio
import websockets
import json

async def handler():
    async with websockets.connect('wss://ftx.com/ws/') as ws:
        
        # subscribe
        data = {'op': 'subscribe', 'channel': 'ticker', 'market': 'ETH-PERP'}
        await ws.send(json.dumps(data))

        # get all messages (not only with `update`)
        async for message in ws:
            print(message)

# --- main ---
     
asyncio.get_event_loop().run_until_complete(handler())

要仅获取 last 值，它需要过滤消息：

        async for message in ws:
            data = json.loads(message)
            if data['type'] == 'update':
                print(data['data']['last'])

但是如果要像class一样使用，还是需要多写代码。

顺便说一句：

似乎 websocket（没有 char s）是非常旧的模块，最后更新是在 2010 年，websockets（有 char s）是在 2021 年更新的（很少 weeks/months 之前）。

https://pypi.org/project/websocket/

https://pypi.org/project/websockets/

编辑：

在其他问题中，我找到了显示有关连接的更多信息的命令：

websocket.enableTrace(True)

这会显示错误，因为函数执行时使用的值多于您定义的值。
他们必须得到 websocket 作为第一个参数（在 self 之后）

你有：

def on_open   (self):
def on_close  (ws):
def on_message(self, message):
def on_error  (ws, error):

但必须

def on_open   (self, ws):
def on_close  (self, ws):
def on_message(self, ws, message):
def on_error  (self, ws, error):

你也忘了on_open=self.on_open

完整的工作代码

import websocket
import json

class WS:

    def __init__(self):
        self.socket = 'wss://ftx.com/ws/'
        self.close = None # default value at start

    def stream(self):
        self.ws = websocket.WebSocketApp(
                    self.socket,
                    on_message=self.on_message,
                    on_error=self.on_error,
                    on_close=self.on_close,
                    on_open=self.on_open
                  )

        print('run forever')
        self.ws.run_forever()
        
    def on_open(self, ws):
        print('on_open:', ws)

        data = {'op': 'subscribe', 'channel': 'ticker', 'market': 'ETH-PERP'}
        
        self.ws.send(json.dumps(data))

    def on_close(self, ws):
        print('on_close:', ws)

    def on_message(self, ws, message):
        print('on_message:', ws, message)
        
        data = json.loads(message)

        if data['type'] == 'update':
            self.close = data['data']['last']

    def get_data_out(self):
        return self.close
   
    def on_error(self, ws, error):
        print('on_error:', ws, error)
        
# --- main ---

websocket.enableTrace(True)  # if you want to see more information  
                             # but code works also without this line

WS().stream()

如何在 class 中使用 websocket 获取数据并实例化变量

How to get data with websocket in a class and instantiate variable

python

websocket