!.d4dZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ej dkrddlmZnddlmZddlmZddlmZed Zed Zed ZejZejd GddZedededefdZdede eeffdZdededefdZ dede ee effdZ!dededefdZ"dede ee effdZ#dede ee effdZ$dede edefdZ%dej&deej'fd Z(d!e ej&de efd"Z)ejd Gd#d$Z*Gd%d&ej+Z,ejd Gd'd(Z-Gd)d*ej+Z.dS)+z)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) ENDMARKERNLNEWLINECOMMENTDEDENTUNIMPORTANT_WS ESCAPED_NL)captureprunpypypythonpython3timetimeitT)frozenc$eZdZUeed<eed<dS) ReplacementmasksrcN)__name__ __module__ __qualname__str__annotations__9lib/python3.11/site-packages/black/handle_ipynb_magics.pyrr4s" III HHHHHr&rverbosequietreturncd ddl}ddl}dS#t$r|s|sd}t|YdSwxYw)NrTz}Skipping .ipynb files as Jupyter dependencies are not installed. You can fix this by running ``pip install "black[jupyter]"``F)IPython tokenize_rtModuleNotFoundErrorr )r(r)r,r-msgs r'"jupyter_dependencies_are_installedr0:sn t   % O  HHHuus  //rcddlm}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)rr3r4r5tokenstrailing_semicolonidxtokens r'remove_trailing_semicolonr>Ks 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)rz{INTERNAL ERROR: Was not able to reinstate trailing semicolon. Please report a bug on https://github.com/psf/black/issues. N) r-r3r4r5r8r9_replacerAssertionErrorr#)rr?r3r4r5r:r<r=s r'put_trailing_semicolon_backrCks " 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_cellrTs')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)rrUnbytesr=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)rrOtreecell_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) rrO 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)rrO 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.header<s6 ; 21 11DK11 1DIr&N)r r!r"r#r$rpropertyrgr%r&r'rr6s] III SM III     X   r&rcJeZdZdZddeeddfdZdejddfdZ dS) rda^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. Nrfr*c||_dSr)rf)rrfs r'__init__zCellMagicFinder.__init__Vs $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_ExprYs tz38 , , T!$*/22 T $(888 11D'T!WT!W4PQ7SSSDO 4     r&r) r r!r"__doc__rrrrFExprrr%r&r'rdrdDsn"%%8I#6%$%%%% !sx !D ! ! ! ! ! !r&rdc$eZdZUeed<eed<dS)OffsetAndMagicrqrUN)r r!r"intr$r#r%r&r'rres"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__s7B7Nt7T7T 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|rrs r' visit_AssignzMagicFinder.visit_Assigns1 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'rnrnms$UUUU!!!!!!B'!sx'!D'!'!'!'!'!'!r&rn)/rrFr dataclassessecretssys functoolsrtypingrrrr version_infor typing_extensions black.outputr black.reportr frozensetTRANSFORMED_MAGICSr9PYTHON_CELL_MAGICS token_hexr] dataclassrboolr0r#r>rCrTr`rKrLr{exprrrrr NodeVisitorrdrrnr%r&r'rs5//  ............w       ++++++''''''Y9  Y     d###       $#   4        '3'5d+;''''@&S&$&3&&&&0%%3%%5d;&7!78%%%%%%%%P3ss4HSHU3[0A+A%BHHHH>&-&-c4 +<&< =&-&-&-&-RS[(9c$ CH 3=)A    SX49d###        $#  !!!!!co!!!Bd###$#^!^!^!^!^!#/^!^!^!^!^!r&