U dh)@s*UddlmZddlZejdks$tddlZddlZddlZddlZddl m Z m Z ddl m Z mZmZmZmZmZmZmZmZmZddlmZdd lmZdd lmZdd lmZd d dgZGdd d eZ iZ!de"d<ej#d dddddZ$ej#d ddddZ%Gdd d Z&Gddde&Z'dS)) annotationsNwin32)AbstractEventLoopget_running_loop) CallableContextManagerDict GeneratorListOptionalSetTextIOTupleUnion)KeyPress)Input)PosixStdinReader) Vt100Parser Vt100Inputraw_mode cooked_modec@seZdZUdZeZded<dddddZd d d d d Zd dddZ ddddZ ddddZ e ddddZ d dddZd dddZddddZd dd!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_terminalr None)stdinreturnc sz |WnHtjk rT}z(dtjkr8td|n td|W5d}~XYnX|}|}|s|tjkrd}tj ||tj tj ||_ |_ g_tj |jd_tfdd_dS)Nz idlelib.runz/tmp/pip-unpacked-wheel-4x_7prb2/prompt_toolkit/input/vt100.pyUz%Vt100Input.__init__..)filenoioUnsupportedOperationsysmodulesisattyrrstderrwriteflushaddr_filenorrr stdin_readerr vt100_parser)r"rer,fdmsgr#r!r$__init__/s0      zVt100Input.__init__Callable[[], None]zContextManager[None])input_ready_callbackrcCs t||S)zj Return a context manager that makes this input active in the current event loop. )_attached_input)r"r9r#r#r$attachXszVt100Input.attachrcCst|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.detachzlist[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"rBr#r#r$ flush_keysss zVt100Input.flush_keysboolcCs|jjSr)r2closedr!r#r#r$rFszVt100Input.closedcCst|jSr)rrr'r!r#r#r$rszVt100Input.raw_modecCst|jSr)rrr'r!r#r#r$rszVt100Input.cooked_modeintcCs |jSr)rr'r!r#r#r$r'szVt100Input.filenostrcCs d|jS)Nzfd-)r1r!r#r#r$typeahead_hashszVt100Input.typeahead_hashN)__name__ __module__ __qualname____doc__setr__annotations__r7r;r>rCrDpropertyrFrrr'rIr#r#r#r$r%s ) z>dict[tuple[AbstractEventLoop, int], Callable[[], None] | None]_current_callbacksr8zGenerator[(None, None, None)])inputcallbackrc #sttf}ddfdd }z|Wntk r\tYnXtf<z dVW5|r||tf<n tf=XdS)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. rr<csjrdS)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)rF remove_readerr#rSr5rRloopr#r$callback_wrappers z)_attached_input..callback_wrapperN)rr'rQget add_readerPermissionErrorEOFErrorrT)rRrSpreviousrWr#rUr$r:s       r:)rRrc csdt}|}t||f}|r6||dt||f<z dVW5|r^||||t||f<XdSr)rr'rQrXrTrY)rRrVr5r\r#r#r$r=s    r=c@sfeZdZdZdddddZdddd Zeddd d d Zeddd d dZdddddZ dS)rz :: with raw_mode(stdin): ''' the pseudo-terminal stdin is now used in raw mode ''' We ignore errors when executing `tcgetattr` fails. rGr)r'rcCs<||_|zt||_Wntjk r6d|_YnXdSr)r'termios tcgetattr attrs_beforeerror)r"r'r#r#r$r7s zraw_mode.__init__r<cCszzt|j}Wntjk r&YnPX||tj|tj<||tj|tj<d|tj tj <t |jtj |dS)Nr) r]r^r'r` _patch_lflagttyZLFLAG _patch_iflagZIFLAGCCVMIN tcsetattrTCSANOW)r"Znewattrr#r#r$ __enter__szraw_mode.__enter__attrsrcCs|tjtjBtjBtjB@Srr]ECHOICANONIEXTENISIGclsrjr#r#r$raszraw_mode._patch_lflagcCs$|tjtjBtjBtjBtjB@Sr)r]IXONIXOFFICRNLINLCRIGNCRrpr#r#r$rcszraw_mode._patch_iflagobject)arcGs>|jdk r:zt|jtj|jWntjk r8YnXdSr)r_r]rfr'rgr`)r"rxr#r#r$__exit__"s  zraw_mode.__exit__N) rJrKrLrMr7rh classmethodrarcryr#r#r#r$rs c@s8eZdZdZedddddZedddddZdS) 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. ''' rGricCs|tjtjBtjBtjBBSrrkrpr#r#r$ra6szcooked_mode._patch_lflagcCs |tjBSr)r]rtrpr#r#r$rc:szcooked_mode._patch_iflagN)rJrKrLrMrzrarcr#r#r#r$r-s )( __future__rr*platformAssertionError contextlibr(r]rbZasynciorrtypingrrrr r r r r rrZ key_bindingrbaserZ posix_utilsrr3r__all__rrQrOcontextmanagerr:r=rrr#r#r#r$s4 0    o 1R