5.9. Thay đổi thông tin lệnh cơ sở (WebSocket)

"""WebSocket client for TCBS cash 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/aither"
TOKEN = "YOUR_OPENAPI_TOKEN_HERE"


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=5.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 STOCK_ORDER
        sub_payload = base64.b64encode(json.dumps({"topic": "STOCK_ORDER"}).encode()).decode()
        await ws.send(f"subscribe|{sub_payload}")
        print("Subscribed to STOCK_ORDER")

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

        asyncio.create_task(heartbeat())

        # 5. Receive order updates
        while True:
            raw = await ws.recv()
            if raw.startswith("message_proto|STOCK_ORDER|"):
                order_json = raw.split("|", 2)[2]
                order = json.loads(order_json)
                print(f"[ORDER] {order['symbol']} "
                      f"Status={order['orStatus']} "
                      f"Qty={order['orderQtty']} "
                      f"Remain={order.get('remainQtty')}")


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