意外的装饰器行为，“if”语句在 if 1==2 之后执行：

Question

А 几小时前我提交了一个related question 并得到了一个答案，为什么我需要将yield 添加到我的装饰器中才能正常运行。 我最近回忆起我省略了它是有原因的 - 我无法解释的奇怪行为。

如果我是盲人，我提前道歉，但我花了几个小时盯着和玩这段代码，这就是我得到的：

def decor(func):
    def wrapper(*args, **kwargs):
        if 1==2:
            print ("Generator")
            for item in func(*args, **kwargs):
                print(item)
                #yield(item)
        else:
            print ("Not generator")
            res = func(*args, **kwargs)
            print(res)
            return res
    return wrapper

@decor            
def f():
    return "a"


f()

"""
Output:
    Not generator
    a

"""

如果我删除yield 之前的评论，则根本没有输出。

这是为什么呢？我在if 1==2: 语句中更改的任何内容怎么可能对脚本产生任何影响？

Answer 1

如果一个函数在主体中包含yieldanywhere，那么它就是一个生成器函数。 yield 是否被执行并不重要。 1 == 2 为假的事实与此无关。

考虑以下函数：

def addone(numbers):
    for number in numbers:
        yield number + 1

当您拨打addone([]) 时会发生什么？ yield 永远不会执行，但 addone 仍然返回一个生成器。为什么会有所不同：

def addone(numbers):
    if numbers:
        for number in numbers:
            yield number + 1

因此很明显，yield 是否实际执行是无关紧要的。唯一相关的事实是 yield 是否存在于函数体中。

如何修复该功能

修复比较简单，只需将带有yield的部分拉到一个单独的函数中即可：

import types

def wrapper(*args, **kwargs):
    result = func(*args, **kwargs)
    if isinstance(result, types.GeneratorType):
        print("Is a generator")
        return wrap_generator(result)
    print("Not a generator")
    return result

def wrap_generator(gen):
    for item in gen:
        print(item)
        yield item

以后如何避免

一般来说，这里的问题是函数要么是生成器（并使用yield），要么是普通函数（并使用return）。在同一个函数中同时使用yield 和return 会有点混乱！

对于 Python，事实证明，如果在同一个函数中同时使用 yield 和 return，则该函数是生成器函数。这可能有点令人困惑，所以就风格而言，我通常会避免在同一个函数中同时使用 return 和 yield。