W pythonie istnieje sposób sprawdzenia, czy funkcja jest "funkcją generatora" przed wywołaniem?

Question

Powiedzmy mam dwie funkcje:

def foo(): 
    return 'foo' 

def bar(): 
    yield 'bar' 

Pierwszy z nich to normalne funkcjonowanie, a druga jest funkcją generatora. Teraz chcę napisać coś takiego:

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

Co będzie prosta implementacja is_generator_function() wyglądać? Korzystając z pakietu types mogę przetestować, czy gen jest generatorem, ale chcę to zrobić przed wywołaniem func().

Rozważmy teraz następujący przypadek:

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

inwokacji z goo() powróci generator. Zakładam, że parser Pythona wie, że funkcja goo() ma deklarację zysku, i zastanawiam się, czy możliwe jest łatwe uzyskanie tych informacji.

Dzięki!

Answer 1

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

• nowego w wersji 2.6 Pythona

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   
>>> 

Jak widać, klucz różnicą, że kod bajtowy dla bar będzie zawierać co najmniej jeden YIELD_VALUE opcodu. Polecam użycie modułu dis (przekierowanie jego wyjścia do instancji StringIO i sprawdzenie jej getvalue, oczywiście), ponieważ zapewnia to pewną miarę odporności na zmiany kodu bajtowego - zmienią się dokładne wartości liczbowe kodów, ale zdemontowana wartość symboliczna pozostanie całkiem stabilny ;-).

Answer 3

Właściwie, zastanawiam się, jak użyteczne byłoby takie hipotetyczne is_generator_function(). Rozważmy:

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

Co powinien is_generator_function() powrót do baz i quux? baz() zwraca generator, ale nie jest nim sam, a quux() może zwrócić generator lub nie.

Answer 4

I zostały wdrożone dekorator że haki na dekorowanej funkcji zwracany/przyniosły wartości. Jego podstawowa funkcja to:

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 

Działa, ponieważ funkcja dekoratora jest bezwarunkowo nazywana: jest to wartość zwracana, która jest testowana.

---

EDIT: Zgodnie z komentarzem Roberta Lujo, skończyło się z czymś takim:

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