X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/16f704326f44fd1e5e5e60b936c9b5895d6a6ff8..0841b144012a6cd54c927c2141d72411b0c86070:/tools/crunchstat-summary/crunchstat_summary/reader.py diff --git a/tools/crunchstat-summary/crunchstat_summary/reader.py b/tools/crunchstat-summary/crunchstat_summary/reader.py index 311c006c07..8ccdbc2fcf 100644 --- a/tools/crunchstat-summary/crunchstat_summary/reader.py +++ b/tools/crunchstat-summary/crunchstat_summary/reader.py @@ -2,11 +2,9 @@ # # SPDX-License-Identifier: AGPL-3.0 -from __future__ import print_function - import arvados import itertools -import Queue +import queue import threading from crunchstat_summary import logger @@ -87,19 +85,21 @@ class LiveLogReader(object): self._queue.put(self.EOF) def __iter__(self): - self._queue = Queue.Queue() + self._queue = queue.Queue() self._thread = threading.Thread(target=self._get_all_pages) self._thread.daemon = True self._thread.start() return self - def next(self): + def __next__(self): line = self._queue.get() if line is self.EOF: self._thread.join() raise StopIteration return line + next = __next__ # for Python 2 + def __enter__(self): return self