U ccu)@sRUddlZejdkstddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZd d dgZGdd d eZiZ e eee!fee gdffe"d<ej#ee gdfe ddddZ$ej#ee ddddZ%Gdd d Z&Gddde&Z'dS)Nwin32)AbstractEventLoop) CallableContextManagerDict GeneratorListOptionalSetTextIOTupleUnion)get_event_loop)KeyPress)Input)PosixStdinReader) Vt100Parser Vt100Inputraw_mode cooked_modec@seZdZUdZeZeeed<e ddddZ e gdfe dddd Z e dd d d Zeed d dZeed ddZeed ddZe dd ddZe dd ddZed ddZed ddZdS)rzz Vt100 input for Posix systems. (This uses a posix file descriptor that can be registered in the event loop.) _fds_not_a_terminalN)stdinreturnc sz |WnHtjk rT}z(dtjkr8td|n td|W5d}~XYnX|}|}|s|tjkrd}tj ||tj tj ||_ |_ g_tj |jd_tfdd_dS)Nz idlelib.runzUz%Vt100Input.__init__..)filenoioUnsupportedOperationsysmodulesisattyrrstderrwriteflushaddr_filenorrr stdin_readerr vt100_parser)r rer*fdmsgr!rr"__init__/s0      zVt100Input.__init__)input_ready_callbackrcCs t||S)zj Return a context manager that makes this input active in the current event loop. )_attached_input)r r6r!r!r"attachXszVt100Input.attachrcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )_detached_inputrr!r!r"detach_szVt100Input.detachcCs&|j}|j||j}g|_|S)zRead list of KeyPress.)r0readr1feedr)r dataresultr!r!r" read_keysfs   zVt100Input.read_keyscCs|j|j}g|_|S)zc Flush pending keys and return them. (Used for flushing the 'escape' key.) )r1r-r)r r?r!r!r" flush_keysss zVt100Input.flush_keyscCs|jjSr)r0closedrr!r!r"rBszVt100Input.closedcCst|jSr)rrr%rr!r!r"rszVt100Input.raw_modecCst|jSr)rrr%rr!r!r"rszVt100Input.cooked_modecCs |jSr)rr%rr!r!r"r%szVt100Input.filenocCs d|jS)Nzfd-)r/rr!r!r"typeahead_hashszVt100Input.typeahead_hash)__name__ __module__ __qualname____doc__setrr int__annotations__r r5rrr8r;rrr@rApropertyboolrBrrr%strrCr!r!r!r"r%s ) _current_callbacks)NNN)inputcallbackrc #sttf}ddfdd }z|Wntk r\tYnXtf<z dVW5|r||tf<n tf=XdS)z Context manager that makes this input active in the current event loop. :param input: :class:`~prompt_toolkit.input.Input` object. :param callback: Called when the input is ready to read. Nr9csjrdS)a:Wrapper around the callback that already removes the reader when the input is closed. Otherwise, we keep continuously calling this callback, until we leave the context manager (which can happen a bit later). This fixes issues when piping /dev/null into a prompt_toolkit application.N)rB remove_readerr!rPr3rOloopr!r"callback_wrappers z)_attached_input..callback_wrapper)rr%rNget add_readerPermissionErrorEOFErrorrQ)rOrPpreviousrTr!rRr"r7s       r7)rOrc csdt}|}t||f}|r6||dt||f<z dVW5|r^||||t||f<XdSr)rr%rNrUrQrV)rOrSr3rYr!r!r"r:s    r:c@sfeZdZdZeddddZddddZeeed d d Zeeed d d Z e ddddZ dS)rz :: with raw_mode(stdin): ''' the pseudo-terminal stdin is now used in raw mode ''' We ignore errors when executing `tcgetattr` fails. N)r%rcCs<||_|zt||_Wntjk r6d|_YnXdSr)r%termios tcgetattr attrs_beforeerror)r r%r!r!r"r5s zraw_mode.__init__r9cCszzt|j}Wntjk r&YnPX||tj|tj<||tj|tj<d|tj tj <t |jtj |dS)Nr) rZr[r%r] _patch_lflagttyZLFLAG _patch_iflagZIFLAGCCVMIN tcsetattrTCSANOW)r Znewattrr!r!r" __enter__szraw_mode.__enter__attrsrcCs|tjtjBtjBtjB@SrrZECHOICANONIEXTENISIGclsrgr!r!r"r^szraw_mode._patch_lflagcCs$|tjtjBtjBtjBtjB@Sr)rZIXONIXOFFICRNLINLCRIGNCRrmr!r!r"r`szraw_mode._patch_iflag)arcGs>|jdk r:zt|jtj|jWntjk r8YnXdSr)r\rZrcr%rdr])r rtr!r!r"__exit__"s  zraw_mode.__exit__) rDrErFrGrIr5re classmethodr^r`objectrur!r!r!r"rs c@s8eZdZdZeeedddZeeedddZdS)rz The opposite of ``raw_mode``, used when we need cooked mode inside a `raw_mode` block. Used in `Application.run_in_terminal`.:: with cooked_mode(stdin): ''' the pseudo-terminal stdin is now used in cooked mode. ''' rfcCs|tjtjBtjBtjBBSrrhrmr!r!r"r^6szcooked_mode._patch_lflagcCs |tjBSr)rZrqrmr!r!r"r`:szcooked_mode._patch_iflagN)rDrErFrGrvrIr^r`r!r!r!r"r-s )(r(platformAssertionError contextlibr&rZr_asynciortypingrrrrrr r r r r Zprompt_toolkit.eventlooprZ key_bindingrbaserZ posix_utilsrr1r__all__rrNrIrJcontextmanagerr7r:rrr!r!r!r"s> 0     o 1R