U ku fîã@s¨dZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z Gdd„dej ƒZGd d „d eƒZGd d „d e ƒZeƒZeƒZejZejZejZejZejZdS) z(Read and write notebooks in JSON format.é)Ú annotationsN)Ú from_dicté)ÚNotebookReaderÚNotebookWriterÚ rejoin_linesÚ split_linesÚstrip_transientc@seZdZdZdd„ZdS)Ú BytesEncoderz@A JSON encoder that accepts b64 (and other *ascii*) bytestrings.cCs"t|tƒr| d¡Stj ||¡S)z#Get the default value of an object.Úascii)Ú isinstanceÚbytesÚdecodeÚjsonÚ JSONEncoderÚdefault)ÚselfÚobj©rú6/tmp/pip-unpacked-wheel-zj9qk9v2/nbformat/v4/nbjson.pyrs  zBytesEncoder.defaultN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrr sr c@s eZdZdZdd„Zdd„ZdS)Ú JSONReaderzA JSON notebook reader.cKs tj|f|Ž}|j|f|Ž}|S)z)Read a JSON string into a Notebook object)rÚloadsÚ to_notebook)rÚsÚkwargsÚnbrrrÚreadsszJSONReader.readscKst|ƒ}t|ƒ}t|ƒ}|S)z™Convert a disk-format notebook dict to in-memory NotebookNode handles multi-line values as strings, scrubbing of transient values, etc. )rrr )rÚdrrrrrr"szJSONReader.to_notebookN)rrrrr rrrrrrsrc@seZdZdZdd„ZdS)Ú JSONWriterzA JSON notebook writer.cKs`t|d<d|d<d|d<d|d<| dd ¡t |¡}| d d¡rJt|ƒ}t|ƒ}tj|f|ŽS) z0Serialize a NotebookNode object as a JSON stringÚclsrÚindentTÚ sort_keys)ú,z: Ú separatorsÚ ensure_asciiFr) r Ú setdefaultÚcopyÚdeepcopyÚpoprr rÚdumps)rrrrrrÚwrites0s   zJSONWriter.writesN)rrrrr.rrrrr"-sr")rÚ __future__rr*rZnbformat.notebooknoderZrwbaserrrrr rr rr"Z_readerZ_writerr ÚreadrÚwriter.rrrrÚs