Python协程怎样处理异常情况 - 问答

在Python中，协程（coroutine）是一种特殊的函数，可以在执行过程中暂停和恢复。处理协程中的异常情况需要使用asyncio库中的一些功能。以下是一些处理协程异常的方法：

使用try-except语句捕获异常：

在协程内部使用try-except语句捕获异常，就像在普通函数中一样。当异常发生时，它会被捕获并存储在except子句中。

import asyncio

async def my_coroutine():
    try:
        # 你的协程代码
    except Exception as e:
        print(f"捕获到异常: {e}")

asyncio.run(my_coroutine())

使用asyncio.gather处理多个协程的异常：

asyncio.gather函数可以同时运行多个协程，并收集它们的结果。如果其中一个协程引发异常，gather会立即停止执行剩余的协程，并将异常传递给return_exceptions参数。

import asyncio

async def my_coroutine(num):
    if num == 2:
        raise ValueError("这是一个故意引发的异常")
    return f"协程 {num} 完成"

async def main():
    coroutines = [my_coroutine(i) for i in range(1, 4)]
    results = await asyncio.gather(*coroutines, return_exceptions=True)
    for result in results:
        if isinstance(result, Exception):
            print(f"捕获到异常: {result}")
        else:
            print(result)

asyncio.run(main())

使用asyncio.create_task和await处理异常：

当你使用asyncio.create_task创建一个任务时，可以使用await关键字等待协程完成。如果协程引发异常，它会被捕获并存储在asyncio.Task对象的exception()方法中。

import asyncio

async def my_coroutine(num):
    if num == 2:
        raise ValueError("这是一个故意引发的异常")
    return f"协程 {num} 完成"

async def main():
    task = asyncio.create_task(my_coroutine(2))
    try:
        result = await task
    except Exception as e:
        print(f"捕获到异常: {e}")
    else:
        print(result)

asyncio.run(main())

这些方法可以帮助你处理Python协程中的异常情况。在实际编程中，你可能需要根据具体需求选择合适的方法。

0 赞

0 踩