from typing import TYPE_CHECKING, Tuple from threading import Thread from queue import Queue from queue import Empty import blockdb if TYPE_CHECKING: from onionrblocks import Block from ..dandelion.phase import DandelionPhase def store_blocks( block_queues: Tuple["Queue[Block]", "Queue[Block]"], dandelion_phase: 'DandelionPhase'): new_queue: "Queue[Block]" = Queue() def _watch_queue(block_queue: "Queue[Block]"): # Copy all incoming blocks into 1 queue which gets processed to db while not dandelion_phase.is_stem_phase() \ and dandelion_phase.remaining_time() > 1: try: new_queue.put( block_queue.get(timeout=dandelion_phase.remaining_time())) except Empty: pass for block_queue in block_queues: Thread(target=_watch_queue, args=[block_queue], daemon=True).start() while not dandelion_phase.is_stem_phase() \ and dandelion_phase.remaining_time() > 1: try: blockdb.add_block_to_db( new_queue.get(timeout=dandelion_phase.remaining_time()) ) except Empty: pass