U nu f@sHdZddlZddlmZddlmZmZddlmZGdddeZ dS) zF Module that extracts attachments from notebooks into their own files N) b64decode)BoolUnicode) Preprocessorcs\eZdZdZedddjddZedddjddZfd d Z fd d Z d dZ Z S)ExtractAttachmentsPreprocessorz Extracts attachments from all (markdown and raw) cells in a notebook. The extracted attachments are stored in a directory ('attachments' by default). https://nbformat.readthedocs.io/en/latest/format_description.html#cell-attachments z{notebook_name}_attachmentsz:Directory to place attachments if use_separate_dir is True)helpT)configFznWhether to use output_files_dir (which ExtractOutput also uses) or create a separate directory for attachmentsc stjf|d|_d|_dS)z$ Public constructor attachmentsN)super__init__ path_nameresources_item_key)selfkw __class__N/tmp/pip-unpacked-wheel-nxpdnluh/nbconvert/preprocessors/extractattachments.pyr "sz'ExtractAttachmentsPreprocessor.__init__csx|jr4|jj|dd|_|j|d<i|d<d|_n|d|_d|_t||jts^i||j<t||\}}||fS)zL Determine some settings and apply preprocessor to notebook Z unique_key)Z notebook_nameZattachment_files_dirr Zoutput_files_diroutputs) use_separate_dirattachments_directory_templateformatrr isinstancedictr preprocess)rnb resourcesrrrr/s   z)ExtractAttachmentsPreprocessor.preprocessc Csd|kr|jD]}|jd||j|D]$}|j||d}t|}qPq*tj|j|}|||j |<tjj dkr| tjj d}|j d|||_ q||fS)aB Extract attachments to individual files and change references to them. E.g. '![image.png](attachment:021fdd80.png)' becomes '![image.png]({path_name}/021fdd80.png)' Assumes self.path_name and self.resources_item_key is set properly (usually in preprocess). r zEncountered attachment %szutf-8/z attachment:) r logdebugencoderospathjoinrrsepreplacesource) rcellrindexfnamemimetypedatadecodedZ new_filenamerrrpreprocess_cellGs   z.ExtractAttachmentsPreprocessor.preprocess_cell) __name__ __module__ __qualname____doc__rtagrrrr rr/ __classcell__rrrrrs  r) r3r#base64rZ traitletsrrbaserrrrrrs