时间戳

首页 > 解决方案 > 如何将连续的“async with”语句重写为循环？

python - 如何将连续的“async with”语句重写为循环？

问题描述

我有以下摘录，除了它要求我知道我计划提前打开多少个 WebSocket 之外，它在各个方面都很好用。

相反，如果我从主机列表开始并且必须创建 N 个 WebSocket 连接，那么我如何才能在循环中为 N 完成此代码为 3 个 Websockets（通过'async with'）所做的事情？

我已经在一个循环中等待，这是我无法弄清楚如何“循环”的“异步”。

async with websockets.connect('ws://192.168.0.174:81') as websocket1:
    async with websockets.connect('ws://192.168.0.194:81') as websocket2:
        async with websockets.connect('ws://192.168.0.179:81') as websocket3:

            datatosend = GetBallData()

            for socket in [websocket1, websocket2, websocket3]:
                await socket.send(datatosend)

标签： pythonwebsocketasync-await

解决方案

从 Python3.7 开始，模块中提供了AsyncExitStackcontextlib，这就是它可以用于您的问题的方式：

hosts = ['ws://192.168.0.174:81', 'ws://192.168.0.194:81', 'ws://192.168.0.179:81']
async with contextlib.AsyncExitStack() as stack:
    sockets = [
        await stack.enter_async_context(websockets.connect(host))
        for host in hosts
    ]

推荐阅读