Fix setting None update workers not causing all threads to stop

This commit is contained in:
Lonami Exo 2017-09-30 18:44:37 +02:00
parent 8c3c990e74
commit f1bca0fd06

View File

@ -79,9 +79,12 @@ class UpdateState:
"""Raises "StopIterationException" on the worker threads to stop them,
and also clears all of them off the list
"""
if self._workers:
with self._updates_lock:
# Insert at the beginning so the very next poll causes an error
# on all the worker threads
# TODO Should this reset the pts and such?
for _ in range(self._workers):
self._updates.appendleft(StopIteration())
self._updates_available.set()