a ÂóFbnGã@sBdZddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z dZ dZ dZd Zd Zd Zd Zd ZdZe e eeeeeeeg ZGdd„dejdZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZejdd„ƒZdLdd„Zdd„Zd d!„Z d"d#„Z!d$d%„Z"d&d'„Z#d(d)„Z$d*d+„Z%e ee e ee!ee!ee"ee#ee$ee%iZ&ejd,d-„ƒZ'e (d.¡Z)e (d/ej*¡Z+e,ƒZ-e,ƒZ.d0d1„Z/d2d3„Z0d4d5„Z1ejd6d7„ƒZ2ejdMd9d:„ƒZ3dNd;d<„Z4ejd=d>„ƒZ5ejd?d@„ƒZ6ejdAdB„ƒZ7dCZ8e (dD 9e8¡ej*¡Z:dEZ;ejdFdG„ƒZejdJdK„ƒZ?dS)Oa,DEPRECATED: Input transformer classes to support IPython special syntax. This module was deprecated in IPython 7.0, in favour of inputtransformer2. This includes the machinery to recognise and transform ``%magic`` commands, ``!system`` commands, ``help?`` querying, prompt stripping, and so forth. éN)Úgenerate_tokensÚ untokenizeÚ TokenError)ÚStringIO)ÚLineInfoú!z!!ú?ú??ú%z%%ú,ú;ú/c@s8eZdZdZejdd„ƒZejdd„ƒZedd„ƒZ dS) ÚInputTransformerz6Abstract base class for line-based input transformers.cCsdS)a1Send a line of input to the transformer, returning the transformed input or None if the transformer is waiting for more input. Must be overridden by subclasses. Implementations may raise ``SyntaxError`` if the input is invalid. No other exceptions may be raised. N©©ÚselfÚlinerrú.transformer_factory)Ú functoolsÚwraps)rrrrrrÚwrapBszInputTransformer.wrapN) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚabcÚabstractmethodrrÚ classmethodrrrrrr*s  r)Ú metaclassc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚStatelessInputTransformerzDWrapper for a stateless input transformer implemented as a function.cCs ||_dSr©r©rrrrrÚ__init__Osz"StatelessInputTransformer.__init__cCs d |j¡S)Nz%StatelessInputTransformer(func={0!r}))ÚformatrrrrrÚ__repr__Rsz"StatelessInputTransformer.__repr__cCs | |¡S)zQSend a line of input to the transformer, returning the transformed input.r)rrrrrUszStatelessInputTransformer.pushcCsdS)z!No-op - exists for compatibility.NrrrrrrZszStatelessInputTransformer.resetN©r r!r"r#r+r-rrrrrrr(Ms r(c@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚCoroutineInputTransformerz†óz-TokenInputTransformer.push..FrT) r5ÚappendÚallrr9ÚtokenizeÚNEWLINEÚNLÚ ERRORTOKENrr6Úoutput)rrÚtokensZ stop_at_NLZintokÚtrrrr„s"     zTokenInputTransformer.pushcCs&|j ¡| ¡t| |¡ƒ d¡S)Nr;)r5Úclearr6rrÚrstrip©rrHrrrrG›s zTokenInputTransformer.outputcCs0d |j¡}|j ¡| ¡|r,| d¡SdS)NÚr;)Újoinr5rJr6rK)rr>rrrr s   zTokenInputTransformer.resetN) r r!r"r#r+r6rrGrrrrrr4us r4cs$eZdZ‡fdd„Zdd„Z‡ZS)Úassemble_python_linescstt|ƒ d¡dSr)ÚsuperrOr+r©Ú __class__rrr+¨szassemble_python_lines.__init__cCs| ¡Sr)rrLrrrrG«szassemble_python_lines.output)r r!r"r+rGÚ __classcell__rrrQrrO§s rOccsnd}|V}|r| ¡rqg}|dur^| d¡rPt|ƒsP| |dd…¡dV}q| |¡q^qd |¡}qdS)z4Join lines following explicit line continuations (\)rMNú\éÿÿÿÿ)r<ÚendswithÚ has_commentrArN)rÚpartsrrrÚassemble_logical_lines®s  rYc Csn|dkr dnd|vrdnd}d ||g¡}| d¡\}}}| t¡}|durZd|||fSd ||||fSdS) zUPrepares a pinfo(2)/psearch call from a target name and the escape (i.e. ? or ??)r Zpinfo2Ú*ZpsearchZpinfoú Nú&%sget_ipython().run_line_magic(%r, %r)zG%sget_ipython().set_next_input(%r);get_ipython().run_line_magic(%r, %r))rNÚ partitionÚlstripÚ ESC_MAGIC) ÚtargetÚescÚlspaceÚ next_inputÚmethodÚargÚ t_magic_nameÚ_Ú t_magic_arg_srrrÚ_make_help_callÄs  þ  ÿricCs|j ¡ t¡}d|j|fS)zTranslate lines escaped with: !z%sget_ipython().system(%r))rr^Ú ESC_SHELLÚpre©Ú line_infoÚcmdrrrÚ _tr_systemÕsrocCs |j ¡dd…}d|j|fS)z Translate lines escaped with: !!éNz%sget_ipython().getoutput(%r))rr^rkrlrrrÚ _tr_system2ÚsrqcCs$|jdd…sdSt|j|j|jƒS)z"Translate lines escaped with: ?/??éNzget_ipython().show_usage())rriÚifunrark©rmrrrÚ_tr_helpßsrucCsVd}|j t¡r|jSd |j|jg¡ ¡}| d¡\}}}| t ¡}||j ||fS)zTranslate lines escaped with: %r\r[) rÚ startswithÚ ESC_MAGIC2rNrsÚthe_restÚstripr]r^r_rk)rmÚtplrnrfrgrhrrrÚ _tr_magicçs  r{cCsd|j|jd |j ¡¡fS)zTranslate lines escaped with: ,ú %s%s("%s")z", "©rkrsrNrxÚsplitrtrrrÚ _tr_quoteòs ÿrcCsd|j|j|jfS)zTranslate lines escaped with: ;r|)rkrsrxrtrrrÚ _tr_quote2÷s ÿr€cCsd|j|jd |j ¡¡fS)zTranslate lines escaped with: /z%s%s(%s)z, r}rtrrrÚ _tr_parenüs ÿrcCs4|r | ¡r|St|ƒ}|jtvr&|St|j|ƒS)zMTransform escaped commands - %magic, !system, ?help + various autocalls. )r<rraÚtr)rZlineinfrrrÚescaped_commands s   rƒz\s*a (%{0,2} (?!\d)[\w*]+ # Variable name (\.(?!\d)[\w*]+)* # .etc.etc ) (\?\??)$ # ? or ?? c Cs~t|ƒj}tƒ}z t|ƒD]}| |d¡qWnHtyx}z0d|jdvrZ| t¡n | t¡WYd}~n d}~00|S)z5Helper for has_comment and ends_in_comment_or_string.rzmulti-line stringN) rÚreadlineÚsetrÚaddrÚargsÚ_MULTILINE_STRINGÚ_MULTILINE_STRUCTURE)rr„ÚtoktypesrIÚerrrÚ _line_tokens$s    rŒcCstjt|ƒvS)a=Indicate whether an input line has (i.e. ends in, or is) a comment. This uses tokenize, so it can distinguish comments from # inside strings. Parameters ---------- src : string A single line input string. Returns ------- comment : bool True if source has a comment. )rCÚCOMMENTrŒ)ÚsrcrrrrW3srWcCst|ƒ}tj|vpt|vS)aIndicates whether or not an input line ends in a comment or within a multiline string. Parameters ---------- src : string A single line input string. Returns ------- comment : bool True if source ends in a comment or multiline string. )rŒrCrrˆ)rŽrŠrrrÚends_in_comment_or_stringDsrcCspt |¡}|dust|ƒr|S| d¡}| d¡}t |¡ d¡}| ¡| d¡kr^| d¡nd}t||||ƒS)z$Translate lines with ?/?? at the endNrrérr) Ú _help_end_reÚsearchrÚgroupÚ_initial_space_reÚmatchryrKri)rÚmr`rarbrcrrrÚhelp_endVs    r—Fccs¸d}t d¡}d}|V}|s$|V}q| t¡s@|dur|V}q.q| |¡rLq|}g}dV}|dur„| ¡dksr|s„| |¡dV}qZ| d¡\}}}| t¡}|||d  |¡f}qdS)zCaptures & transforms cell magics. After a cell magic is started, this stores up any lines it gets until it is reset (sent None). z(get_ipython().run_cell_magic(%r, %r, %r)z%%\w+\?rMNr[r;) ÚreÚcompilervrwr•ryrAr]r^rN)Zend_on_blank_linerzZcellmagic_help_rerÚfirstÚbodyZ magic_namergrrrÚ cellmagicfs2    ÿÿ  rœccs¾|dur |}d}|V}|dur q|jd|dd\}}|rX|sX| |¡rX|dur|V}qFq|V}|durhq|jd|dd\}}|V}|sŠ|r¨|dur¸|jd|ddV}qŠq|dur|V}q¨qdS)a&Remove matching input prompts from a block of input. Parameters ---------- prompt_re : regular expression A regular expression matching any input prompt (including continuation) initial_re : regular expression, optional A regular expression matching only the initial prompt, but not continuation. If no initial expression is given, prompt_re will be used everywhere. Used mainly for plain Python prompts, where the continuation prompt ``...`` is a valid Python expression in Python 3, so shouldn't be stripped. Notes ----- If `initial_re` and `prompt_re differ`, only `initial_re` will be tested against the first line. If any prompt is found on the first two lines, prompts will be stripped from the rest of the block. NrMrr)Úcount)Úsubnr•Úsub)Ú prompt_reÚ initial_reÚ turnoff_rerÚoutZn1Zn2rrrÚ_strip_promptsŽs, r¤cCs*t d¡}t d¡}t d¡}t|||ƒS)z:Strip the >>>/... prompts of the Python interactive shell.z^(>>>|\.\.\.)( |$)z ^>>>( |$)z^[%!]©r˜r™r¤)r r¡r¢rrrÚclassic_promptÉs   r¦cCs t d¡}t d¡}t||dS)z%Strip IPython's In [1]:/...: prompts.z^(In \[\d+\]: |\s*\.{3,}: ?)z^%%)r¢r¥)r r¢rrrÚ ipy_promptÓs  r§ccsxt d¡}d}|V}|durq| |¡}|rb| d¡}|durr| |¡rX|t|ƒd…}|V}q6q|dur|V}qbqdS)z©Remove leading indentation. If the first line starts with a spaces or tabs, the same whitespace will be removed from each following line until it is reset. z^[ \t]+rMNr)r˜r™r•r“rvÚlen)Zspace_rerr–ZspacerrrÚleading_indentÝs     r©zÎ(?P(\s*) ([\w\.]+) # Initial identifier (\s*,\s* \*?[\w\.]+)* # Further identifiers for unpacking \s*?,? # Trailing comma ) \s*=\s* z{}!\s*(?P.*)z %s = get_ipython().getoutput(%r)cCs&t |¡}|dur|St| dd¡S)z.*)z)%s = get_ipython().run_line_magic(%r, %r)cCsNt |¡}|dur|S| dd¡\}}| d¡\}}}| t¡}t|||fS)z;Transform assignment from magic commands (e.g. a = %who_ls)Nrªrnr[)Úassign_magic_rer•r“r]r^r_Úassign_magic_template)rr–Zm_lhsZm_cmdrfrgrhrrrÚassign_from_magics  r°)N)F)NN)@r#r$rr˜rCrrrÚiorZIPython.core.splitinputrrjZ ESC_SH_CAPZESC_HELPZ ESC_HELP2r_rwZ ESC_QUOTEZ ESC_QUOTE2Z ESC_PARENZ ESC_SEQUENCESÚABCMetarr(r/r4rOrrYrirorqrur{rr€rr‚rƒr™r”ÚVERBOSEr‘Úobjectrˆr‰rŒrWrr—rœr¤r¦r§r©Z _assign_patr,r«r¬r­r®r¯r°rrrrÚs˜  þ#2   ù  ú   ' ;   ÿ