在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