如何在外部处理Python生成器抛出的异常后恢复其运行？

核心前提 #

Python 语言层面规定：生成器内部抛出未被捕获的异常时，会立即被标记为终止状态，后续所有next()/send()调用都会直接触发StopIteration，没有办法绕过这个规则。所有可行的恢复方案本质都是把异常控制在生成器内部不抛出到外部，再通过内部逻辑实现和外部的交互。

最优简化方案：装饰器统一封装异常转换逻辑 #

你自己实现的「异常作为值yield」的思路是完全可行的，只是可以通过装饰器把重复的异常捕获逻辑抽离，不需要每个生成器都手动写try-except块，侵入性极低：

from functools import wraps

# 通用装饰器，指定要捕获的异常类型，自动转为yield值
def yield_exceptions(*exc_types):
    def decorator(gen_func):
        @wraps(gen_func)
        def wrapper(*args, **kwargs):
            gen = gen_func(*args, **kwargs)
            while True:
                try:
                    yield next(gen)
                except exc_types as e:
                    yield e
                except StopIteration:
                    break
        return wrapper
    return decorator

使用时只需要给业务生成器加装饰器即可，原有业务代码不需要修改：

@yield_exceptions(ValueError)
def generator():
    yield "some data"
    raise ValueError("Something bad happened")
    yield "more data"

def do_something(thing): print(thing)
def handle_error(): print("caught ValueError")

def use_generator():
    gen = generator()
    while True:
        try:
            thing = next(gen)
            # 识别到异常值则执行处理逻辑
            if isinstance(thing, ValueError):
                handle_error()
                continue
            do_something(thing)
        except StopIteration:
            break

use_generator()

运行输出完全符合你的预期：

some data
caught ValueError
more data

进阶交互方案：send()双向通信 #

如果需要外部处理完异常后，给生成器返回处理结果、调整生成器后续运行逻辑，可以搭配生成器的send()方法实现双向交互：

def generator():
    yield "some data"
    try:
        raise ValueError("Something bad happened")
    except ValueError as e:
        # 抛出异常给外部，等待外部返回处理结果
        process_res = yield e
        # 可根据process_res调整后续逻辑，这里省略
    yield "more data"

def use_generator():
    gen = generator()
    send_val = None
    while True:
        try:
            thing = gen.send(send_val)
            send_val = None
            if isinstance(thing, ValueError):
                handle_error()
                # 把处理结果传回生成器
                send_val = "recovered"
                continue
            do_something(thing)
        except StopIteration:
            break

内容的提问来源于stack exchange，提问作者BurnNote