yIf44dZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZejdkrddl mZnddlmZddlmZdd lmZed Zed Zed ZejZejd GddZededefdZde dee effdZ!de dede fdZ"de dee e effdZ#de de de fdZ$de dee e effdZ%de dee e effdZ&de de ede fdZ'dej(deej)fd Z*d!e ej(de e fd"Z+ejd Gd#d$Z,Gd%d&ej-Z.ejd Gd'd(Z/Gd)d*ej-Z0dS)+z)Functions to process IPython magics with.N) lru_cache) find_spec)DictListOptionalTuple) ) TypeGuard)out)NothingChanged)zget_ipython().run_cell_magiczget_ipython().systemzget_ipython().getoutputzget_ipython().run_line_magic) ENDMARKERNLNEWLINECOMMENTDEDENTUNIMPORTANT_WS ESCAPED_NL)captureprunpypypythonpython3timetimeitT)frozenc$eZdZUeed<eed<dS) ReplacementmasksrcN)__name__ __module__ __qualname__str__annotations__M/var/www/piapp/venv/lib/python3.11/site-packages/black/handle_ipynb_magics.pyrr/s" III HHHHHr'rwarnreturncttdduotddu}|s|rd}t||S)N tokenize_rtIPythonz}Skipping .ipynb files as Jupyter dependencies are not installed. You can fix this by running ``pip install "black[jupyter]"``)rr )r) installedmsgs r("jupyter_dependencies_are_installedr05sZ -  ,Q91E1ET1Q  K  C r'r cddlm}m}m}||}d}||D]/\}}|jt vr|jdkr|jdkr||=d}|s|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. rreversed_enumerate src_to_tokens tokens_to_srcFOP;T)r,r3r4r5nameTOKENS_TO_IGNOREr )r r3r4r5tokenstrailing_semicolonidxtokens r(remove_trailing_semicolonr>Cs MLLLLLLLLL ]3  F((00 U :) ) )  :  %)s"2"2s !%  Ez = $ &&r'has_trailing_semicolonc|s|Sddlm}m}m}||}||D]6\}}|jt vr||jdz||<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. rr2r7)r z{INTERNAL ERROR: Was not able to reinstate trailing semicolon. Please report a bug on https://github.com/psf/black/issues. N) r,r3r4r5r8r9_replacer AssertionErrorr$)r r?r3r4r5r:r<r=s r(put_trailing_semicolon_backrCcs " LLLLLLLLLL ]3  F((00   U :) ) ) nnSn99s   L    }}V$$ % %%r'cg} tj|||fS#t$rYnwxYwddlm}|}||}t |\}}||z }||}t|\}}t| t| krt||z }||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 SyntaxErrorIPython.core.inputtransformer2rEtransform_cellreplace_cell_magicsreplace_magicslen splitlinesr )r replacementsrEtransformer_manager transformedcell_magic_replacementsmagic_replacementss r( mask_cellrT{s')L! # L       BAAAAA,,..%44S99K+>{+K+K(K(++L%44[AAK&4[&A&A#K# ; ! ! # #$$CNN,<,<(=(===&&L  $$s  ))magicc8|sJtt|dzdz d}t|}d}||vr1t|}|dz }|dkrtd|d||v1t|dzt|kr|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.")maxrM TOKEN_HEXrB)r rUnbytesr=counters r( get_tokenr`s LLL Uq1$a ( (F f  EG 3,,&!!1  S== 7/477   3,, 5zzA~E ""  u<<<r'c0g}tj|}t}|||j||fS|jj}t ||}|t|||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. Nrr  ) rFrGCellMagicFindervisit cell_magicheaderr`appendrbody)r rOtreecell_magic_finderrgrs r(rKrKs"')L 9S>>D'))D!!!#+L   ) 0F S& ! !D 6:::;;; 9 9'27 9 9< GGr'cLg}t}|tj|g}t |dD]\}}||jvr|j|}t|dkrtd|d|dj |dj }}t||} | t| ||d|| z}| |d||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. rX)startz#Expecting one magic per line, got: z< Please report a bug on https://github.com/psf/black/issues.rrbNrc) MagicFinderrerFrG enumeraterNmagicsrMrB col_offsetrUr`rhrjoin) r rO magic_findernew_srcsilineoffsets_and_magicsrqrUrs r(rLrLsJ$L==Lsy~~&&&HS^^--Q7774 # # #!-!4Q!7 %&&!++$R:LRRR #1%0"1%+JS%((D    5 A A A B B B  $t+D 99X   ,,r'rOcP|D]"}||j|j}#|S)zRemove replacements from cell. For example "9b20" foo = bar becomes %%time foo = bar )replacerr )r rO replacements r( unmask_cellr{s2$== kk+*KO<< Jr'nodect|tjoWt|jtjo8t|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) isinstancerF AttributevalueCallfuncNameid)r|s r(_is_ipython_magicrs\ 4'' 0 tz38 , , 0 tz 1 1 0 JO - / r'argscg}|D]8}t|tjsJ||j9|SN)rrFStrrhs)rstr_argsargs r( _get_str_argsr&sJH#sw''''' Or'cVeZdZUeed<eeed<eed<edefdZdS) CellMagicr8paramsrir*cH|jrd|jd|jSd|jS)Nz%% )rr8selfs r(rgzCellMagic.header4s6 ; 21 11DK11 1DIr'N)r!r"r#r$r%rpropertyrgr&r'r(rr.s] III SM III     X   r'rcJeZdZdZddeeddfdZdejddfdZ dS) rdagFind 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. Nrfr*c||_dSr)rf)rrfs r(__init__zCellMagicFinder.__init__Os $r'r|cNt|jtjrpt |jjrW|jjjdkrBt|jj}t|d|d|d|_ | |dS)z)Find cell magic, extract header and body.run_cell_magicrrXrW)r8rriN) rrrFrrrattrrrrrf generic_visit)rr|rs r( visit_ExprzCellMagicFinder.visit_ExprRs tz38 , , T!$*/22 T $(888 11D'T!WT!W4PQ7SSSDO 4     r'r) r!r"r#__doc__rrrrFExprrr&r'r(rdrd<sn$%%8I#6%$%%%% !sx !D ! ! ! ! ! !r'rdc$eZdZUeed<eed<dS)OffsetAndMagicrqrUN)r!r"r#intr%r$r&r'r(rr^s"OOO JJJJJr'rcNeZdZdZddZdejddfdZdejddfdZ dS) rnaVisit 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). r*NcBtjt|_dSr) collections defaultdictlistrprs r(rzMagicFinder.__init__ys7B7Nt7T7T r'r|cLt|jtjrt |jjrt |jj}|jjjdkr d|d}nZ|jjjdkr"d|d}|dr|d|dz }n#td|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%rXrzUnexpected IPython magic zC found. Please report a bug on https://github.com/psf/black/issues.N)rrrFrrrrrrrBrplinenorhrrqrrr|rr s r( visit_AssignzMagicFinder.visit_Assign|s1 dj#( + + 0A$*/0R0R  11Dz#{22#$q'mm%)999#$q'mm7)=tAw==(C$R 0DRRR K ) * 1 1tz4c::    4     r'ct|jtjr%t |jjr t |jj}|jjjdkrR|ddkr d|d}n|ddkr d|d}nkd|d}|dr|d |dz }nI|jjjd kr d |d}n(|jjjd kr d |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. rrpinfo?rXpinfo2z??rrsystemrrz!!N)rrrFrrrrrrr rprrhrrqrrs r(rzMagicFinder.visit_Exprs`& dj#( + + 0A$*/0R0R  11Dz#'7777g%%'d1g--CC!W(((tAw..CC'd1g--CAw-}47}},%11#$q'mm%44$47nn$$ K ) * 1 1tz4c::    4     r')r*N) r!r"r#rrrFAssignrrrr&r'r(rnrnfs$UUUU!!!!!!B'!sx'!D'!'!'!'!'!'!r'rn)1rrFr dataclassessecretssys functoolsrimportlib.utilrtypingrrrr version_infor typing_extensions black.outputr black.reportr frozensetTRANSFORMED_MAGICSr9PYTHON_CELL_MAGICS token_hexr] dataclassrboolr0r$r>rCrTr`rKrLr{exprrrrr NodeVisitorrdrrnr&r'r(rs://  $$$$$$............w       ++++++''''''Y  9Y    d###       $#         '3'5d+;''''@&S&$&3&&&&0%%3%%5d;&7!78%%%%%%%%P3ss4HSHU3[0A+A%BHHHH>&-&-c4 +<&< =&-&-&-&-RS[(9c$ CH 3=)A    SX49d###        $#  !!!!!co!!!Dd###$#^!^!^!^!^!#/^!^!^!^!^!r'