From bee33f2df629a702dd61345e14d807720b94e3e9 Mon Sep 17 00:00:00 2001 From: Devaev Maxim Date: Tue, 13 Oct 2020 17:27:36 +0300 Subject: tools.clear_queue() --- kvmd/tools.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'kvmd/tools.py') diff --git a/kvmd/tools.py b/kvmd/tools.py index 6d10e227..555ddc3e 100644 --- a/kvmd/tools.py +++ b/kvmd/tools.py @@ -22,6 +22,8 @@ import operator import functools +import multiprocessing.queues +import queue from typing import Tuple from typing import List @@ -53,3 +55,12 @@ _DictValueT = TypeVar("_DictValueT") def sorted_kvs(dct: Dict[_DictKeyT, _DictValueT]) -> List[Tuple[_DictKeyT, _DictValueT]]: return sorted(dct.items(), key=operator.itemgetter(0)) + + +# ===== +def clear_queue(q: multiprocessing.queues.Queue) -> None: # pylint: disable=invalid-name + for _ in range(q.qsize()): + try: + q.get_nowait() + except queue.Empty: + break -- cgit v1.2.3