B '=[! @sdZddlZddlZddlmZmZmZddlmZddlm Z ddZ d'd d Z d d Z ddZ dddddddddZddZddZddZdd Zd!d"Zd#d$Zd%d&ZdS)(z-Code for converting notebooks to and from v3.N)nbformatnbformat_minor NotebookNode)v3) get_loggerc CsXddlm}m}y|||dWn2|k rR}ztd||Wdd}~XYnXdS)z(Log validation errors, if there are any.r)validateValidationError)versionz"Notebook JSON is not valid v%i: %sN)rrr rerror)nbr rr er2lib/python3.7/site-packages/nbformat/v4/convert.py_warn_if_invalids rcCs|dkrt|||dd}|dd}|p0d|j_|p= 3). r orig_nbformatNorig_nbformat_minorrcells worksheetsnameZ signaturezZCannot convert a notebook directly from v%s to v4. Try using the nbformat.convert module.) rpopmetadatarrrrappend upgrade_cell ValueError)r Z from_versionZ from_minorrrrZwscellrrrupgrades.        rcCs|dt|jdkrf|ddd|kr<|d|jd<|dd|_|dd|_t|j|_nT|jd krd |_|d d }d j d|d | dd d|_n|jdkrd |_|S)zupgrade a cell from v3 to v4 heading cell: - -> markdown heading code cell: - remove language metadata - cell.input -> cell.source - cell.prompt_number -> cell.execution_count - update outputs rcodelanguager collapsedinput prompt_numberNheadingmarkdownlevelrz{hashes} {single_line}# source)ZhashesZ single_linehtml) setdefaultr cell_typerrr*execution_countupgrade_outputsoutputsformatjoinget splitlines)rr'rrrrLs"      rcCs|jdkrJd|_|dd|_|dd|_|jdd|_t|j|_nR|jd kr| dd}d |kr| d rt d | \}}d |_||_t||_|dd|S)adowngrade a cell from v4 to v3 code cell: - set cell.language - cell.input <- cell.source - cell.prompt_number <- cell.execution_count - update outputs markdown cell: - single-line heading -> heading cell r Zpythonr*rr.Nr"Fr& r(z (#+)\s*(.*)r%Z attachments)r-r!rr#r$rr"downgrade_outputsr0r3 startswithrematchgroupsr*lenr')rr*prefixtextrrrdowngrade_cellks     r>z text/plainz text/htmlz image/svg+xmlz image/pngz image/jpegz text/latexzapplication/jsonzapplication/javascript)r=r+ZsvgZpngZjpegZlatexjsonZ javascriptcCs0x*tD]\}}||kr ||||<q W|S)z4convert dict with v3 aliases to plain mime-type keys) _mime_mapitemsr)daliasmimerrr to_mime_keysrEcCs2i}x(tD]\}}||kr||||<qW|S)z.convert dict with mime-type keys to v3 aliases)r@rA)rBZd2rCrDrrr from_mime_keys rFcCs|ddkr|dt|ddkr>d|d<|dd|d<i}x(t|D]}|d krZqL||||<qLWt|||d <t|jd |krt|d |d <xjd D],}||krt||t r|| d ||<qWn4|ddkrd|d<n|ddkr |dd|d<|S)zupgrade a single code cell output from v3 to v4 - pyout -> execute_result - pyerr -> error - output.type -> output.data.mime/type - mime-type keys - stream.stream -> stream.name output_type>pyout display_datarrHexecute_resultr$Nr.>rrGr.datazapplication/json)z image/pngz image/jpegasciipyerrr streamstdoutr) r,rrlistrErr?loads isinstancebytesdecode)outputrKkeyrrrupgrade_outputs.      rWcCs|ddkrz|ddkr0d|d<|dd|d<|di}d |krVt|d |d <t|}||t|d in0|dd krd |d<n|dd kr|d|d <|S)zdowngrade a single code cell output to v3 from v4 - pyout <- execute_result - pyerr <- error - output.data.mime/type -> output.type - un-mime-type keys - stream.stream <- stream.name rG>rIrJrJrHr.Nr$rKzapplication/jsonrr rMrNr)rr?dumpsrFupdater3)rUrKrrrdowngrade_outputs       rZcCsdd|DS)z,upgrade outputs of a code cell from v3 to v4cSsg|] }t|qSr)rW).0oprrr sz#upgrade_outputs..r)r0rrrr/sr/cCsdd|DS)z.downgrade outputs of a code cell to v3 from v4cSsg|] }t|qSr)rZ)r[r\rrrr]sz%downgrade_outputs..r)r0rrrr6sr6cCs|jtkr|St|ttj|_tj|_dd|dD}tj|dg|_|jddt|tj|jdt|_ |jdt|_ |S) zConvert a v4 notebook to v3. Parameters ---------- nb : NotebookNode The Python representation of the notebook to convert. cSsg|] }t|qSr)r>)r[rrrrr]szdowngrade..r)rrrrr) rrrrrZ new_worksheetrrr,rr)r rrrr downgrades   r^)rr)__doc__r?r8ZnbbaserrrrZ traitlets.logrrrrr>r@rErFrWrZr/r6r^rrrrs0   3$