a g g4@sdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ej dkrhddlmZn ddlmZddlmZddlmZed Zed Zed ZejZejd d GdddZeeeedddZee eefdddZeeedddZ ee ee efdddZ!eeedddZ"ee ee efdddZ#ee ee efdd d!Z$ee eed"d#d$Z%ej&eej'd%d&d'Z(e ej&e ed(d)d*Z)ejd d Gd+d,d,Z*Gd-d.d.ej+Z,ejd d Gd/d0d0Z-Gd1d2d2ej+Z.dS)3z)Functions to process IPython magics with.N) lru_cache)DictListOptionalTuple) ) TypeGuard)out)NothingChanged)zget_ipython().run_cell_magiczget_ipython().systemzget_ipython().getoutputzget_ipython().run_line_magic) ENDMARKERNLNEWLINECOMMENTDEDENTZUNIMPORTANT_WSZ ESCAPED_NL)captureZprunpypypythonpython3timeZtimeitT)frozenc@seZdZUeed<eed<dS) ReplacementmasksrcN)__name__ __module__ __qualname__str__annotations__rr\/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/black/handle_ipynb_magics.pyr4s r)verbosequietreturncCsFzddl}ddl}Wn(ty<|s*|s6d}t|YdS0dSdS)Nrz}Skipping .ipynb files as Jupyter dependencies are not installed. You can fix this by running ``pip install "black[jupyter]"``FT)IPython tokenize_rtModuleNotFoundErrorr )r!r"r$r%msgrrr "jupyter_dependencies_are_installed:s  r()rr#cCsxddlm}m}m}||}d}||D]6\}}|jtvrs&      r>)rrAr#cCs|D]}||j|j}q|S)zRemove replacements from cell. For example "9b20" foo = bar becomes %%time foo = bar )replacerr)rrA replacementrrr unmask_cell s ranoder#cCs8t|tjo6t|jtjo6t|jjtjo6|jjjdkS)zCheck if attribute is IPython magic. Note that the source of the abstract syntax tree will already have been processed by IPython's TransformerManager().transform_cell. Z get_ipython) isinstancer: AttributevalueCallfuncNameid)rcrrr _is_ipython_magics   rk)argsr#cCs.g}|D] }t|tjsJ||jq|SN)rdr:StrrTs)rlZstr_argsargrrr _get_str_args.s rqc@s<eZdZUeed<eeed<eed<eedddZdS) CellMagicr/paramsrUr#cCs&|jrd|jd|jSd|jS)Nz%% )rsr/selfrrr rS<szCellMagic.headerN)rrrrrrpropertyrSrrrr rr6s  rrc@s8eZdZdZd eeddddZejddddZ dS) rPa^Find cell magics. Note that the source of the abstract syntax tree will already have been processed by IPython's TransformerManager().transform_cell. For example, %%time foo() would have been transformed to get_ipython().run_cell_magic('time', '', 'foo()\n') and we look for instances of the latter. N)rRr#cCs ||_dSrm)rR)rwrRrrr __init__VszCellMagicFinder.__init__rbcCs^t|jtjrPt|jjrP|jjjdkrPt|jj}t |d|d|dd|_ | |dS)z)Find cell magic, extract header and body.Zrun_cell_magicrrErD)r/rsrUN) rdrfr:rgrkrhattrrqrlrrrR generic_visit)rwrcrlrrr visit_ExprYs    zCellMagicFinder.visit_Expr)N) rrr__doc__rrrryr:Exprr|rrrr rPDsrPc@seZdZUeed<eed<dS)OffsetAndMagicr[rCN)rrrintrrrrrr res rc@sBeZdZdZddddZejddddZejddd d Z dS) rXaVisit cell to look for get_ipython calls. Note that the source of the abstract syntax tree will already have been processed by IPython's TransformerManager().transform_cell. For example, %matplotlib inline would have been transformed to get_ipython().run_line_magic('matplotlib', 'inline') and we look for instances of the latter (and likewise for other types of magics). NrtcCstt|_dSrm) collections defaultdictlistrZrvrrr ryszMagicFinder.__init__rbcCst|jtjrt|jjrt|jj}|jjjdkrDd|d}nP|jjjdkr|d|d}|dr|d|d7}nt d|jjjd d |j |jj  t |jj|||d S) a{Look for system assign magics. For example, black_version = !black --version env = %env var would have been (respectively) transformed to black_version = get_ipython().getoutput('black --version') env = get_ipython().run_line_magic('env', 'var') and we look for instances of any of the latter. getoutput!rrun_line_magic%rEruzUnexpected IPython magic zC found. Please report a bug on https://github.com/psf/black/issues.N)rdrfr:rgrkrhrqrlrzr7rZlinenorTrr[r{rwrcrlrrrr visit_Assigns"  zMagicFinder.visit_AssigncCst|jtjrt|jjrt|jj}|jjjdkr|ddkrPd|d}q|ddkrld|d}qd|d}|dr|d |d7}n@|jjjd krd |d}n"|jjjd krd |d}nt |j |jj  t |jj|||dS)aLook for magics in body of cell. For examples, !ls !!ls ?ls ??ls would (respectively) get transformed to get_ipython().system('ls') get_ipython().getoutput('ls') get_ipython().run_line_magic('pinfo', 'ls') get_ipython().run_line_magic('pinfo2', 'ls') and we look for instances of any of the latter. rrZpinfo?rEZpinfo2z??rrusystemrrz!!N)rdrfr:rgrkrhrqrlrzr rZrrTrr[r{rrrr r|s&    zMagicFinder.visit_Expr) rrrr}ryr:Assignrr~r|rrrr rXms!rX)/r}r:rZ dataclassesZsecretssys functoolsrtypingrrrr version_infor Ztyping_extensionsZ black.outputr Z black.reportr frozensetZTRANSFORMED_MAGICSr0ZPYTHON_CELL_MAGICSZ token_hexrJZ dataclassrboolr(rr4r8rBrMr=r>raexprrerkrqrr NodeVisitorrPrrXrrrr sR         ()  !