python - 如何使用 for 循环而不是 while 循环遍历 Python Queue.Queue？

Question

通常我们这样编码:

while True:
    job = queue.get()
    ...

但是是否也可以按照以下方式做一些事情:

for job in queue.get():
    #do stuff to job

我想这样做的真正原因是因为我想使用 python-progressbar 的自动检测 maxval。他们这样做就像 for this in progressbar(that):

Answer 1

您可以使用 iter与可调用。 (您应该传递两个参数，一个用于可调用对象，另一个用于标记值)

for job in iter(queue.get, None): # Replace `None` as you need.
    # do stuff with job

注意 当没有元素剩余并且没有放置哨兵值时，这将阻塞。此外，与 while-get 循环类似，与容器上的普通 for 循环不同，它会从队列中删除项目。

None 是通用值，所以这里有一个更具体的哨兵值示例:

sentinel = object()
for job in iter(queue.get, sentinel):
    # do stuff with job