U ku f @sXdZddlmZddlZddlmZGdddeZdd Zd d Z d d Z ddZ dS)z/API for reading notebooks of different versions) annotationsN)ValidationErrorc@seZdZdZdS) NotJSONErrorz1An error raised when an object is not valid JSON.N)__name__ __module__ __qualname____doc__r r 3/tmp/pip-unpacked-wheel-zj9qk9v2/nbformat/reader.pyr src Ksjztj|f|}WnRtk rd}z4d|}t|dkrJ|ddd}t||W5d}~XYnX|S)z Parse a JSON string into a dict.z%Notebook does not appear to be JSON: PNMz...)jsonloads ValueErrorlenr)skwargsnb_dictemessager r r parse_jsons  rcCs |dd}|dd}||fS)zGet the version of a notebook. Parameters ---------- nb : dict NotebookNode or dict containing notebook data. Returns ------- Tuple containing major (int) and minor (int) version numbers ZnbformatrZnbformat_minorr)get)nbmajorminorr r r get_version s  rc Ksddlm}m}t|f|}t|\}}||kr~z||j||dWStk rz}zd|}t|dW5d}~XYqXn |d|dS)aQRead a notebook from a json string and return the NotebookNode object. This function properly reads notebooks of any version. No version conversion is performed. Parameters ---------- s : unicode | bytes The raw string or bytes object to read the notebook from. Returns ------- nb : NotebookNode The notebook that was read. Raises ------ ValidationError Notebook JSON for a given version is missing an expected key and cannot be read. NBFormatError Specified major version is invalid or unsupported. r) NBFormatErrorversions)rz8The notebook is invalid and is missing an expected key: NzUnsupported nbformat version %s)rrrrZto_notebook_jsonAttributeErrorr) rrrrrrrrmsgr r r reads1s   r"cKst|f|S)aZRead a notebook from a file and return the NotebookNode object. This function properly reads notebooks of any version. No version conversion is performed. Parameters ---------- fp : file Any file-like object with a read method. Returns ------- nb : NotebookNode The notebook that was read. )r"read)fprr r r r#Wsr#) r __future__rrZ validatorrrrrrr"r#r r r r s  &