coalescestreams.py
75 lines
| 2.4 KiB
| text/x-python
|
PythonLexer
|
r10674 | """Module that allows latex output notebooks to be conditioned before | ||
they are converted. Exposes a decorator (@cell_preprocessor) in | ||||
addition to the coalesce_streams pre-proccessor. | ||||
""" | ||||
#----------------------------------------------------------------------------- | ||||
# Copyright (c) 2013, the IPython Development Team. | ||||
# | ||||
# Distributed under the terms of the Modified BSD License. | ||||
# | ||||
# The full license is in the file COPYING.txt, distributed with this software. | ||||
#----------------------------------------------------------------------------- | ||||
|
r10485 | |||
|
r10674 | #----------------------------------------------------------------------------- | ||
# Functions | ||||
#----------------------------------------------------------------------------- | ||||
|
r10485 | |||
|
r10674 | def cell_preprocessor(function): | ||
""" | ||||
Wrap a function to be executed on all cells of a notebook | ||||
Wrapped Parameters | ||||
---------- | ||||
cell : NotebookNode cell | ||||
Notebook cell being processed | ||||
resources : dictionary | ||||
Additional resources used in the conversion process. Allows | ||||
transformers to pass variables into the Jinja engine. | ||||
index : int | ||||
Index of the cell being processed | ||||
""" | ||||
def wrappedfunc(nb, resources): | ||||
|
r10485 | for worksheet in nb.worksheets : | ||
for index, cell in enumerate(worksheet.cells): | ||||
|
r10674 | worksheet.cells[index], resources = function(cell, resources, index) | ||
return nb, resources | ||||
|
r10485 | return wrappedfunc | ||
@cell_preprocessor | ||||
|
r10674 | def coalesce_streams(cell, resources, index): | ||
""" | ||||
Merge consecutive sequences of stream output into single stream | ||||
to prevent extra newlines inserted at flush calls | ||||
Parameters | ||||
---------- | ||||
cell : NotebookNode cell | ||||
Notebook cell being processed | ||||
resources : dictionary | ||||
Additional resources used in the conversion process. Allows | ||||
transformers to pass variables into the Jinja engine. | ||||
index : int | ||||
Index of the cell being processed | ||||
""" | ||||
|
r10485 | outputs = cell.get('outputs', []) | ||
if not outputs: | ||||
|
r10674 | return cell, resources | ||
|
r10485 | last = outputs[0] | ||
new_outputs = [last] | ||||
|
r10674 | |||
|
r10485 | for output in outputs[1:]: | ||
if (output.output_type == 'stream' and | ||||
last.output_type == 'stream' and | ||||
last.stream == output.stream | ||||
): | ||||
last.text += output.text | ||||
else: | ||||
new_outputs.append(output) | ||||
|
r12339 | last = output | ||
|
r10485 | |||
cell.outputs = new_outputs | ||||
|
r10674 | return cell, resources | ||