python 协程如何不被取消

Z时代
2024-01-10
分类：技术分享

python 协程如何不被取消

我开始学习python中的协程，学习到asyncio.shield和asyncio.wait_for这两个函数的时候有一个想法。

我想写一个函数，这个函数会因为超时而被asyncio.wait_for而取消，但是又会因这个函数asyncio.shield而不被取消。

import asyncio
import time
async def factorial(name, number):
    f = 1
    for i in range(2, number + 1):
        print(f"Task {name}: Compute factorial({i})...")
        await asyncio.sleep(1)
        f *= i
    print(f"Task {name}: factorial({number}) = {f}")
async def main():
    try:
        await asyncio.wait_for(
            asyncio.shield(
                factorial("func", 3)
            ), timeout=0
        )
    except asyncio.TimeoutError:
        print("Time out")
asyncio.run(main())time.sleep(1)

发现并没有成功。
后面修改了一下，但是蒙了。

async def main():
    try:
        await asyncio.wait_for(
            asyncio.shield(
                factorial("func", 3)
            ), timeout=0
        )
    except asyncio.TimeoutError:
        print("Time out")        await asyncio.sleep(10)

感觉是因为main函数的执行结束取消了。
但是最后一句await ... 改成time.sleep(10)又不行了。
求解释解释。。。