(output, capacity, global_init_net, global_exit_net)
| 35 | |
| 36 | |
| 37 | def _init_output(output, capacity, global_init_net, global_exit_net): |
| 38 | if output is None: |
| 39 | out_queue = queue_util.Queue( |
| 40 | capacity=( |
| 41 | capacity if capacity is not None |
| 42 | else DEFAULT_QUEUE_CAPACITY)) |
| 43 | writer = out_queue.writer() |
| 44 | elif isinstance(output, Writer): |
| 45 | assert capacity is None, 'capacity would not be used.' |
| 46 | out_queue = None |
| 47 | writer = output |
| 48 | elif hasattr(output, 'writer'): |
| 49 | assert capacity is None, 'capacity would not be used.' |
| 50 | out_queue = output |
| 51 | writer = output.writer() |
| 52 | else: |
| 53 | raise ValueError('output must be a reader, queue or stream.') |
| 54 | writer.setup_ex(global_init_net, global_exit_net) |
| 55 | return out_queue, writer |
| 56 | |
| 57 | |
| 58 | def make_processor(processor, reader=None): |
no test coverage detected
searching dependent graphs…