python中终止协程和异常处理方式

相信你对Python异步编程和协程有一定了解，我分别为你解释Python中终止协程和异常处理方式的攻略。

Python中终止协程的方式

协程是Python异步编程中重要的一部分，在协程中可能会出现需要提前终止协程的情况。Python中提供了3种终止协程的方式，分别是协程的throw()方法、close()方法和cancel()方法。

throw()方法

一种终止协程的方式是通过协程对象的throw()方法，抛出一个异常来中断协程的执行。这种方法可以在协程任何位置调用，需要注意的是被抛出的异常必须是BaseException的子类，否则抛出的将是TypeError。

下面是一个简单的示例代码：

import asyncio

async def foo():
    try:
        while True:
            await asyncio.sleep(1)
    except asyncio.CancelledError:
        print("Coroutine interrupted.")

async def main():
    task = asyncio.create_task(foo())
    await asyncio.sleep(3)
    task.throw(asyncio.CancelledError)
    await task

asyncio.run(main())

这个示例协程对象foo()中循环睡眠1秒钟，我们通过task.throw(asyncio.CancelledError)发出`Asyncio系统定义的CancelledError异常，从而中断了这个协程。

close()方法

第二种终止协程的方法是通过协程对象的close()方法。当我们调用close()方法时，协程对象抛出GeneratorExit异常，并让协程自然结束。

下面是一个简单的示例代码：

import asyncio

async def foo():
    try:
        while True:
            await asyncio.sleep(1)
    except GeneratorExit:
        print("Coroutine interrupted.")

async def main():
    task = asyncio.create_task(foo())
    await asyncio.sleep(3)
    task.close()
    await task

asyncio.run(main())

这个示例协程对象foo()中循环睡眠1秒钟，我们通过task.close()的方式中断了这个协程，并让其自然结束。

cancel()方法

第三种终止协程的方法是通过协程对象的cancel()方法。当我们调用cancel()方法时，协程对象抛出CancelledError异常，并让协程自然结束。

下面是一个简单的示例代码：

import asyncio

async def foo():
    try:
        while True:
            await asyncio.sleep(1)
    except asyncio.CancelledError:
        print("Coroutine interrupted.")

async def main():
    task = asyncio.create_task(foo())
    await asyncio.sleep(3)
    task.cancel()
    await task

asyncio.run(main())

这个示例协程对象foo()中循环睡眠1秒钟，我们通过task.cancel()的方式中断了这个协程，并让其自然结束。

异常处理方式

在Python异步编程中，我们可能需要处理协程中的异步异常。一些Python自带的异步库，比如异步请求库aiohttp，已经提供了一些具体的异常类型，但在其他异步库中，我们可能需要自定义异常类型，或者对特定异常类型进行处理。

我们可以通过try-except语句或者asyncio提供的特定异常处理函数，对异步异常进行处理。

异常处理函数

首先介绍asyncio提供的异步异常处理函数：

• try-except: 我们可以使用try-except语句捕获协程中的异常，然后针对不同类型的异常进行不同的处理。例如：

import asyncio

async def foo():
    try:
        raise Exception("Error in coroutine.")
    except:
        print("Coroutine exception caught.")

async def main():
    await foo()

asyncio.run(main())

在以上的示例代码中，我们在协程函数foo()中抛出了一个异常，然后在try-except语句块中对异常进行处理，最后输出Coroutine exception caught.。

• asyncio.add_exception_handler(): asyncio提供了asyncio.add_exception_handler()函数用于注册异步异常处理器。通过该函数，我们可以将自定义的异步异常处理器添加到asyncio事件循环，对指定的协程异步异常进行处理。例如：

import asyncio

async def foo():
    raise Exception("Error in coroutine.")

def my_exception_handler(loop, context):
    print("Coroutine exception caught.")

async def main():
    asyncio.get_event_loop().set_exception_handler(my_exception_handler)
    await foo()

asyncio.run(main())

在以上示例代码中，我们定义了一个自定义的异步异常处理器my_exception_handler()，然后通过asyncio.get_event_loop().set_exception_handler()函数指定该处理器处理协程中的异步异常。最后，我们调用foo()函数并执行asyncio.run()，可以看到输出Coroutine exception caught.。

协程内部处理异常

除了使用全局异步异常处理函数外，我们还可以在协程内部处理异步异常。即在协程函数中使用try-except语句对协程内部的异常进行捕获和处理。例如：

import asyncio

async def foo():
            try:
                raise Exception("Coroutine error.")
            except:
                print("Coroutine exception caught.")

async def main():
    await foo()

asyncio.run(main())

在以上示例代码中，我们定义了一个协程函数foo()，在其中抛出了一个异常，并在try-except语句块中对异常进行了捕获和处理。最终输出Coroutine exception caught.。

处理协程内部异常需要注意的是，异步异常只能在协程内部处理。如果协程内部没有显式处理异常，那么异步异常将会一直向外抛出，直至被全局异常处理函数捕获。如果没有全局异常处理函数的话，程序将会异常终止。