Pythonでは、関数を呼び出す前に関数が "ジェネレータ関数"であるかどうかをチェックする方法はありますか？

Question

2つの関数があるとしましょう：

def foo(): 
    return 'foo' 

def bar(): 
    yield 'bar' 

最初の関数は通常の関数であり、2つ目はジェネレータ関数です。今私はこのようなものを書いておきたいと思います：

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という名前で呼び出されるため動作します。これはテストされる戻り値です。

---

EDIT：はロバートLujoによってコメントに続き、私のようなものになってしまった：

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