如何从含yield的生成器函数中提取重复代码？求通用方案

这问题我太懂了！碰到过好几次——带yield和return的代码块想复用，确实容易踩坑，毕竟生成器的执行逻辑和普通函数不一样，return在生成器里还会触发StopIteration，直接硬抽成普通helper方法肯定会乱套，甚至像你说的，代码更臃肿、效率还下降。下面分享几个通用的解决思路，亲测好用：

1. 用yield from包装生成器Helper #

这是最直接的方案，专门针对生成器场景。把重复的yield+return逻辑抽成独立的生成器函数，然后在原方法里用yield from来复用，它会自动帮你处理所有yield的传递，还能捕获生成器里return的返回值。

举个例子，假设你原来的重复代码是这样的：

# 第一段重复代码
yield calculate_value(1)
yield calculate_value(2)
if some_condition:
    return early_result
yield calculate_value(3)

# 第二段重复代码
yield calculate_value(1)
yield calculate_value(2)
if some_condition:
    return early_result
yield calculate_value(3)

抽成helper后：

def generator_helper(some_condition):
    yield calculate_value(1)
    yield calculate_value(2)
    if some_condition:
        return early_result
    yield calculate_value(3)

然后原方法里直接用：

def original_method():
    # 第一段逻辑复用
    result = yield from generator_helper(condition_a)
    # 如果需要用return的结果，直接用result
    process_result(result)

    # 第二段逻辑复用
    result = yield from generator_helper(condition_b)
    process_result(result)

这样既完全复用了代码，又保留了生成器的惰性特性，效率不会下降——反而因为减少了重复代码，字节码更简洁，运行效率说不定还能提一点。

2. 用嵌套闭包处理上下文依赖 #

如果重复代码需要访问原方法里的局部变量（比如一些临时状态、参数），不用把所有变量都当参数传给helper，直接把helper写成原方法的嵌套函数就行，它可以直接访问外部函数的上下文变量，代码更干净。

比如：

def original_method(input_data):
    temp_state = {}

    def helper():
        # 直接用外部的input_data和temp_state
        yield process_input(input_data, step=1)
        temp_state['intermediate'] = process_input(input_data, step=2)
        if temp_state['intermediate'] > 100:
            return temp_state['intermediate']
        yield process_input(input_data, step=3)

    # 第一次复用
    yield from helper()
    # 修改temp_state后第二次复用
    temp_state['reset'] = True
    yield from helper()

这种方式避免了大量参数传递，代码可读性反而更好。

3. 特殊场景：把return转为生成器终止信号 #

如果你的return只是用来终止生成器，不需要返回值，也可以在helper里用raise StopIteration代替return（不过Python 3.3+里生成器的return已经等价于这个了，所以还是推荐用yield from）。如果是老版本Python，这个方法也能凑合用。

比如：

def helper():
    yield 1
    yield 2
    if stop_condition:
        raise StopIteration
    yield 3

# 原方法里遍历helper
for item in helper():
    yield item

不过这个不如yield from简洁，优先用第一种方案。

最后提醒 #

别担心helper会增加代码量——短期看可能多了几行，但长期维护时，修改一次helper就能同步更新所有复用的地方，比改N段重复代码省心太多。而且生成器的性能开销真的很小，只要不是把生成器改成一次性返回列表（那才会浪费内存和效率），用yield from完全不会有性能问题。

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