B £'=[~ã@s dZddlmZmZddlmZddlmZmZm Z dd„Z ddd d d d gZ d d„Z dd„Z dd„Zdd„Zdd„Zdd„ZGdd„deƒZGdd„deƒZdS)z3Base classes and utilities for readers and writers.é)Ú encodestringÚ decodestring)Ú py3compat)Ú 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)zôRestore 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. Note: this is never used ÚcodeÚpngÚasciiÚjpeg)Ú worksheetsÚcellsÚ cell_typeÚoutputsrr r )ÚnbÚwsÚcellÚoutput©rú1lib/python3.7/site-packages/nbformat/v3/rwbase.pyÚ restore_bytes s    rÚtextZhtmlZsvgZlatexZ javascriptZjsoncCs*|r|d d¡rd |¡Sd |¡SdS)zšjoin lines that have been written by splitlines() Has logic to protect against `splitlines()`, which should have been `splitlines(True)` r)Ú ú ÚrN)ÚendswithÚjoin)ÚlinesrrrÚ _join_lines#s rcCsÀxº|jD]°}xª|jD] }|jdkr„d|krBt|jtƒrBt|jƒ|_xp|jD]4}x.tD]&}|  |d¡}t|tƒrTt|ƒ||<qTWqJWqx.dD]&}|  |d¡}t|tƒrŠt|ƒ||<qŠWqWqW|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. rÚinputN)ÚsourceÚrendered) r r rÚ isinstancerÚlistrrÚ_multiline_outputsÚget)rrrrÚkeyÚitemrrrÚ rejoin_lines1s           r(cCsÆxÀ|jD]¶}x°|jD]¦}|jdkrˆd|krDt|jtƒrD|j d¡|_xt|jD]6}x0tD](}|  |d¡}t|tƒrV| d¡||<qVWqLWqx0dD](}|  |d¡}t|tƒrŽ| d¡||<qŽWqWqW|S)zásplit 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. rrTN)r r!) r r rr"rrÚ splitlinesrr$r%)rrrrr&r'rrrÚ split_linesMs          r*cCsšx”|jD]Š}x„|jD]z}|jdkrxj|jD]`}d|kr\t|jtƒrP|j d¡|_t|jƒ|_d|kr*t|j tƒr~|j  d¡|_ t|j ƒ|_ q*WqWqW|S)zmRestore all bytes objects in the notebook from base64-encoded strings. Note: This is never used rr r r ) r r rrr"r rÚencoderr )rrrrrrrÚ base64_decodeis       r,cCsrxl|jD]b}x\|jD]R}|jdkrxB|jD]8}d|krHt|jƒ d¡|_d|kr*t|jƒ d¡|_q*WqWqW|S)zŠBase64 encode all bytes objects in the notebook. These will be b64-encoded unicode strings Note: This is never used rr r r )r r rrrr Údecoder )rrrrrrrÚ base64_encode}s    r.cCs`| dd¡| dd¡xB|dD]6}x0|dD]$}| di¡ dd¡| dd¡q0Wq"W|S)zsStrip transient values that shouldn't be stored in files. This should be called in *both* read and write. Z orig_nbformatNZorig_nbformat_minorr r ZmetadataZtrusted)Úpopr%)rrrrrrÚstrip_transients  r0c@s eZdZdZdd„Zdd„ZdS)ÚNotebookReaderzA class for reading notebooks.cKs tdƒ‚dS)zRead a notebook from a string.z'loads must be implemented in a subclassN)ÚNotImplementedError)ÚselfÚsÚkwargsrrrÚreads¢szNotebookReader.readscKs0| ¡}tjs"t|tƒs"t |¡}|j|f|ŽS)z'Read a notebook from a file like object)ÚreadrÚPY3r"rÚstr_to_unicoder6)r3Úfpr5Únbsrrrr7¦s zNotebookReader.readN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r6r7rrrrr1Ÿsr1c@s eZdZdZdd„Zdd„ZdS)ÚNotebookWriterzA class for writing notebooks.cKs tdƒ‚dS)zWrite a notebook to a string.z'loads must be implemented in a subclassN)r2)r3rr5rrrÚwrites±szNotebookWriter.writescKs2|j|f|Ž}tjs(t|tƒs(t |¡}| |¡S)z&Write a notebook to a file like object)rArr8r"rr9Úwrite)r3rr:r5r;rrrrBµs zNotebookWriter.writeN)r<r=r>r?rArBrrrrr@®sr@N)r?Úbase64rrZipython_genutilsrZipython_genutils.py3compatrrrrr$rr(r*r,r.r0Úobjectr1r@rrrrÚs