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

时间:2019-04-16 13:30:35

标签: python-3.x python-asyncio

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'属性。

1 个答案:

答案 0 :(得分:2)

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

此外,asyncio.wait()本身返回两组任务:实际完成的任务和尚未完成的任务,因此您只能遍历done集:

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