o ^i<@sndZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ej dkr8ddl mZnddlmZddlmZdd lmZdd lmZed Zed Zed ZejddGdddZededefddZdededdfddZdede eeffddZ!dededefddZ"dede ee#effdd Z$d!e%defd"d#Z&ded$edefd%d&Z'dede ee#effd'd(Z(dede ee#effd)d*Z)ded+e#edefd,d-Z*dedefd.d/Z+d0ej,deej-fd1d2Z.d3e#ej,de#efd4d5Z/ejddGd6d7d7Z0Gd8d9d9ej1Z2ejddGd:d;d;Z3Gdz)Functions to process IPython magics with.N) lru_cache) find_spec)Optional) ) TypeGuard)Mode)out)NothingChanged)zget_ipython().run_cell_magiczget_ipython().systemzget_ipython().getoutputzget_ipython().run_line_magic) ENDMARKERNLNEWLINECOMMENTDEDENTUNIMPORTANT_WS ESCAPED_NL)captureprunpypypythonpython3timetimeitT)frozenc@seZdZUeed<eed<dS) ReplacementmasksrcN)__name__ __module__ __qualname__str__annotations__r"r"^/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/black/handle_ipynb_magics.pyr0  rwarnreturncCs0tdduo tddu}|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% installedmsgr"r"r#"jupyter_dependencies_are_installed6sr+rmodecsZtfddtDr tt}|dr)|jdddddt|jBvr+tdSdS) aCheck that cell does not already contain TransformerManager transformations, or non-Python cell magics, which might cause tokenizer_rt to break because of indentations. If a cell contains ``!ls``, then it'll be transformed to ``get_ipython().system('ls')``. However, if the cell originally contained ``get_ipython().system('ls')``, then it would get transformed in the same way: >>> TransformerManager().transform_cell("get_ipython().system('ls')") "get_ipython().system('ls')\n" >>> TransformerManager().transform_cell("!ls") "get_ipython().system('ls')\n" Due to the impossibility of safely roundtripping in such situations, cells containing transformed magics will be ignored. c3s|]}|vVqdSNr").0transformed_magicrr"r# Usz validate_cell..%%)maxsplitrN)anyTRANSFORMED_MAGICSr _get_code_start startswithsplitPYTHON_CELL_MAGICSpython_cell_magics)rr,liner"r0r# validate_cellDs r>cCstddlm}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. rreversed_enumerate src_to_tokens tokens_to_srcFOP;T)r'r@rArBnameTOKENS_TO_IGNOREr)rr@rArBtokenstrailing_semicolonidxtokenr"r"r#remove_trailing_semicolon`s  rKhas_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?rDr0z{INTERNAL ERROR: Was not able to reinstate trailing semicolon. Please report a bug on https://github.com/psf/black/issues. N) r'r@rArBrErF_replacerAssertionErrorr )rrLr@rArBrGrIrJr"r"r#put_trailing_semicolon_backs  rOcCsg}z t|W||fStyYnwddlm}|}||}t|\}}||7}||}t|\}}t| t| krKt ||7}||fS)zMask IPython magics so content becomes parseable Python code. For example, %matplotlib inline 'foo' becomes b"25716f358c32750" 'foo' The replacements are returned, along with the transformed code. r)TransformerManager) astparse SyntaxErrorIPython.core.inputtransformer2rPtransform_cellreplace_cell_magicsreplace_magicslenstrip splitlinesr )r replacementsrPtransformer_manager transformedcell_magic_replacementsmagic_replacementsr"r"r# mask_cells$        r`n_charscCsP|dksJt|ddd}t|}t|d|kr"|dd}d|dS) zBCreate a randomly generated token that is n_chars characters long.rr5r3rNzb"")maxsecrets token_hexrX)ran_bytesrJr"r"r# create_tokens    rhmagiccCsX|sJt|}t|}d}||vr*t|}|d7}|dkr&td|d||vs|S)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. rr3dzINTERNAL 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)rXrhrN)rrirarJcounterr"r"r# get_tokens" rlcCsjg}t|}t}|||jdur||fS|jj}t||}|t||d|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 ) rQrRCellMagicFindervisit cell_magicheaderrlappendrbody)rr[treecell_magic_finderrrrr"r"r#rVs    rVc Csg}t}|t|g}t|dddD]B\}}||jvrU|j|}t|dkr4td|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. rnr3)startz#Expecting one magic per line, got: z< Please report a bug on https://github.com/psf/black/issues.rrmN) MagicFinderrprQrR enumerater:magicsrXrN col_offsetrirlrsrjoin) rr[ magic_findernew_srcsir=offsets_and_magicsr{rirr"r"r#rWs&      rWr[cCs|D] }||j|j}q|S)zRemove replacements from cell. For example "9b20" foo = bar becomes %%time foo = bar )replacerr)rr[ replacementr"r"r# unmask_cell.s rcCs:td|D]}|d}|r|ds|SqdS)a0Provides the first line where the code starts. Iterates over lines of code until it finds the first line that doesn't contain only empty spaces and comments. It removes any empty spaces at the start of the line and returns it. If such line doesn't exist, it returns an empty string. z.+r#)refinditergrouplstripr9)rmatchr=r"r"r#r8@s r8nodecCs8t|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) isinstancerQ AttributevalueCallfuncNameid)rr"r"r#_is_ipython_magicOs   rargscCs:g}|D]}t|tjrt|jtsJ||jq|Sr-)rrQConstantrr rs)rstr_argsargr"r"r# _get_str_args^s rc@s<eZdZUeed<eeed<eed<edefddZdS) CellMagicrEparamsrtr&cCs&|jr d|jd|jSd|jS)Nr2 )rrEselfr"r"r#rrls zCellMagic.headerN)rrrr r!rpropertyrrr"r"r"r#rfs  rc@s<eZdZdZd deeddfddZdejddfdd Z dS) roahFind 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\n foo() would have been transformed to get_ipython().run_cell_magic('time', '', 'foo()\n') and we look for instances of the latter. Nrqr&cCs ||_dSr-)rq)rrqr"r"r#__init__s zCellMagicFinder.__init__rcCs^t|jtjr(t|jjr(|jjjdkr(t|jj}t |d|d|dd|_ | |dS)z)Find cell magic, extract header and body.run_cell_magicrr3r5)rErrtN) rrrQrrrattrrrrrq generic_visit)rrrr"r"r# visit_Exprs    zCellMagicFinder.visit_Exprr-) rrr__doc__rrrrQExprrr"r"r"r#rotsroc@seZdZUeed<eed<dS)OffsetAndMagicr{riN)rrrintr!r r"r"r"r#rr$rc@sBeZdZdZd ddZdejddfddZdejddfd 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). r&NcCstt|_dSr-) collections defaultdictlistrzrr"r"r#rszMagicFinder.__init__rcCst|jtjrYt|jjrYt|jj}|jjjdkr"d|d}n(|jjjdkr>d|d}|dr=|d|d7}n t 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%r3rzUnexpected IPython magic zC found. Please report a bug on https://github.com/psf/black/issues.N)rrrQrrrrrrrNrzlinenorsrr{rrrrrr"r"r# visit_Assigns$  zMagicFinder.visit_AssigncCst|jtjrzt|jjrzt|jj}|jjjdkrK|ddkr(d|d}nC|ddkr6d|d}n5d|d}|drJ|d |d7}n |jjjd krZd |d}n|jjjd krid |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?r3pinfo2z??rrsystemrrz!!N)rrrQrrrrrrr rzrrsrr{rrr"r"r#rs(    zMagicFinder.visit_Expr)r&N) rrrrrrQAssignrrrr"r"r"r#rxs  !rx)5rrQr dataclassesrresys functoolsrimportlib.utilrtypingr version_infortyping_extensions black.moder black.outputr black.reportr frozensetr7rFr; dataclassrboolr+r r>tuplerKrOrr`rrhrlrVrWrr8exprrrrr NodeVisitorrorrxr"r"r"r#sR           * )  "