B '=[m@sdZddlmZmZddlZddlmZmZmZddZ ddd d d d gZ d dZ ddZ ddZ ddZGdddeZGdddeZdS)zOBase classes and utilities for readers and writers. Authors: * Brian Granger ) encodestring decodestringN) str_to_bytes unicode_type string_typescCsjxd|jD]Z}xT|jD]J}|jdkrx:|jD]0}d|krDt|jd|_d|kr*t|jd|_q*WqWqW|S)zRestore bytes of image data from unicode-only formats. Base64 encoding is handled elsewhere. Bytes objects in the notebook are always b64-encoded. We DO NOT encode/decode around file formats. codepngasciijpeg) worksheetscells cell_typeoutputsrrr )nbwscelloutputr1lib/python3.7/site-packages/nbformat/v2/rwbase.py restore_bytess    rtextZhtmlZsvgZlatexZ javascriptZjsoncCsx|jD]}x|jD]}|jdkrd|krDt|jtrDd|j|_xt|jD]6}x0tD](}| |d}t|trVd|||<qVWqLWqx0dD](}| |d}t|trd|||<qWqWqW|S)a2rejoin multiline text into strings For reversing effects of ``split_lines(nb)``. This only rejoins lines that have been split, so if text objects were not split they will pass through unchanged. Used when reading JSON files that may have been passed through split_lines. rinput N)sourcerendered) r r r isinstancerlistjoinr_multiline_outputsget)rrrrkeyitemrrr rejoin_lines/s          r"cCsx|jD]}x|jD]}|jdkrd|krBt|jtrB|j|_xp|jD]4}x.tD]&}| |d}t|trT|||<qTWqJWqx.dD]&}| |d}t|tr|||<qWqWqW|S)zsplit likely multiline text into lists of strings For file output more friendly to line-based VCS. ``rejoin_lines(nb)`` will reverse the effects of ``split_lines(nb)``. Used when writing JSON files. rrN)rr) r r r rrr splitlinesrrr)rrrrr r!rrr split_linesKs           r$cCsx|jD]}x|jD]z}|jdkrxj|jD]`}d|kr\t|jtrP|jd|_t|j|_d|kr*t|j tr~|j d|_ t|j |_ q*WqWqW|S)zmRestore all bytes objects in the notebook from base64-encoded strings. Note: This is never used rrr r ) r r r rrrrencoderr )rrrrrrr base64_decodegs       r&cCsrxl|jD]b}x\|jD]R}|jdkrxB|jD]8}d|krHt|jd|_d|kr*t|jd|_q*WqWqW|S)zBase64 encode all bytes objects in the notebook. These will be b64-encoded unicode strings Note: This is never used rrr r )r r r rrrdecoder )rrrrrrr base64_encode{s    r(c@s eZdZdZddZddZdS)NotebookReaderzA class for reading notebooks.cKs tddS)zRead a notebook from a string.z'loads must be implemented in a subclassN)NotImplementedError)selfskwargsrrrreadsszNotebookReader.readscKs|j|f|S)z'Read a notebook from a file like object)read)r+fpr-rrrr/szNotebookReader.readN)__name__ __module__ __qualname____doc__r.r/rrrrr)sr)c@s eZdZdZddZddZdS)NotebookWriterzA class for writing notebooks.cKs tddS)zWrite a notebook to a string.z'loads must be implemented in a subclassN)r*)r+rr-rrrwritesszNotebookWriter.writescKs||j|f|S)z&Write a notebook to a file like object)writer6)r+rr0r-rrrr7szNotebookWriter.writeN)r1r2r3r4r6r7rrrrr5sr5)r4base64rrZpprintZipython_genutils.py3compatrrrrrr"r$r&r(objectr)r5rrrrs