7.3. Thay đổi thông tin lệnh phái sinh (WebSocket)

"""WebSocket client for TCBS derivative order updates — Python 3.10+"""

import asyncio
import base64
import json

from websockets.asyncio.client import connect


WS_URL = "wss://openapi.tcbs.com.vn/ws/nesoi"
TOKEN = "YOUR_OPENAPI_TOKEN_HERE"

TOPICS = ["DE_ORDER", "DE_CONDITIONAL_ORDER", "DE_OPEN_POSITION"]


async def main():
    async with connect(WS_URL, ping_interval=None, ping_timeout=None) as ws:

        # 1. Authenticate
        jwt_payload = json.dumps({"jwt": TOKEN})
        encoded = base64.b64encode(jwt_payload.encode()).decode()
        await ws.send(f"authenticate|{encoded}")

        # 2. Wait for auth response
        while True:
            msg = await asyncio.wait_for(ws.recv(), timeout=10.0)
            print(f"< {msg}")
            if msg.startswith("authenticate|"):
                payload = msg.split("|", 1)[1]
                result = json.loads(payload)
                if result.get("success"):
                    break
                else:
                    raise Exception(f"Auth failed: {result}")

        # 3. Subscribe to all topics
        for topic in TOPICS:
            sub_payload = base64.b64encode(json.dumps({"topic": topic}).encode()).decode()
            await ws.send(f"subscribe|{sub_payload}")
            print(f"Subscribed to {topic}")

        # 4. Heartbeat task
        async def heartbeat():
            while True:
                await ws.send("ping|1")
                await asyncio.sleep(2)

        asyncio.create_task(heartbeat())

        # 5. Receive updates
        while True:
            raw = await ws.recv()
            if raw.startswith("message_proto|"):
                parts = raw.split("|", 2)
                topic = parts[1]
                data = json.loads(parts[2])

                if topic == "DE_ORDER":
                    print(f"[ORDER] {data.get('symbol')} "
                          f"Side={data.get('side')} "
                          f"Status={data.get('status')} "
                          f"Vol={data.get('volume')}")
                elif topic == "DE_OPEN_POSITION":
                    print(f"[POSITION] {data.get('symbol')} "
                          f"Net={data.get('net')} "
                          f"PnL%={data.get('pcRemain')}")
                elif topic == "DE_CONDITIONAL_ORDER":
                    print(f"[COND] {data.get('symbol')} "
                          f"Status={data.get('status')} "
                          f"Type={data.get('notes')}")


if __name__ == "__main__":
    asyncio.run(main())