python-3.x - 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:
    ...