在 Python 中包装生成器函数

Question

我正在编写一些代码来遍历可能具有循环引用的结构。 与其在递归函数的开头明确地进行检查，我认为我会创建一个装饰器，它不允许使用相同的参数多次调用函数。

以下是我想出的。如其所写，这将尝试遍历 Nonetype 并引发异常。我知道我可以通过返回一个空列表来修复它，但我想要更优雅。有没有办法从装饰器中判断被装饰的函数是否是生成器函数？这样，如果它是生成器，我可以有条件地引发 StopIteration，否则就返回 None。

previous = set()
def NO_DUPLICATE_CALLS(func):
    def wrapped(*args, **kwargs):
        if args in previous:
            print 'skipping previous call to %s with args %s %s' % (func.func_name, repr(args), repr(kwargs))
            return
        else:
            ret = func(*args, **kwargs)
            previous.add(args)
            return ret
    return wrapped

@NO_DUPLICATE_CALLS
def foo(x):
    for y in x:
        yield y

for f in foo('Hello'):
    print f

for f in foo('Hello'):
    print f

Answer 1

好的，看看这个：

>>> from inspect import isgeneratorfunction
>>> def foo(x):
...    for y in x:
...        yield y
...
>>> isgeneratorfunction(foo)
True

不过，这需要 Python 2.6 或更高版本。

Answer 2

不幸的是，没有真正的好方法可以知道一个函数是否会在不调用它的情况下返回某种类型的可迭代对象，请参阅 this answer 另一个问题以获得对一些潜在问题的很好解释。

但是，您可以通过使用修改过的 memoize 装饰器来解决这个问题。通常，记忆装饰器会使用先前参数的返回值创建一个缓存，但您可以只存储返回值的类型，而不是存储完整值。当您遇到参数时，您已经看到只是返回该类型的新初始化，这将导致空字符串、列表等。

这里是 memoize 装饰器的链接，可以帮助您入门：
http://wiki.python.org/moin/PythonDecoratorLibrary#Memoize