from rest.rest_api import deepcoin_api

account_test = deepcoin_api.AccountTest()

params = {
    'instType': 'SWAP',
    'ccy': 'USDT'
}

res = account_test.get_account_balance(params)

import websocket
import json
import time
import threading

ws_url = "wss://stream.deepcoin.com/streamlet/trade/public/swap?platform=api"

def on_message(ws, message):
    print(f"Received: {message}")

def on_error(ws, error):
    print(f"Error: {error}")

def on_close(ws, close_status_code, close_msg):
    print("WebSocket closed")

def on_open(ws):
    print("WebSocket connected")

    def send_heartbeat():
        while True:
            ws.send("ping")
            print("Sent: ping")
            time.sleep(30)

    threading.Thread(target=send_heartbeat, daemon=True).start()

    ### 最新行情
    request_data = {
        "SendTopicAction": {
            "Action": "1",
            "FilterValue": "DeepCoin_ETHUSDT",
            "LocalNo": 111,
            "ResumeNo": -2,
            "TopicID": "7"
        }
    }

    ws.send(json.dumps(request_data))
    print(f"Sent: {json.dumps(request_data)}")

ws = websocket.WebSocketApp(
    ws_url,
    on_open=on_open,
    on_message=on_message,
    on_error=on_error,
    on_close=on_close,
)

ws.run_forever()