o i@s4dZddlmZmZddlmZGdddeZdS)zFModule containing a preprocessor that removes metadata from code cells)BoolSet) PreprocessorcseZdZdZedddjddZedddjddZedgddjddZ eddjddZ d d Z d d Z d dZ ddZddZfddZZS)ClearMetadataPreprocessorzE Removes all the metadata from all code cells in a notebook. TzRFlag to choose if cell metadata is to be cleared in addition to notebook metadata.)help)configzRFlag to choose if notebook metadata is to be cleared in addition to cell metadata.) language_infonamezIndicates the key paths to preserve when deleting metadata across both cells and notebook metadata fields. Tuples of keys can be passed to preserved specific nested valuescCs&t|tr|St|dkrdS|dS)z#Get the current key for a mask key.rN) isinstancestrlen)selfmask_keyr`/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/nbconvert/preprocessors/clearmetadata.py current_key(s  z%ClearMetadataPreprocessor.current_keycfdd|DS)z Get the current mask for a mask.cs$h|]}|dur|qS)N)r.0krrr 3s$z9ClearMetadataPreprocessor.current_mask..rrmaskrrr current_mask1sz&ClearMetadataPreprocessor.current_maskcr)z Get the nested masks for a mask.cs>i|]}|rt|tst|dkr|d|ddqS)rrN)r r r rrrrr 7s z:ClearMetadataPreprocessor.nested_masks..rrrrr nested_masks5s z&ClearMetadataPreprocessor.nested_masksccsr||}||}|D])\}}||}||vr6|dur1t|tr0|t|||fVq ||fVq dS)z-Get the nested filter for items given a mask.N)rrgetr dict nested_filteritems)rr!r keep_currentkeep_nested_lookuprv keep_nestedrrrr =s      z'ClearMetadataPreprocessor.nested_filtercCs:|jr|jdkrd|vrt||j|j|_||fS)zO All the code cells are returned with an empty metadata field. codemetadata)clear_cell_metadata cell_typerr r'r!preserve_cell_metadata_mask)rcell resources cell_indexrrrpreprocess_cellJs z)ClearMetadataPreprocessor.preprocess_cellcsBt||\}}|jrd|vrt||j|j|_||fS)ax Preprocessing to apply on each notebook. Must return modified nb, resources. Parameters ---------- nb : NotebookNode Notebook being converted resources : dictionary Additional resources used in the conversion process. Allows preprocessors to pass variables into the Jinja engine. r')super preprocessclear_notebook_metadatarr r'r!preserve_nb_metadata_mask)rnbr, __class__rrr0Vs z$ClearMetadataPreprocessor.preprocess)__name__ __module__ __qualname____doc__rtagr(r1rr2r*rrrr r.r0 __classcell__rrr4rr s>   rN)r9 traitletsrrbaserrrrrrs