使用生成器作为进度通知程序

Question

我目前正在使用生成器作为获取长进程进度的快速方法，我想知道它通常是如何完成的，因为我发现它不是很优雅......

首先让我解释一下，我有一个 engine.py 模块，它可以执行以下操作：一些视频处理（分割、背景/背景减法等）需要花费大量时间（从几秒到几分钟）。

我通过 wxpython 和控制台脚本编写的 GUI 使用此模块。 当我查看如何在 wxpython 中实现进度对话框时，我发现我必须以某种方式获取进度值来更新我的对话框，您会承认这是纯粹的逻辑...... 因此，我决定使用引擎函数中处理的帧数，每 33 帧生成当前帧数，并在处理完成时生成 None 。

通过这样做，它看起来像这样：

dlg = wx.ProgressDialog("Movie processing", "Movie is being written...",
                           maximum = self.engine.endProcessingFrame,self.engine.startProcessingFrame,
                           parent=self,
                           style = wx.PD_APP_MODAL | wx.PD_ELAPSED_TIME | wx.PD_SMOOTH | wx.PD_CAN_ABORT)
state = self.engine.processMovie()
f = state.next()
while f != None:
    c, s = dlg.Update(f, "Processing frame %d"%f)
    if not c:break
    f = state.next()
dlg.Destroy()

效果很好，绝对没有明显的速度损失，但我希望能够调用 processMovie() 函数，而不必处理生成器（如果我不想）。

例如，我使用引擎模块的控制台脚本不关心进度，我可以使用它，但它注定要在没有显示的环境中执行，所以我真的不关心进度......

有人有我想出的另一种设计吗？ （使用线程、全局变量、进程等）

我认为必须有一个设计可以干净地完成这项工作:-)

Answer 1

这听起来像是一个完美的事件案例。 该进程发送一个“状态更新事件”，任何想知道的人（在本例中为对话框）都会监听该事件。

Answer 2

首先，如果你使用生成器，你不妨将它用作迭代器：

state = self.engine.processMovie()

for f in state:
    c, s = dlg.Update(f, "Processing frame %d"%f)
    if not c:
        break

dlg.Destroy()

并且不要产生 None; 完成后停止让步并离开该函数； 或者引发 StopIteration。 这是结束生成的正确方法（当使用 for 循环时，这是必要的）。

除此之外，我喜欢这个主意。 在我看来，这是生成器的一种非常有效的用途。

您可能希望使 33 可配置（即可作为参数传递给 processMovie）； 33 似乎是一个任意的选择，如果您处理一部两小时的电影，我想没有必要每 33 帧更新一次进度条。

Answer 3

使用生成器对此很好，但使用生成器的全部意义在于您可以使用内置语法：

for f in self.engine.processMovie():
    c, s = dlg.Update(f, "Processing frame %d"%f)
    if not c: break

如果您不关心这一点，那么您可以说：

for f in self.engine.processMovie(): pass

或公开一个函数（例如engine.processMovieFull）来执行给你的。

您还可以使用简单的回调：

def update_state(f):
    c, s = dlg.Update(f, "Processing frame %d"%f)
    return c
self.engine.processMovie(progress=update_state)

...但如果您想零碎地处理数据，那就不太好； 回调模型更喜欢一次性完成所有工作——这就是生成器的真正好处。