Czy mogę zapamiętać generator Pythona?

Question

Mam funkcję o nazwie runquery, która wykonuje wywołania do bazy danych, a następnie generuje wiersze, jeden po drugim. Napisałem dekorator memoize (a dokładniej, ukradłem go po prostu z this stackoverflow question), ale po kolejnych wywołaniach po prostu wydaje pustą sekwencję, prawdopodobnie dlatego, że wartości generatora można uzyskać tylko raz.

Jak mogę zmodyfikować dekorator zapamiętywania, który działa dla generatorów Pythona? Zdaję sobie sprawę, że w pewnym momencie będę musiał go zapisać w pamięci, ale chciałbym poradzić sobie z tym w dekoratorze i nie modyfikować oryginalnej funkcji.

Obecny kod funkcji memoization jest:

def memoized(f): 
    # Warning: Doesn't work if f yields values 
    cache={} 
    def ret(*args): 
     if args in cache: 
      return cache[args] 
     else: 
      answer=f(*args) 
      cache[args]=answer 
      return answer 
    return ret 

Answer 1

Zdaję sobie sprawę, jest to nieco starej pytanie, ale dla tych, którzy chcą pełne rozwiązanie: oto jeden, na podstawie sugestii jsbueno za:

from itertools import tee 
from types import GeneratorType 

Tee = tee([], 1)[0].__class__ 

def memoized(f): 
    cache={} 
    def ret(*args): 
     if args not in cache: 
      cache[args]=f(*args) 
     if isinstance(cache[args], (GeneratorType, Tee)): 
      # the original can't be used any more, 
      # so we need to change the cache as well 
      cache[args], r = tee(cache[args]) 
      return r 
     return cache[args] 
    return ret 

Answer 2

Tak. Jest dekorator opublikowany here. Zwróć uwagę, że jak mówi plakat, tracisz część korzyści z leniwego oceniania.

def memoize(func): 
    def inner(arg): 
     if isinstance(arg, list): 
      # Make arg immutable 
      arg = tuple(arg) 
     if arg in inner.cache: 
      print "Using cache for %s" % repr(arg) 
      for i in inner.cache[arg]: 
       yield i 
     else: 
      print "Building new for %s" % repr(arg) 
      temp = [] 
      for i in func(arg): 
       temp.append(i) 
       yield i 
      inner.cache[arg] = temp 
    inner.cache = {} 
    return inner 


@memoize 
def gen(x): 
    if not x: 
     yield 0 
     return 

    for i in xrange(len(x)): 
     for a in gen(x[i + 1:]): 
      yield a + x[0] 


print "Round 1" 
for a in gen([2, 3, 4, 5]): 
    print a 

print 
print "Round 2" 
for a in gen([2, 3, 4, 5]): 
    print a 

Answer 3

from itertools import tee 

sequence, memoized_sequence = tee (sequence, 2) 

Gotowe.

Generatory są łatwiejsze, ponieważ standardowa biblioteka ma tę metodę "tee"!