o ˷eu)@sVUddlZejdks JddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZgd ZGd ddeZiZe eee fee gdffe!d<ej"dede gdfde dfddZ#ej"dede dfddZ$GdddZ%Gddde%Z&dS)Nwin32)AbstractEventLoop) CallableContextManagerDict GeneratorListOptionalSetTextIOTupleUnion)get_event_loop)KeyPress)Input)PosixStdinReader) Vt100Parser) Vt100Inputraw_mode cooked_modec@seZdZUdZeZeeed<de ddfddZ de gdfde dfd d Z de dfd d Zdeefd dZdeefddZedefddZde dfddZde dfddZdefddZdefddZdS)rzz Vt100 input for Posix systems. (This uses a posix file descriptor that can be registered in the event loop.) _fds_not_a_terminalstdinreturnNc sz|Wntjy$}zdtjvrtd|td|d}~ww|}|}|sI|tjvrId}tj ||tj tj ||_ |_ g_tj |jd_tfdd_dS)Nz idlelib.runzUs z%Vt100Input.__init__..)filenoioUnsupportedOperationsysmodulesisattyrrstderrwriteflushaddr_filenorrr stdin_readerr vt100_parser)r!rer*fdmsgr"r r#__init__/s4         zVt100Input.__init__input_ready_callbackcCs t||S)zj Return a context manager that makes this input active in the current event loop. )_attached_input)r!r6r"r"r#attachXs zVt100Input.attachcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )_detached_inputr r"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)r0closedr r"r"r#rAszVt100Input.closedcCt|jSr)rrr%r r"r"r#rzVt100Input.raw_modecCrBr)rrr%r r"r"r#rrCzVt100Input.cooked_modecCs |jSr)rr%r r"r"r#r%s zVt100Input.filenocCs d|jS)Nzfd-)r/r r"r"r#typeahead_hashs zVt100Input.typeahead_hash)__name__ __module__ __qualname____doc__setrr int__annotations__r r5rrr8r:rrr?r@propertyboolrArrr%strrDr"r"r"r#r%s ) r_current_callbacksinputcallbackr)NNNc #sttf}dfdd }z|Wn ty*twtf<z dVW|rK||tf<dStf=dS|rf||tf<wtf=w)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. rNcsjrdS)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)rA remove_readerr"rQr3rPloopr"r#callback_wrappers  z)_attached_input..callback_wrapperrN)rr%rOget add_readerPermissionErrorEOFErrorrR)rPrQpreviousrUr"rSr#r7s.         r7c cst}|}t||f}|r||dt||f<zdVW|r1||||t||f<dSdS|rA||||t||f<wr)rr%rOrWrRrX)rPrTr3r[r"r"r#r9s    r9c@sjeZdZdZdeddfddZdddZed edefd d Zed edefd d Z de ddfddZ dS)rz :: with raw_mode(stdin): ''' the pseudo-terminal stdin is now used in raw mode ''' We ignore errors when executing `tcgetattr` fails. r%rNcCs:||_|z t||_WdStjyd|_YdSwr)r%termios tcgetattr attrs_beforeerror)r!r%r"r"r#r5s zraw_mode.__init__cCszzt|j}Wn tjyYdSw||tj|tj<||tj|tj<d|tj tj <t |jtj |dS)Nr) r\r]r%r_ _patch_lflagttyLFLAG _patch_iflagIFLAGCCVMIN tcsetattrTCSANOW)r!newattrr"r"r# __enter__szraw_mode.__enter__attrscCs|tjtjBtjBtjB@Srr\ECHOICANONIEXTENISIGclsrkr"r"r#r`szraw_mode._patch_lflagcCs$|tjtjBtjBtjBtjB@Sr)r\IXONIXOFFICRNLINLCRIGNCRrqr"r"r#rcszraw_mode._patch_iflagacGs@|jdurz t|jtj|jWdStjyYdSwdSr)r^r\rgr%rhr_)r!rxr"r"r#__exit__"s zraw_mode.__exit__rV) rErFrGrHrJr5rj classmethodr`rcobjectryr"r"r"r#rs rc@s<eZdZdZededefddZededefddZdS) 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. ''' rkrcCs|tjtjBtjBtjBBSrrlrqr"r"r#r`6szcooked_mode._patch_lflagcCs |tjBSr)r\rurqr"r"r#rc:s zcooked_mode._patch_iflagN)rErFrGrHrzrJr`rcr"r"r"r#r-s r)'r(platform contextlibr&r\raasynciortypingrrrrrr r r r r prompt_toolkit.eventloopr key_bindingrbaser posix_utilsrr1r__all__rrOrJrKcontextmanagerr7r9rrr"r"r"r#s>  0    o  1R