U
    kufÁ  ã                   @  s¨   d Z ddlmZ ddlZddlZddlmZ ddlmZm	Z	m
Z
mZmZ G dd„ dejƒZG d	d
„ d
eƒZG dd„ de	ƒZeƒ Zeƒ ZejZejZejZejZejZdS )z(Read and write notebooks in JSON format.é    )ÚannotationsNé   )Ú	from_dict)ÚNotebookReaderÚNotebookWriterÚrejoin_linesÚsplit_linesÚstrip_transientc                   @  s   e Zd ZdZdd„ ZdS )ÚBytesEncoderz@A JSON encoder that accepts b64 (and other *ascii*) bytestrings.c                 C  s"   t |tƒr| d¡S tj | |¡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/v3/nbjson.pyr      s    

zBytesEncoder.defaultN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r
      s   r
   c                   @  s    e Zd ZdZdd„ Zdd„ ZdS )Ú
JSONReaderzA JSON notebook reader.c                 K  s(   t j|f|Ž}| j|f|Ž}t|ƒ}|S )zConvert a string to a notebook.)r   ÚloadsÚto_notebookr	   )r   ÚsÚkwargsÚnbr   r   r   Úreads   s    zJSONReader.readsc                 K  s   t t|ƒƒS )zConvert a dict to a notebook.)r   r   )r   Údr   r   r   r   r   "   s    zJSONReader.to_notebookN)r   r   r   r   r    r   r   r   r   r   r      s   r   c                   @  s   e Zd ZdZdd„ ZdS )Ú
JSONWriterzA JSON notebook writer.c                 K  sT   t |d< d|d< d|d< d|d< t |¡}t|ƒ}| dd¡rFt|ƒ}tj|f|ŽS )	zConvert a notebook to a string.Úclsr   ÚindentTÚ	sort_keys)ú,z: Ú
separatorsr   )r
   ÚcopyÚdeepcopyr	   Úpopr   r   Údumps)r   r   r   r   r   r   Úwrites*   s    
zJSONWriter.writesN)r   r   r   r   r,   r   r   r   r   r"   '   s   r"   )r   Ú
__future__r   r(   r   Znbbaser   Zrwbaser   r   r   r   r	   r   r
   r   r"   Z_readerZ_writerr    Úreadr   Úwriter,   r   r   r   r   Ú<module>   s   
