Python，一种检查任务是否完成的干净方法

Question

loop = asyncio.get_event_loop()
executor = concurrent.futures.ThreadPoolExecutor(max_workers=5)
tasks = [loop.run_in_executor(executor, self.parse_url, url) for url in urls]
await asyncio.wait(tasks, timeout=5)
for (i, task) in zip(urls, tasks):
    result = task.result() if task._state.lower() == 'finished' else []

在上面的代码中，我使用 '_state' 属性来检查任务是否确实完成并且没有超时。

有没有更清洁的方法？我认为我不应该像那样访问 '_state' 属性。

Answer 1

您可以使用task.done()来检查任务是否已完成。

此外，asyncio.wait()它本身返回两组任务：实际完成的任务和尚未完成的任务，因此您只能迭代该done组：

done, pending = await asyncio.wait(tasks, timeout=5)
for t in done:
    ...