o ih)@sUddlmZddlZejdksJddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZddlmZdd lmZdd lmZdd lmZgd ZGd ddeZiZ de!d<ej"dddZ#ej"dddZ$GdddZ%Gddde%Z&dS) ) annotationsNwin32)AbstractEventLoopget_running_loop) CallableContextManagerDict GeneratorListOptionalSetTextIOTupleUnion)KeyPress)Input)PosixStdinReader) Vt100Parser) Vt100Inputraw_mode cooked_modec@seZdZUdZeZded<d$dd Zd%d dZd&ddZ d'ddZ d'ddZ e d(ddZ d&ddZd&ddZd)ddZd*d!d"Zd#S)+rzz Vt100 input for Posix systems. (This uses a posix file descriptor that can be registered in the event loop.) zset[int]_fds_not_a_terminalstdinr returnNonec 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_callbackCallable[[], None]ContextManager[None]cCs t||S)zj Return a context manager that makes this input active in the current event loop. )_attached_input)r#r8r$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.detachlist[KeyPress]cCs&|j}|j||j}g|_|S)zRead list of KeyPress.)r2readr3feedr)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.) )r3r/r)r#rCr$r$r% flush_keysss zVt100Input.flush_keysboolcCs|jjSr)r2closedr"r$r$r%rGszVt100Input.closedcCt|jSr)rrr'r"r$r$r%rzVt100Input.raw_modecCrHr)rrr'r"r$r$r%rrIzVt100Input.cooked_modeintcCs |jSr)rr'r"r$r$r%r's zVt100Input.filenostrcCs d|jS)Nzfd-)r1r"r$r$r%typeahead_hashs zVt100Input.typeahead_hashN)rr rr)r8r9rr:)rr:)rr?)rrF)rrJ)rrK)__name__ __module__ __qualname____doc__setr__annotations__r7r<r>rDrEpropertyrGrrr'rLr$r$r$r%r%s   )       rz>dict[tuple[AbstractEventLoop, int], Callable[[], None] | None]_current_callbacksinputcallbackr9rGenerator[None, None, None]c #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. rrcsjrdS)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)rG remove_readerr$rVr5rUloopr$r%callback_wrappers  z)_attached_input..callback_wrapperNrr)rr'rTget add_readerPermissionErrorEOFErrorrX)rUrVpreviousr[r$rYr%r;s.         r;c cst}|}t||f}|r||dt||f<zdVW|r1||||t||f<dSdS|rA||||t||f<wr)rr'rTr]rXr^)rUrZr5rar$r$r%r=s    r=c@sJeZdZdZdddZddd Zedd d Zedd dZdddZ dS)rz :: with raw_mode(stdin): ''' the pseudo-terminal stdin is now used in raw mode ''' We ignore errors when executing `tcgetattr` fails. r'rJrrcCs:||_|z t||_WdStjyd|_YdSwr)r'termios tcgetattr attrs_beforeerror)r#r'r$r$r%r7s zraw_mode.__init__cCszzt|j}Wn tjyYdSw||tj|tj<||tj|tj<d|tj tj <t |jtj |dS)Nr) rbrcr're _patch_lflagttyLFLAG _patch_iflagIFLAGCCVMIN tcsetattrTCSANOW)r#newattrr$r$r% __enter__szraw_mode.__enter__attrscCs|tjtjBtjBtjB@SrrbECHOICANONIEXTENISIGclsrqr$r$r%rfszraw_mode._patch_lflagcCs$|tjtjBtjBtjBtjB@Sr)rbIXONIXOFFICRNLINLCRIGNCRrwr$r$r%riszraw_mode._patch_iflagaobjectcGs@|jdurz t|jtj|jWdStjyYdSwdSr)rdrbrmr'rnre)r#r~r$r$r%__exit__"s zraw_mode.__exit__N)r'rJrrr\rqrJrrJ)r~rrr) rMrNrOrPr7rp classmethodrfrirr$r$r$r%rs    rc@s,eZdZdZed ddZed ddZd S) 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. ''' rqrJrcCs|tjtjBtjBtjBBSrrrrwr$r$r%rf6szcooked_mode._patch_lflagcCs |tjBSr)rbr{rwr$r$r%ri:s zcooked_mode._patch_iflagNr)rMrNrOrPrrfrir$r$r$r%r-s  r)rUrrVr9rrW)rUrrrW)' __future__rr*platform contextlibr(rbrgasynciorrtypingrrrr r r r r rr key_bindingrbaser posix_utilsrr3r__all__rrTrRcontextmanagerr;r=rrr$r$r$r%s.0   o  1 R