o b3@sdZddlmZddlZddlZddlmZmZmZm Z ddl Z ddl Z ddl Z e j dkr4ddlmZnddlmZddlmZddlmZed Zed Zed Zejd d GdddZedededefddZdedeeeffddZdededefddZdedeeeeffddZdededefddZ dedeeeeffd d!Z!dedeeeeffd"d#Z"ded$eedefd%d&Z#d'ej$deej%fd(d)Z&ejd d Gd*d+d+Z'ejGd,d-d-ej(Z)ejd d Gd.d/d/Z*ejGd0d1d1ej(Z+dS)2z)Functions to process IPython magics with.) lru_cacheN)DictListTupleOptional) ) TypeGuard)NothingChanged)out)zget_ipython().run_cell_magiczget_ipython().systemzget_ipython().getoutputzget_ipython().run_line_magic) ENDMARKERNLNEWLINECOMMENTDEDENTZUNIMPORTANT_WSZ ESCAPED_NL) z%%bashz%%htmlz %%javascriptz%%jsz%%latexz %%markdownz%%perlz%%rubyz%%scriptz%%shz%%svgz %%writefileT)frozenc@eZdZUeed<eed<dS) ReplacementmasksrcN__name__ __module__ __qualname__str__annotations__rrX/var/www/secure340b-portal/env/lib/python3.10/site-packages/black/handle_ipynb_magics.pyr:  rverbosequietreturncCs@z ddl}ddl}WdSty|s|sd}t|YdSw)Nrz{Skipping .ipynb files as Jupyter dependencies are not installed. You can fix this by running ``pip install black[jupyter]``FT)IPython tokenize_rtModuleNotFoundErrorr )rr r"r#msgrrr"jupyter_dependencies_are_installed@s  r&rcCstddlm}m}m}||}d}||D]\}}|jtvrq|jdkr-|jdkr-||=d}|s4|dfS||dfS)aRemove trailing semicolon from Jupyter notebook cell. For example, fig, ax = plt.subplots() ax.plot(x_data, y_data); # plot data would become fig, ax = plt.subplots() ax.plot(x_data, y_data) # plot data Mirrors the logic in `quiet` from `IPython.core.displayhook`, but uses ``tokenize_rt`` so that round-tripping works fine. r src_to_tokens tokens_to_srcreversed_enumerateFOP;T)r#r(r)r*nameTOKENS_TO_IGNOREr)rr(r)r*tokensZtrailing_semicolonidxtokenrrrremove_trailing_semicolonQs  r2has_trailing_semicoloncCsp|s|Sddlm}m}m}||}||D]\}}|jtvr q|j|jdd||<ntddt ||S)zPut trailing semicolon back if cell originally had it. Mirrors the logic in `quiet` from `IPython.core.displayhook`, but uses ``tokenize_rt`` so that round-tripping works fine. rr'r,)rz{INTERNAL ERROR: Was not able to reinstate trailing semicolon. Please report a bug on https://github.com/psf/black/issues. N) r#r(r)r*r-r._replacerAssertionErrorr)rr3r(r)r*r/r0r1rrrput_trailing_semicolon_backus  r6cCsg}z t|W||fStyYnwddlm}|}||}t|\}}||7}||}t|\}}t| t| krGt ||7}||fS)zMask IPython magics so content becomes parseable Python code. For example, %matplotlib inline 'foo' becomes "25716f358c32750e" 'foo' The replacements are returned, along with the transformed code. r)TransformerManager) astparse SyntaxErrorZIPython.core.inputtransformer2r7Ztransform_cellreplace_cell_magicsreplace_magicslen splitlinesr )r replacementsr7Ztransformer_managerZ transformedZcell_magic_replacementsZmagic_replacementsrrr mask_cells$       r@magiccCs|sJtt|ddd}t|}d}||vr3t|}|d7}|dkr/td|d||vst|dt|krB|d}d|dS) aDReturn randomly generated token to mask IPython magic with. For example, if 'magic' was `%matplotlib inline`, then a possible token to mask it with would be `"43fdd17f7e5ddc83"`. The token will be the same length as the magic, and we make sure that it was not already present anywhere else in the cell. rdzINTERNAL ERROR: Black was not able to replace IPython magic. Please report a bug on https://github.com/psf/black/issues. The magic might be helpful: N.")maxr=secretsZ token_hexr5)rrAnbytesr1counterrrr get_tokens&    rKcCsg}t|}t}|||jdur||fS|jjdtvr$tt ||jj}| t ||jjd|d|jj |fS)aNReplace cell magic with token. Note that 'src' will already have been processed by IPython's TransformerManager().transform_cell. Example, get_ipython().run_cell_magic('t', '-n1', 'ls =!ls\n') becomes "a794." ls =!ls The replacement, along with the transformed code, is returned. Nrrr ) r8r9CellMagicFindervisit cell_magicheadersplitNON_PYTHON_CELL_MAGICSr rKappendrbody)rr?treeZcell_magic_finderrrrrr;s   r;c Csg}t}|t|g}t|ddD]B\}}||jvrT|j|}t|dkr3td|d|dj |dj }}t ||} | t | |d|d|| }| |qd||fS) aoReplace magics within body of cell. Note that 'src' will already have been processed by IPython's TransformerManager().transform_cell. Example, this get_ipython().run_line_magic('matplotlib', 'inline') 'foo' becomes "5e67db56d490fd39" 'foo' The replacement, along with the transformed code, are returned. rC)startz#Expecting one magic per line, got: z< Please report a bug on https://github.com/psf/black/issues.rrLNrM) MagicFinderrOr8r9 enumerater>magicsr=r5 col_offsetrArKrTrjoin) rr?Z magic_finderZnew_srcsilineZoffsets_and_magicsr[rArrrrr<s&      r<r?cCs|D] }||j|j}q|S)zRemove replacements from cell. For example "9b20" foo = bar becomes %%time foo = bar )replacerr)rr? replacementrrr unmask_cells ranodecCs8t|tjot|jtjot|jjtjo|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. get_ipython) isinstancer8 AttributevalueCallfuncNameid)rbrrr_is_ipython_magic*s   rkc@r) CellMagicrQrUNrrrrrrl9rrlc@s6eZdZUdZdZeeed<dej ddfddZ dS)rNa^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. NrPrbr!cCst|jtjrHt|jjrH|jjjdkrHg}|jjD]}t|tjs$J| |j qd|d}|dr?|d|d7}t ||dd|_ | |dS) z)Find cell magic, extract header and body.Zrun_cell_magicz%%rrC rB)rQrUN)rdrfr8rgrkrhattrargsStrrTsrlrP generic_visit)selfrbroargrQrrr visit_ExprTs    zCellMagicFinder.visit_Expr) rrr__doc__rPrrlrr8ExprrurrrrrN?s rNc@seZdZUeed<eed<dS)OffsetAndMagicr[rAN)rrrintrrrrrrrxfrrxc@s^eZdZUdZejdddZeee e fe d<de j ddfd d Zde jddfd d ZdS) 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). cCs ttS)N) collections defaultdictlistrrrrs zMagicFinder.)default_factoryrZrbr!NcCst|jtjrEt|jjrE|jjjdkrEg}|jjD]}t|tjs$J| |j q|s/Jd|d}|j |jj  t |jj|||dS)a Look for system assign magics. For example, black_version = !black --version would have been transformed to black_version = get_ipython().getoutput('black --version') and we look for instances of the latter. getoutput!rN)rdrfr8rgrkrhrnrorprTrqrZlinenorxr[rrrsrbrortrrrr visit_Assigns     zMagicFinder.visit_AssigncCs8t|jtjrt|jjrg}|jjD]}t|tjsJ||j q|s(J|jjj dkrf|ddkr=d|d}nI|ddkrKd|d}n;d|d}|dre|d us\J|d |d7}n |jjj d krud |d}n|jjj d krd|d}nt |j |jj t|jj|||d S)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. Zrun_line_magicrZpinfo?rCZpinfo2z??%Nrmsystemrrz!!)rdrfr8rgrkrhrorprTrqrnr rZrrxr[rrrrrrrus2     zMagicFinder.visit_Expr)rrrrv dataclassesfieldrZrryrrxrr8AssignrrwrurrrrrXls rX),rv functoolsrrr8typingrrrrrHsysrz version_infor Ztyping_extensionsZ black.reportr Z black.outputr frozensetZTRANSFORMED_MAGICSr.rS dataclassrboolr&rr2r6r@rKr;r<raexprrerkrl NodeVisitorrNrxrXrrrrsT       $( )  &