U nu f@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.)Z 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)selfZmask_keyrI/tmp/pip-unpacked-wheel-nxpdnluh/nbconvert/preprocessors/clearmetadata.py current_key(s   z%ClearMetadataPreprocessor.current_keycsfdd|DS)z Get the current mask for a mask.cs$h|]}|dk r|qS)N)r.0kr rr 3sz9ClearMetadataPreprocessor.current_mask..rr maskrrr current_mask1sz&ClearMetadataPreprocessor.current_maskcsfdd|DS)z Get the nested masks for a mask.cs>i|]6}|rt|tst|dkr|d|ddqS)rrN)r r r rrrrr 7s  z:ClearMetadataPreprocessor.nested_masks..rrrrr nested_masks5s z&ClearMetadataPreprocessor.nested_masksccsp||}||}|D]R\}}||}||kr|dk r`t|trj|t|||fVq||fVqdS)z-Get the nested filter for items given a mask.N)rrgetr dict nested_filteritems)r rrZ keep_currentZkeep_nested_lookuprvZ keep_nestedrrrr=s     z'ClearMetadataPreprocessor.nested_filtercCs:|jr2|jdkr2d|kr2t||j|j|_||fS)zO All the code cells are returned with an empty metadata field. codemetadata)clear_cell_metadataZ cell_typerrr!rpreserve_cell_metadata_mask)r cell resourcesZ cell_indexrrrpreprocess_cellJs z)ClearMetadataPreprocessor.preprocess_cellcsBt||\}}|jr:d|kr:t||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_metadatarrr!rpreserve_nb_metadata_mask)r nbr% __class__rrr(Vs z$ClearMetadataPreprocessor.preprocess)__name__ __module__ __qualname____doc__rtagr"r)rr*r#rrrrr&r( __classcell__rrr,rr s<   rN)r1Z traitletsrrbaserrrrrrs