:vjrdZddlmZddlmZmZdZgdZdZdZ dZ d Z Gd d Z Gd d Z y)zOBase classes and utilities for readers and writers. Authors: * Brian Granger ) annotations) decodebytes encodebytesc|jD]}}|jD]l}|jdk(s|jD]K}d|vr |jj d|_d|vs,|j j d|_Mn|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_typeoutputsrencoder nbwscelloutputs c/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/nbformat/v2/rwbase.py restore_bytesrs| mmHHD~~'"llF%+ZZ%6%6w%? '&,kk&8&8&A + I)texthtmlsvglatex javascriptjsonc|jD]}|jD]}|jdk(rd|vr:t|jt r dj |j|_|jD]D}tD]9}|j|d}t|t s&dj |||<;FdD]9}|j|d}t|t s&dj |||<;|S)a&rejoin 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 Nsourcerendered) r r r isinstancerlistjoinr_multiline_outputsgetrrrrkeyitems r rejoin_linesr,0smmHHD~~'d?z$**d'C!%4::!6DJ"llF1%zz#t4%dD1*.))D/F3K 2+ 2C88C.D!$-$(IIdOS 2 Irc|jD]}|jD]}|jdk(rd|vr9t|jt r|jj |_|jD]C}tD]8}|j|d}t|t s&|j ||<:EdD]8}|j|d}t|t s&|j ||<:|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. rrNr!) r r r r$rstr splitlinesrr'r(r)s r split_linesr0LsmmHHD~~'d?z$**c'B!%!6!6!8DJ"llF1%zz#t4%dC0*.//*;F3K 2+ 2C88C.D!$,$(OO$5S 2 Irc|jD]}|jD]}|jdk(s|jD]}d|vrTt |j t r |j jd|_t|j |_d|vs`t |jt r |jjd|_ t|j|_ |S)ziRestore all bytes objects in the notebook from base64-encoded strings. Note: This is never used rrr r ) r r r rr$rr.rrr rs r base64_decoder2js mmHHD~~'"llF%fjj#6)/):):7)CFJ%0%< '%fkk37*0++*<* + IrcB|jD]}|jD]~}|jdk(s|jD]]}d|vr)t |j j d|_d|vs5t |jj d|__|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 rs r base64_encoder5~smmHHD~~'"llF%0%<%C%CG%L '&1&++&>&E&Eg&N + IrceZdZdZdZdZy)NotebookReaderzA class for reading notebooks.c d}t|)zRead a notebook from a string.'loads must be implemented in a subclassNotImplementedError)selfskwargsmsgs rreadszNotebookReader.reads7!#&&rc D|j|jfi|S)z'Read a notebook from a file like object)read)r<fpr>s rrCzNotebookReader.readstyy-f--rN)__name__ __module__ __qualname____doc__r@rCrrr7r7s(' .rr7ceZdZdZdZdZy)NotebookWriterzA class for writing notebooks.c d}t|)zWrite a notebook to a string.r9r:)r<rr>r?s rwriteszNotebookWriter.writesrArc F|j|j|fi|S)z&Write a notebook to a file like object)writerM)r<rrDr>s rrOzNotebookWriter.writes!xx  B1&122rN)rErFrGrHrMrOrIrrrKrKs(' 3rrKN)rH __future__rbase64rrrr'r,r0r2r5r7rKrIrrrRsH"#+$L8<($ . . 3 3r