How to receive data from multiple WebSockets asynchronously in Python?

Question

I am currently trying to use the websockets library. If another library is better suited for this purpose, let me know.

Given these functions:

def handle_message(msg):
    # do something

async def consumer_handler(websocket, path):
    async for message in websocket:
        handle_message(message)

How can I (indefinitely) connect to multiple websockets? Would the below code work?

import asyncio
import websockets


connections = set()
connections.add(websockets.connect(consumer_handler, 'wss://api.foo.com', 8765))
connections.add(websockets.connect(consumer_handler, 'wss://api.bar.com', 8765))
connections.add(websockets.connect(consumer_handler, 'wss://api.foobar.com', 8765))

async def handler():
    await asyncio.wait([ws for ws in connections])

asyncio.get_event_loop().run_until_complete(handler())

BlueOxile · Accepted Answer · 2018-08-17 17:29:44Z

9

For anyone who finds this, I found an answer. Only works in > Python 3.6.1 I believe.

import asyncio
import websockets

connections = set()
connections.add('wss://api.foo.com:8765')
connections.add('wss://api.bar.com:8765'))
connections.add('wss://api.foobar.com:8765'))

async def handle_socket(uri, ):
    async with websockets.connect(uri) as websocket:
        async for message in websocket:
            print(message)

async def handler():
    await asyncio.wait([handle_socket(uri) for uri in connections])

asyncio.get_event_loop().run_until_complete(handler())

answered Aug 17, 2018 at 17:29

BlueOxile

3512 silver badges12 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Saurav Kumar Over a year ago

I was looking something similar with auto connect feature to ensure that the clients are always connected to the websocket Server. If the connection was closed I want the clients to re-establish the connection with the servers. For example wss://api.foobar.com:8765 websocket server was restarted. The client should reconnects automatically after few intervals. Can you help illustrate how to write such functionalities?

MICHAEL MASTERS · Accepted Answer · 2021-12-27 14:55:36Z

1

Instead of:

connections = set()

I would list it:

connections = []
connections = ["wss://api.foo.com:8765"]
connections.append ("wss://api.bar.com:8765")
connections.append ("wss://api.foobar.com:8765")

answered Dec 27, 2021 at 14:55

MICHAEL MASTERS

494 bronze badges

Collectives™ on Stack Overflow

How to receive data from multiple WebSockets asynchronously in Python?

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related