A method for listening to streaming data on an Amberdata websocket

websocket_listen.py

async def listen(api_key):
    "Opens the websocket connection and listens for pending transactions"
    # the amberdata websocket uri
    uri = 'wss://ws.web3api.io'
    # our headers for the connection
    headers = {
        "x-api-key": api_key["AMBERDATA_API_KEY"],
        "x-amberdata-blockchain-id": "408fa195a34b533de9ad9889f076045e"
    }
    # outer loop
    while True:
        # create the websocket item
        async with websockets.connect(uri, extra_headers=headers) as websocket:
            logger.info(f"Connected to Websocket at {uri}")
            # the message to pass for pending transactions
            message = json.dumps({
                'jsonrpc': '2.0',
                'id': 2,
                'method': 'subscribe',
                'params': ['pending_transaction']
            })
            # send our message to the websocket
            await websocket.send(message)
            # continuously listen for data and process
            while True:
                try:
                    # the response from the websocket
                    response = await asyncio.wait_for(websocket.recv(), timeout=25)
                except Exception as e:
                    logger.error(str(e))
                    try:
                        pong = await websocket.ping()
                        await asyncio.wait_for(pong, timeout=100)
                        logger.debug('Ping OK, keeping connection alive...')
                        continue
                    except:
                        # sleep for 30 seconds
                        await asyncio.sleep(30)
                        break  # inner loop
                # interpret the response
                await on_response(response)