o ^i@sdZddlmZddlmZddlmZddlmZm Z m Z m Z ddl m Z mZddlmZgd Zee e e d Zdd lmZdd lmZdd lmZmZddl mZddl mZddlmZmZGddde Z!ede"dZ#dddZ$e#dfddZ%dddZ&e#dfddZ'dS)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)Sentinel)versionsvalidateValidationErrorconvert 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$L/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/nbformat/__init__.pyr5srrzKValue to prevent nbformat to convert notebooks to most recent version. Nc Kstj|fi|}|turt||}zt|W|StyC}ztd|t|t r8||d<WYd}~|SWYd}~|Sd}~ww)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    rc Ks|tur t||}nt|\}}zt|Wn"ty9}ztd|t|t r/||d<WYd}~nd}~wwt |j |fi|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 writes_json)r.versionr,r-_r/r$r$r%rhs   rc Ks|z|}Wn-ty3t|dd}t|||fi|WdYS1s,wYYnwt|||fi|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. utf8encodingN)rAttributeErroropenr)fpr+r,r-buffr$r$r%rs  (rc Kst|||fi|}t|tr|d}z|||ds'|dWdSWdStyft|jddd"}|||dsR|dWdYdSWdYdS1s^wYYdSw)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. r4 wr5N) rr(bytesdecoderendswithr7rr8)r.r9r2r,r-r*r;r$r$r%rs"        &r)N)(__doc__ __future__rpathlibr traitlets.logrrrrr _versionr r sentinelr __all__r r converterr notebooknoderrrrr r validatorrr ValueErrorrr!rrrrrr$r$r$r%s:         & $%