Python üreteci işlevinin ilk verimden hemen önce ilerlemesi

Question

Bir jeneratör işlevini başlattığınızda, üzerinde next numaralı çağrıyı çağırana kadar hiçbir kod yürütmez.

Bu, jeneratör işlevi bir çeşit başlatma kodu içeriyorsa, yinelenene kadar çalıştırılmayacağı anlamına gelir.

def generator(filename): 
    with open(filename) as f: 
     data = f.read() 

    while True: 
     yield data 

gen = generator('/tmp/some_file') 
# at this point, no generator code is executed 

# initialization code is executed only at the first iteration 
for x in gen: 
    pass 

dosya yoksa, istisna döngü de yükseltilecek:

Bu örneği ele alalım. Jeneratörün yinelenmesinden önce çalıştırılacak ilk yield'dan önceki kod gibi kodlarını beğenirim, böylece başlatma sırasında tüm özel durumlar jeneratör örneğinde yükseltilecektir.

Bunu yapmanın temiz pythonik bir yolu var mı?

Answer 1

Wrap jeneratör etrafında düzenli bir fonksiyonu ve orada başlatma koyun: (. Bu durumda, genel olarak itertools.repeat ile iç jeneratörü değiştirin ama olamazdı)

def file_contents_repeated(filename): 
    with open(filename) as f: 
     data = f.read() 
    return _gen(data) 

def _gen(data): 
    while True: 
     yield data 

Answer 2

işlevi sarıcı çözümü iyi çalışıyor. Tüm kodu aynı işlevde tutmak istiyorsanız, bir kapatma oluşturabilirsiniz.

def generator(filename): 
    with open(filename) as f: 
     data = f.read() 

    def _generator(): 
     while True: 
      yield data 

    return _generator()