#!/usr/bin/env python3

import asyncio
import json
import websockets


async def test(symbol="AAPL"):
    uri = "ws://localhost:8765"

    async with websockets.connect(uri) as ws:
        if symbol:
            await ws.send(json.dumps({"symbol": symbol}))

        while True:
            msg = await ws.recv()
            data = json.loads(msg)
            print(data)


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