o Zh)@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)rr1rrrattachMs 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+closedrrrrr@vszVt100Input.closedcCt|jSr)rrr rrrrrzzVt100Input.raw_modecCrAr)rrr rrrrr}rBzVt100Input.cooked_modeintcCs |jSr)rr rrrrr s zVt100Input.filenostrcCs d|jS)Nzfd-)r*rrrrtypeahead_hashs zVt100Input.typeahead_hashN)rr rr)r1r2rr3)rr3)rr8)rr?)rrC)rrD)__name__ __module__ __qualname____doc__setr__annotations__r0r5r7r=r>propertyr@rrr rErrrrrs   )       rz>dict[tuple[AbstractEventLoop, int], Callable[[], None] | None]_current_callbacksinputcallbackr2rGenerator[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)r@ remove_readerrrOr.rNlooprrcallback_wrappers  z)_attached_input..callback_wrapperNrr)rr rMget add_readerPermissionErrorEOFErrorrQ)rNrOpreviousrTrrRrr4s.         r4c cst}|}t||f}|r||dt||f<zdVW|r1||||t||f<dSdS|rA||||t||f<wr)rr rMrVrQrW)rNrSr.rZrrrr6s    r6c@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 rCrrcCs:||_|z t||_WdStjyd|_YdSwr)r termios tcgetattr attrs_beforeerror)rr rrrr0s zraw_mode.__init__cCszzt|j}Wn tjyYdSw||tj|tj<||tj|tj<d|tj tj <t |jtj |dS)Nr ) r[r\r r^ _patch_lflagttyZLFLAG _patch_iflagZIFLAGCCVMIN tcsetattrTCSANOW)rZnewattrrrr __enter__szraw_mode.__enter__attrscCs|tjtjBtjBtjB@Srr[ECHOICANONIEXTENISIGclsrgrrrr_szraw_mode._patch_lflagcCs$|tjtjBtjBtjBtjB@Sr)r[IXONIXOFFICRNLINLCRIGNCRrmrrrraszraw_mode._patch_iflagaobjectcGs@|jdurz t|jtj|jWdStjyYdSwdSr)r]r[rdr rer^)rrtrrr__exit__s zraw_mode.__exit__N)r rCrrrUrgrCrrC)rtrurr) rFrGrHrIr0rf classmethodr_rarvrrrrrs    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. ''' rgrCrcCs|tjtjBtjBtjBBSrrhrmrrrr_+szcooked_mode._patch_lflagcCs |tjBSr)r[rqrmrrrra/s zcooked_mode._patch_iflagNrw)rFrGrHrIrxr_rarrrrr"s  r)rNrrOr2rrP)rNrrrP)! __future__rr#platform contextlibr!r[r`Zasynciorrtypingrrrr Z key_bindingr baser Z posix_utilsrr,r__all__rrMrKcontextmanagerr4r6rrrrrrs.    o  1 R