o j *@sUddlmZddlZejdksJddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddlmZddlmZdd lmZd d lmZd d lmZd d lmZgdZGdddeZiZded<ejd ddZejd!ddZGdddZ Gddde Z!dS)") annotationsNwin32)AbstractEventLoopget_running_loop)Callable Generator)AbstractContextManager)TextIO)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.runzLs z%Vt100Input.__init__..)filenoioUnsupportedOperationsysmodulesisattyrrstderrwriteflushaddr_filenorrr stdin_readerr vt100_parser)rrer&fdmsgrrr__init__&s4         zVt100Input.__init__input_ready_callbackCallable[[], None]AbstractContextManager[None]cCs t||S)zj Return a context manager that makes this input active in the current event loop. )_attached_input)rr2rrrattachOs zVt100Input.attachcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )_detached_inputrrrrdetachXszVt100Input.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_keysls zVt100Input.flush_keysboolcCs|jjSr)r,closedrrrrrAzszVt100Input.closedcCt|jSr)rrr!rrrrr~zVt100Input.raw_modecCrBr)rrr!rrrrrrCzVt100Input.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 ttfy,twtf<z dVW|rM||tf<dStf=dS|rh||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_readerPermissionErrorOSErrorEOFErrorrR)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]ECHOICANONIEXTENISIGclsrlrrrra szraw_mode._patch_lflagcCs$|tjtjBtjBtjBtjB@Sr)r]IXONIXOFFICRNLINLCRIGNCRrrrrrrdszraw_mode._patch_iflagaobjectcGs@|jdurz t|jtj|jWdStjyYdSwdSr)r_r]rhr!rir`)rryrrr__exit__s zraw_mode.__exit__N)r!rDrrrVrlrDrrD)ryrzrr) rGrHrIrJr1rk classmethodrardr{rrrrrs    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. ''' rlrDrcCs|tjtjBtjBtjBBSrrmrrrrrra1szcooked_mode._patch_lflagcCs |tjBSr)r]rvrrrrrrd5s zcooked_mode._patch_iflagNr|)rGrHrIrJr}rardrrrrr(s  r)rOrrPr3rrQ)rOrrrQ)" __future__rr$platform contextlibr"r]rbasynciorrcollections.abcrrrtypingr key_bindingr baser posix_utilsrr-r__all__rrNrLcontextmanagerr5r7rrrrrrs2      q  3 R