在python中有一种方法可以在调用它之前检查函数是否是“生成器函数”？

Question

可以说我有两个功能：

def foo(): 
    return 'foo' 

def bar(): 
    yield 'bar' 

第一个是一个正常的功能，第二个是发电机功能。现在，我想写的东西是这样的：

def run(func): 
    if is_generator_function(func): 
    gen = func() 
    gen.next() 
    #... run the generator ... 
    else: 
    func() 

什么会直接实现的is_generator_function()样子？使用types软件包，我可以测试gen是否为发生器，但我希望在调用func()之前这样做。

现在考虑以下情况：

def goo(): 
    if False: 
    yield 
    else: 
    return 

的goo()的调用将返回一个发电机。我认为Python解析器知道goo()函数有一个yield语句，我不知道是否可以轻松获取这些信息。

谢谢！

Answer 1

>>> import inspect 
>>> 
>>> def foo(): 
... return 'foo' 
... 
>>> def bar(): 
... yield 'bar' 
... 
>>> print inspect.isgeneratorfunction(foo) 
False 
>>> print inspect.isgeneratorfunction(bar) 
True 

• 新的Python 2.6版本

Answer 2

>>> def foo(): 
... return 'foo' 
... 
>>> def bar(): 
... yield 'bar' 
... 
>>> import dis 
>>> dis.dis(foo) 
    2   0 LOAD_CONST    1 ('foo') 
       3 RETURN_VALUE   
>>> dis.dis(bar) 
    2   0 LOAD_CONST    1 ('bar') 
       3 YIELD_VALUE   
       4 POP_TOP    
       5 LOAD_CONST    0 (None) 
       8 RETURN_VALUE   
>>> 

正如你看到的，关键的区别在于：bar字节码将包含至少一个YIELD_VALUE操作码。我推荐使用dis模块（将其输出重定向到StringIO实例并检查其当前的getvalue），因为这为您提供了对字节码更改的鲁棒性度量 - 操作码的确切数值将会改变，但反汇编的符号值将保持相当稳定;-)。

Answer 3

其实，我想知道这样一个假设的is_generator_function()真的有多有用。试想一下：

def foo(): 
    return 'foo' 
def bar(): 
    yield 'bar' 
def baz(): 
    return bar() 
def quux(b): 
    if b: 
     return foo() 
    else: 
     return bar() 

什么都要is_generator_function()回报baz和quux？ baz()返回一个发生器，但不是一个本身，并且quux()可能会返回一个发生器或可能不会。

Answer 4

，我实现了对装饰函数的返回/产生价值钩子装饰。它的基本云：

import types 
def output(notifier): 
    def decorator(f): 
     def wrapped(*args, **kwargs): 
      r = f(*args, **kwargs) 
      if type(r) is types.GeneratorType: 
       for item in r: 
        # do something 
        yield item 
      else: 
       # do something 
       return r 
    return decorator 

它的工作原理，因为装饰功能unconditionnaly称为：这是测试的返回值。

---

编辑：继罗伯特·路约的评论，我结束了类似：

def middleman(f): 
    def return_result(r): 
     return r 
    def yield_result(r): 
     for i in r: 
      yield i 
    def decorator(*a, **kwa): 
     if inspect.isgeneratorfunction(f): 
      return yield_result(f(*a, **kwa)) 
     else: 
      return return_result(f(*a, **kwa)) 
    return decorator