python - 即使忽略了 CancelledError，如何取消任务执行？

Question

这是取消任务的示例:

import asyncio


async def some_func():
    await asyncio.sleep(2)
    print('Haha! Task keeps running!')
    await asyncio.sleep(2)

async def cancel(task):
    await asyncio.sleep(1)
    task.cancel()

async def main():
    func_task = asyncio.ensure_future(some_func())
    cancel_task = asyncio.ensure_future(cancel(func_task))
    try:
        await func_task
    except asyncio.CancelledError:
        print('Task cancelled as expected')

if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())

# Task cancelled as expected
# [Finished in 1.2s]

工作正常，任务已取消。如果在 some_func 中捕获 CancelledError 任务将不会被取消:

async def some_func():
    try:
        await asyncio.sleep(2)
    except:
        pass
    print('Haha! Task keeps running!')
    await asyncio.sleep(2)

# Haha! Task keeps running!
# [Finished in 3.2s]

很容易忘记我不应该在异步代码中的任何地方抑制异常(例如，some_func 可以是第三方代码)，但是任务应该取消。我能做到吗？或者忽略 CancelledError 意味着任务根本无法取消？

Answer 1

您无法取消抑制 CancelledError 的任务。 这类似于无法关闭生成器而忽略 GeneratorExit .

这是故意的行为。任务可能需要在取消时做一些额外的工作(例如资源清理)，从而捕获 CancelledError可能是个好主意，但抑制通常是编程错误的标志。

如果你有不妥协的意图，Python 通常允许你搬起石头砸自己的脚。

捕获所有异常甚至禁止按 <Ctrl+C> 关闭 python 进程因为它被翻译成KeyboardInterrupt内部。