B KZ@sBdZddlZddlZddlmZddZedZeddZdS) zHPreprocessor for merging consecutive stream outputs for easier handling.N) get_loggercstfdd}|S)a Wrap a function to be executed on all cells of a notebook The wrapped function should have these parameters: cell : NotebookNode cell Notebook cell being processed resources : dictionary Additional resources used in the conversion process. Allows preprocessors to pass variables into the Jinja engine. index : int Index of the cell being processed csFtdjx,t|jD]\}}|||\|j|<}qW||fS)NzApplying preprocessor: %s)rdebug__name__ enumerateZcells)Znb resourcesindexcell)functionFlib/python3.7/site-packages/nbconvert/preprocessors/coalescestreams.py wrappedfuncs  z&cell_preprocessor..wrappedfunc) functoolswraps)r r r )r r cell_preprocessor srz .*\r(?=[^\n])cCs|dg}|s||fS|d}|g}xT|ddD]D}|jdkrj|jdkrj|j|jkrj|j|j7_q4|||}q4Wx0|D](}|jdkrd|jkrtd|j|_qW||_||fS)a 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 outputsrNstream )getZ output_typenametextappendcr_patsubr)rrrrZlastZ new_outputsoutputr r r coalesce_streams$s"      r) __doc__rer Z traitlets.logrrcompilerrr r r r s