Python 生成器预取？

Question

我有一个生成器，每次迭代都需要很长时间才能运行。有没有一种标准的方法让它产生一个值，然后在等待再次调用时生成下一个值？

每次在 gui 中按下按钮时都会调用生成器，并且希望用户在每次按下按钮后考虑结果。

编辑:解决方法可能是:

def initialize():
    res = next.gen()

def btn_callback()
    display(res)
    res = next.gen()
    if not res:
       return

Answer 1

如果我想做类似你的解决方法的事情，我会写一个这样的类:

class PrefetchedGenerator(object):
    def __init__(self, generator):
         self._data = generator.next()
         self._generator = generator
         self._ready = True

    def next(self):
        if not self._ready:
            self.prefetch()
        self._ready = False
        return self._data

    def prefetch(self):
        if not self._ready:
            self._data = self._generator.next()
            self._ready = True

它比你的版本更复杂，因为我这样做是为了让它处理不调用预取或调用预取太多次。基本思想是在需要下一项时调用 .next() 。当您有“时间”可以消磨时，您可以调用预取。

你的另一个选择是线程..

class BackgroundGenerator(threading.Thread):
    def __init__(self, generator):
        threading.Thread.__init__(self)
        self.queue = Queue.Queue(1)
        self.generator = generator
        self.daemon = True
        self.start()

    def run(self):
        for item in self.generator:
            self.queue.put(item)
        self.queue.put(None)

    def next(self):
            next_item = self.queue.get()
            if next_item is None:
                 raise StopIteration
            return next_item

这将与您的主应用程序分开运行。无论获取每次迭代需要多长时间，您的 GUI 都应该保持响应。