U  W[8.@sdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZGdd d ZGd d d ejZd Zd ZdZdZdZdZGdddejZGdddZddZGdddeZdS)aO Line-input oriented interactive interpreter loop. Provides classes for handling Python source input and arbitrary output interactively from a Twisted application. Also included is syntax coloring code with support for VT102 terminals, control code handling (^C, ^D, ^Q), and reasonable handling of Deferreds. @author: Jp Calderone N)BytesIO)recvline)defer) _tokenize_get_async_param) TokenPrinterc@s8eZdZdZdZdZddZddZdd Zd d Z d S) FileWrapperz Minimal write-file-like object. Writes are translated into addOutput calls on an object passed to __init__. Newlines are also converted from network to local style. rZnormalcCs ||_dSN)o)selfr r 7/usr/lib/python3/dist-packages/twisted/conch/manhole.py__init__&szFileWrapper.__init__cCsdSr r r r r r flush*szFileWrapper.flushcCs|j|dddS)Nz  )r addOutputreplace)r datar r r write.szFileWrapper.writecCs|d|dS)N)rjoin)r linesr r r writelines2szFileWrapper.writelinesN) __name__ __module__ __qualname____doc__Z softspacestaterrrrr r r r rsrc@sXeZdZdZdZdddZddZd d Zd d Zd dZ ddZ ddZ dddZ dS)ManholeInterpreteraf Interactive Interpreter with special output and Deferred support. Aside from the features provided by L{code.InteractiveInterpreter}, this class captures sys.stdout output and redirects it to the appropriate location (the Manhole protocol instance). It also treats Deferreds which reach the top-level specially: each is formatted to the user with a unique identifier and a new callback and errback added to it, each of which will format the unique identifier and the result with which the Deferred fires and then pass it on to the next participant in the callback chain. rN cCs,tj||i|_||_||_|dSr )codeInteractiveInterpreterr_pendingDeferredshandlerfilename resetBuffer)r r$localsr%r r r rFs zManholeInterpreter.__init__cCs g|_dS)z) Reset the input buffer. N)bufferrr r r r&NszManholeInterpreter.resetBuffercCs@|j|d|j}|d}|||j}|s<||S)a Push a line to the interpreter. The line should not have a trailing newline; it may have internal newlines. The line is appended to a buffer and the interpreter's runsource() method is called with the concatenated contents of the buffer as source. If this indicates that the command was executed or invalid, the buffer is reset; otherwise, the command is incomplete, and the buffer is left as it was after the line was appended. The return value is 1 if more input is required, 0 if the line was dealt with in some way (this is the same as runsource()). @param line: line of text @type line: L{bytes} @return: L{bool} from L{code.InteractiveInterpreter.runsource}  utf-8)r(appendrdecodeZ runsourcer%r&)r linesourcemorer r r pushUs   zManholeInterpreter.pushcOsXtj|j}t_z:tjt|j}t_ztjj|f||W5|t_XW5|t_XdSr )sys displayhookstdoutrr$r!r"runcode)r akwZorighookZorigoutr r r r4ps zManholeInterpreter.runcodecCs||jd<t|tjrt|dr0|t|qt||jkr^|d|jt|dfq|j}|j }||f|t|<|j d7_ |j |j |j ||f||fd|d|fn|dk r|t|dS)N_resultzr)Z callbackArgsZ errbackArgs) r' isinstancerZDeferredhasattrrrepridr# numDeferredsZ addCallbacks_cbDisplayDeferred_ebDisplayDeferred)r objdkr r r r2|s"     zManholeInterpreter.displayhookcCs$|d||fd|jt|=|S)NzDeferred #%d called back: %rT)rr#r=)r r8rCrAr r r r?s z%ManholeInterpreter._cbDisplayDeferredcCs(|d||fd|jt|=|S)NzDeferred #%d failed: %rT)rZgetErrorMessager#r=)r ZfailurerCrAr r r r@s z%ManholeInterpreter._ebDisplayDeferredcKst|f|}|j||dSr )rr$r)r risAsynckwargsr r r rs zManholeInterpreter.write)Nr )N) rrrrr>rr&r0r4r2r?r@rr r r r r7s   r c@s`eZdZdZdZdddZddZddZd d Zd d Z d dZ ddZ dddZ ddZ dS)Manholea Mediator between a fancy line source and an interactive interpreter. This accepts lines from its transport and passes them on to a L{ManholeInterpreter}. Control commands (^C, ^D, ^\) are also handled with something approximating their normal terminal-mode behavior. It can optionally be constructed with a dict which will be used as the local namespace for any code executed. NcCs"tj||dk r||_dSr )rHistoricRecvLinercopy namespace)r rOr r r rs zManhole.__init__cCsftj|t||j|_|j|jt<|j |jt <|j |jt <|j |jt<|j|jt<|j|jt<dSr )rrMconnectionMaderrO interpreter handle_INTZ keyHandlersCTRL_C handle_EOFCTRL_D handle_FFCTRL_LZ handle_HOMECTRL_AZ handle_ENDCTRL_E handle_QUITCTRL_BACKSLASHrr r r rPs      zManhole.connectionMadecCsTd|_g|_d|_|j|j|jd|j|j|j|jdS)z~ Handle ^C as an interrupt keystroke by resetting the current input variables to their initial state. rsKeyboardInterruptN) pn lineBufferlineBufferIndexrQr&terminalnextLinerpsrr r r rRs    zManhole.handle_INTcCs |jr|jdn|dS)N)r]r_rrZrr r r rTszManhole.handle_EOFcCs |j|j|dS)zh Handle a 'form feed' byte - generally used to request a screen refresh/redraw. N)r_Z eraseDisplayZ cursorHomeZ drawInputLinerr r r rVs  zManhole.handle_FFcCs|jdSr )r_ZloseConnectionrr r r rZszManhole.handle_QUITcCs |jj}|d o|d S)Nr)sE)r_Z lastWriteendswith)r wr r r _needsNewlineszManhole._needsNewlinecKst|f|}|r<|j|jt|jt|j|j|j||r| r^|j |j|j|j|jr|j}g|_d|_ | |dS)Nr) rr_ eraseLinecursorBackwardlenr]rar\rrer`r^Z_deliverBuffer)r rrDrEZ oldBufferr r r rs     zManhole.addOutputcCs@|j|}t||_|r(|j|j|j|jdSr ) rQr0boolr\rer_r`rra)r r-r/r r r lineReceived s    zManhole.lineReceived)N)N)rrrrrOrrPrRrTrVrZrerrjr r r r rLs     rLc@sVeZdZdZdddddddd Zd Zd d Zd dZdddZddZ e e krRe Z dS) VT102Writerz Colorizer for Python tokens. A series of tokens are written to instances of this object. Each is colored in a particular way. The final line of the result of this is generally added to the output. sssssss)Z identifierkeywordZ parameterZvariablestringZnumberopscCs g|_dSr )writtenrr r r r'szVT102Writer.__init__cCs|j|d}|S)N) typeToColorget)r typerr r r color+szVT102Writer.colorNcCsH|rD|dkrD||}|r&|j||j||rD|j|jdS)N )ruror+ normalColor)r tokenrscr r r r0s    zVT102Writer.writecCsd|j}|ddS)Nrpr))rrostrip splitlines)r sr r r __bytes__:s zVT102Writer.__bytes__)N) rrrrrqrwrrurr~bytesstr__str__r r r r rks    rkc Csdt|ts|d}t}t|jj}t|}t|j D]"}|\}}}}} |||||| q8t|S)z Tokenize and colorize the given Python source. Returns a VT102-format colorized version of the last line of C{source}. @param source: Python source code @type source: L{str} or L{bytes} @return: L{bytes} of colorized source r*) r:rencoderkrrZ printtokenrrreadline) r.rdpr}rxZ tokenTypermstartendr-r r r lastColorizedLineDs   rc@s eZdZdZddZddZdS)ColoredManholez< A REPL which syntax colors input as users type it. cCsd|jjdd|jS)z Return a string containing the currently entered source. This is only the code which will be considered for execution next. r)rp)rrQr(r]rr r r getSourceas   zColoredManhole.getSourcecCs|jdkr|j|j|n|g|j|j|jd<|jd7_|rJdS|dkrb|j|dS|}z t|}Wn"tj k r|j|YnlX|j |j t |jt |j |jd|j|j |j|t |j|j}|r|j |dS)Ninsertr9 )moder]rr^r_rrrtokenize TokenErrorrfrgrhrar\)r ZchZmoreCharactersComingr.Z coloredLinenr r r characterReceivedms(    &z ColoredManhole.characterReceivedN)rrrrrrr r r r r\s r)rr!r1riorZ twisted.conchrZtwisted.internetrZtwisted.python.compatrrZtwisted.python.htmlizerrrr"rrSrUr[rWrXrYrMrLrkrrr r r r s$     kh1