o wh)@sUddlmZddlZejdksJddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZmZddlmZdd lmZdd lmZdd lmZgd ZGd ddeZiZded<ejdddZejdddZGdddZGdddeZ dS) ) annotationsNwin32)AbstractEventLoopget_running_loop)CallableContextManager GeneratorTextIO)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.runzJs z%Vt100Input.__init__..)filenoioUnsupportedOperationsysmodulesisattyrrstderrwriteflushaddr_filenorrr stdin_readerr vt100_parser)rrer&fdmsgrrr__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)rr2rrrattachMs zVt100Input.attachcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )_detached_inputrrrrdetachTszVt100Input.detachlist[KeyPress]cCs&|j}|j||j}g|_|S)zRead list of KeyPress.)r,readr-feedr)rdataresultrrr read_keys[s  zVt100Input.read_keyscCs|j|j}g|_|S)zc Flush pending keys and return them. (Used for flushing the 'escape' key.) )r-r)r)rr=rrr flush_keyshs zVt100Input.flush_keysboolcCs|jjSr)r,closedrrrrrAvszVt100Input.closedcCt|jSr)rrr!rrrrrzzVt100Input.raw_modecCrBr)rrr!rrrrr}rCzVt100Input.cooked_modeintcCs |jSr)rr!rrrrr!s zVt100Input.filenostrcCs d|jS)Nzfd-)r+rrrrtypeahead_hashs zVt100Input.typeahead_hashN)rr rr)r2r3rr4)rr4)rr9)rr@)rrD)rrE)__name__ __module__ __qualname____doc__setr__annotations__r1r6r8r>r?propertyrArrr!rFrrrrrs   )       rz>dict[tuple[AbstractEventLoop, int], Callable[[], None] | None]_current_callbacksinputcallbackr3rGenerator[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)rA remove_readerrrPr/rOlooprrcallback_wrappers  z)_attached_input..callback_wrapperNrr)rr!rNget add_readerPermissionErrorEOFErrorrR)rOrPpreviousrUrrSrr5s.         r5c cst}|}t||f}|r||dt||f<zdVW|r1||||t||f<dSdS|rA||||t||f<wr)rr!rNrWrRrX)rOrTr/r[rrrr7s    r7c@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!rDrrcCs:||_|z t||_WdStjyd|_YdSwr)r!termios tcgetattr attrs_beforeerror)rr!rrrr1s 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)rnewattrrrr __enter__szraw_mode.__enter__attrscCs|tjtjBtjBtjB@Srr\ECHOICANONIEXTENISIGclsrkrrrr`szraw_mode._patch_lflagcCs$|tjtjBtjBtjBtjB@Sr)r\IXONIXOFFICRNLINLCRIGNCRrqrrrrcszraw_mode._patch_iflagaobjectcGs@|jdurz t|jtj|jWdStjyYdSwdSr)r^r\rgr!rhr_)rrxrrr__exit__s zraw_mode.__exit__N)r!rDrrrVrkrDrrD)rxryrr) rGrHrIrJr1rj classmethodr`rcrzrrrrrs    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. ''' rkrDrcCs|tjtjBtjBtjBBSrrlrqrrrr`+szcooked_mode._patch_lflagcCs |tjBSr)r\rurqrrrrc/s zcooked_mode._patch_iflagNr{)rGrHrIrJr|r`rcrrrrr"s  r)rOrrPr3rrQ)rOrrrQ)! __future__rr$platform contextlibr"r\raasynciorrtypingrrrr key_bindingr baser posix_utilsrr-r__all__rrNrLcontextmanagerr5r7rrrrrrs.    o  1 R