a Fb @sddZddlZddlmZddlmZGdddeZeZ GdddeZ e Z Gd d d eZ dS) zF A context manager for managing things injected into :mod:`builtins`. N) Configurable)Instancec@s eZdZdS)__BuiltinUndefinedN__name__ __module__ __qualname__r r 8lib/python3.9/site-packages/IPython/core/builtin_trap.pyr rc@s eZdZdS) __HideBuiltinNrr r r r r r r csZeZdZedddZdfdd ZddZd d Zd d Zd dZ ddZ ddZ Z S) BuiltinTrapz1IPython.core.interactiveshell.InteractiveShellABCT)Z allow_noneNcs<tt|j|ddi|_d|_||_tt|jjd|_dS)N)shellZconfigr)exitquit get_ipython) superr __init___orig_builtins _nested_levelr HideBuiltinr auto_builtins)selfr __class__r r rszBuiltinTrap.__init__cCs$|jdkr||jd7_|S)Nr)ractivate)rr r r __enter__'s zBuiltinTrap.__enter__cCs$|jdkr||jd8_dS)NrF)r deactivate)rtypevalue tracebackr r r __exit__.s zBuiltinTrap.__exit__cCsJtj}||t}|tur4|turF||j|<||=n||j|<|||<dS)z$Add a builtin and save the original.N) builtin_mod__dict__getBuiltinUndefinedrr)rkeyr Zbdictorigr r r add_builtin5s   zBuiltinTrap.add_builtincCs |turtj|=n |tj|<dS)z0Remove an added builtin and re-set the original.N)r&r#r$)rr'r(r r r remove_builtinAs zBuiltinTrap.remove_builtincCs(|j}|jD]\}}|||qdS)z6Store ipython references in the __builtin__ namespace.N)r)ritems)rr)namefuncr r r rHszBuiltinTrap.activatecCs8|j}|jD]\}}|||q|jd|_dS)z~Remove any builtins which might have been added by add_builtins, or restore overwritten ones to their previous values.FN)r*rr+clearZ_builtins_added)rr*r'valr r r rOs   zBuiltinTrap.deactivate)N) rrrrrrrr"r)r*rr __classcell__r r rr r s r ) __doc__builtinsr#Ztraitlets.config.configurablerZ traitletsrobjectrr&r rr r r r r s