o i@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 s"tjdi|d|_d|_dS)z$ Public constructor attachmentsN)super__init__ path_nameresources_item_key)selfkw __class__r e/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/nbconvert/preprocessors/extractattachments.pyr"sz'ExtractAttachmentsPreprocessor.__init__csx|jr|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 unique_key) notebook_nameattachment_files_dirr output_files_diroutputs) use_separate_dirattachments_directory_templateformatrr isinstancedictr preprocess)rnb resourcesrr rr /s   z)ExtractAttachmentsPreprocessor.preprocessc Csd|vrN|jD]F}|jd||j|D]}|j||d}t|}tj|j|}|||j |<tjj dkrC| 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) rcellr"indexfnamemimetypedatadecoded new_filenamer r rpreprocess_cellGs  z.ExtractAttachmentsPreprocessor.preprocess_cell) __name__ __module__ __qualname____doc__rtagrrrrr r4 __classcell__r r rrrs"  r) r8r'base64r traitletsrrbaserrr r r rs