python实现websocket保存binance行情数据至sqlite中

最新推荐文章于 2025-03-27 09:00:00 发布

达科索斯

最新推荐文章于 2025-03-27 09:00:00 发布

阅读量1.6k

点赞数

分类专栏：量化文章标签： python websocket sqlite

本文链接：https://blog.csdn.net/weixin_40264579/article/details/128962382

版权

量化专栏收录该内容

8 篇文章

订阅专栏

该代码示例展示了如何使用Python的websocket-client库连接到Binance的WebSocketAPI，接收BTCUSDT交易对的实时交易数据。接收到的数据被解析为JSON，然后插入到SQLite3数据库中创建的binance_data表中，以便存储和后续分析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先，您需要安装必要的库：

pip install websocket-client
pip install sqlite3

然后，您可以使用以下代码实现您的目标：

import websocket
import sqlite3
import json

# Connect to the Binance websocket API
def on_message(ws, message):
    # Parse the message as JSON
    data = json.loads(message)

    # Connect to the SQLite database
    conn = sqlite3.connect('binance.db')
    c = conn.cursor()

    # Create the table if it doesn't exist
    c.execute('''CREATE TABLE IF NOT EXISTS binance_data
                 (timestamp TEXT, price REAL)''')

    # Insert the data into the table
    c.execute("INSERT INTO binance_data VALUES (?, ?)", (data['timestamp'], data['price']))

    # Commit the changes and close the connection
    conn.commit()
    conn.close()

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

def on_close(ws):
    print("Closed")

def on_open(ws):
    print("Opened")

websocket.enableTrace(True)
ws = websocket.WebSocketApp("wss://stream.binance.com:9443/ws/btcusdt@trade",
                            on_message=on_message,
                            on_error=on_error,
                            on_close=on_close)
ws.on_open = on_open
ws.run_forever()

在这个示例中，我们首先连接到Binance的websocket API，然后在收到消息时处理数据。我们使用sqlite3库连接到本地数据库，如果表不存在，则创建表，并将数据插入表中。最后，我们提交更改并关闭连接。该API提供了实时的交易数据。我们创建了几个回调函数，如on_message，on_error，on_close和on_open，以处理websocket的不同状态和事件。

在on_message回调函数中，我们解析了收到的消息，并将其作为JSON数据处理。然后，我们使用sqlite3库连接到本地数据库并创建了一个数据表，如果数据表不存在，则将其创建。接下来，我们将数据插入表中，并在最后提交更改并关闭连接。

最后，我们使用websocket.WebSocketApp方法连接到Binance的websocket API，并将回调函数作为参数传递。最后，我们调用ws.run_forever（）方法，以保持websocket连接。