U ku f@s0dZddlmZddlmZddlmZddlmZm Z m Z m Z ddl m Z mZddlmZd d d d d ddddddddddddgZee e e dZddlmZddlmZddlmZmZddl mZddl mZdd lmZmZGd!dde Z!ede"d"Z#d(d$dZ$e#d#fd%dZ%d)d&dZ&e#d#fd'dZ'd#S)*znThe Jupyter notebook format Use this module to read or write notebook files as particular nbformat versions. ) annotations)Path) get_logger)v1v2v3v4) __version__ version_info)SentinelversionsvalidateValidationErrorconvert from_dict NotebookNodecurrent_nbformatcurrent_nbformat_minor NBFormatError NO_CONVERTreadsreadwriteswriter r r )r)reader)r)rr)nbformat)nbformat_minor)rrc@s eZdZdS)rN)__name__ __module__ __qualname__r$r$5/tmp/pip-unpacked-wheel-zj9qk9v2/nbformat/__init__.pyr5szKValue to prevent nbformat to convert notebooks to most recent version. Nc Ksttj|f|}|tk r t||}z t|WnBtk rn}z$td|t|t r^||d<W5d}~XYnX|S)a9Read a notebook from a string and return the NotebookNode object as the given version. The string can contain a notebook of any version. The notebook will be returned `as_version`, converting, if necessary. Notebook format errors will be logged. Parameters ---------- s : unicode The raw unicode string to read the notebook from. as_version : int The version of the notebook format to return. The notebook will be converted, if necessary. Pass nbformat.NO_CONVERT to prevent conversion. capture_validation_error : dict, optional If provided, a key of "ValidationError" with a value of the ValidationError instance will be added to the dictionary. Returns ------- nb : NotebookNode The notebook that was read. Notebook JSON is invalid: %srN) rrrrrrrerror isinstancedict)s as_versioncapture_validation_errorkwargsnber$r$r%rBs   c Ks|tk rt||}nt|\}}z t|WnBtk rp}z$td|t|t r`||d<W5d}~XYnXt |j |f|S)aWrite a notebook to a string in a given format in the given nbformat version. Any notebook format errors will be logged. Parameters ---------- nb : NotebookNode The notebook to write. version : int, optional The nbformat version to write. If unspecified, or specified as nbformat.NO_CONVERT, the notebook's own version will be used and no conversion performed. capture_validation_error : dict, optional If provided, a key of "ValidationError" with a value of the ValidationError instance will be added to the dictionary. Returns ------- s : unicode The notebook as a JSON string. r&rN) rrr get_versionrrrr'r(r)r Z writes_json)r.versionr,r-_r/r$r$r%rhs   c Ksjz |}WnLtk rXt|dd&}t|||f|W5QRYSQRXYnXt|||f|S)arRead a notebook from a file as a NotebookNode of the given version. The string can contain a notebook of any version. The notebook will be returned `as_version`, converting, if necessary. Notebook format errors will be logged. Parameters ---------- fp : file or str A file-like object with a read method that returns unicode (use ``io.open()`` in Python 2), or a path to a file. as_version : int The version of the notebook format to return. The notebook will be converted, if necessary. Pass nbformat.NO_CONVERT to prevent conversion. capture_validation_error : dict, optional If provided, a key of "ValidationError" with a value of the ValidationError instance will be added to the dictionary. Returns ------- nb : NotebookNode The notebook that was read. utf8encoding)rAttributeErroropenr)fpr+r,r-buffr$r$r%rs  0c Kst|||f|}t|tr$|d}z"|||dsD|dWnPtk rt|jddd$}|||ds|dW5QRXYnXdS)aWrite a notebook to a file in a given nbformat version. The file-like object must accept unicode input. Parameters ---------- nb : NotebookNode The notebook to write. fp : file or str Any file-like object with a write method that accepts unicode, or a path to write a file. version : int, optional The nbformat version to write. If nb is not this version, it will be converted. If unspecified, or specified as nbformat.NO_CONVERT, the notebook's own version will be used and no conversion performed. capture_validation_error : dict, optional If provided, a key of "ValidationError" with a value of the ValidationError instance will be added to the dictionary. r3 wr4N) rr(bytesdecoderendswithr6rr7)r.r8r1r,r-r*r:r$r$r%rs      )N)N)(__doc__ __future__rpathlibrZ traitlets.logrrrrr _versionr r sentinelr __all__r r converterrZ notebooknoderrrrr rZ validatorrr ValueErrorrr!rrrrrr$r$r$r%sZ         &$ %