o j{@sddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ej dks/Je s;ddlZdd lmZdd lmZmZmZdd lmZmZmZdd lmZmZdd lmZddlmZddlm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4gdZ5dZ6dZ7dZ8dZ9dZ:Gddde2Z;Gddde;ZGd#d$d$Z?e d5d*d+Z@e d6d,d-ZAGd.d/d/ZBGd0d1d1eBZCd7d3d4ZDdS)8) annotationsN)abstractmethod)get_running_loop)AbstractContextManagercontextmanager)SPHINX_AUTODOC_RUNNINGwin32)windll)CallableIterableIterator)Arraybyrefpointer)DWORDHANDLE)TextIOrun_in_executor_with_context)create_win32_eventwait_for_handles)KeyPress)Keys) MouseButtonMouseEventType) INPUT_RECORDKEY_EVENT_RECORDMOUSE_EVENT_RECORDSTD_INPUT_HANDLE EventTypes)REVERSE_ANSI_SEQUENCES)Input) Vt100Parser) Win32InputConsoleInputReaderraw_mode cooked_modeattach_win32_inputdetach_win32_inputic@s,eZdZdZd ddZeed ddZd S) _Win32InputBasez; Base class for `Win32Input` and `Win32PipeInput`. returnNonecCs t|_dSN) _Win32Handles win32_handlesselfr4/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/input/win32.py__init__Bs z_Win32InputBase.__init__rcCsdSr/r4r2r4r4r5handleEsz_Win32InputBase.handleNr-r.r-r)__name__ __module__ __qualname____doc__r6propertyrr7r4r4r4r5r,=s  r,cseZdZdZd'd(fdd Zd)d d Zd*ddZd+ddZd+ddZe d,ddZ d*ddZ d*ddZ d-ddZ d.d d!Zd/d"d#Ze d0d%d&ZZS)1r%z< `Input` class that reads from the Windows console. Nstdin TextIO | Noner-r.cs4tt|_||jrt|_dSt|_dSr/)superr6_is_win_vt100_input_enabled_use_virtual_terminal_inputVt100ConsoleInputReaderconsole_input_readerr&)r3r? __class__r4r5r6Ps   zWin32Input.__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. )r))r3rHr4r4r5attach[s zWin32Input.attachcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )r*r2r4r4r5detachdszWin32Input.detachlist[KeyPress]cCst|jSr/)listrEreadr2r4r4r5 read_keyskzWin32Input.read_keyscCs |jSr/)rE flush_keysr2r4r4r5rRns zWin32Input.flush_keysboolcCdSNFr4r2r4r4r5closedqzWin32Input.closedcCs t|jdS)N) use_win10_virtual_terminal_input)r'rCr2r4r4r5r'uszWin32Input.raw_modecCstSr/)r(r2r4r4r5r(zszWin32Input.cooked_modeintcCs tjSr/)sysr?filenor2r4r4r5r[}s zWin32Input.filenostrcCrT)Nz win32-inputr4r2r4r4r5typeahead_hashszWin32Input.typeahead_hashcCs|jdSr/)rEcloser2r4r4r5r^rQzWin32Input.closercCs|jjSr/)rEr7r2r4r4r5r7szWin32Input.handler/)r?r@r-r.)rHrIr-rJ)r-rJr-rMr-rS)r-rY)r-r\r8r9)r:r;r<r=r6rKrLrPrRr>rVr'r(r[r]r^r7 __classcell__r4r4rFr5r%Ks         r%c@sHeZdZdZidejdejdejdejdej dej dej d ej d ej d ejd ejd ejdejdejdejdejdejidejdejdejdejdejdejdejdejdejdejdej dej!dej"d ej#d!ej$d"ej%Z&id#ej'd$ej(d%ej)d&ej*d'ej+d(ej,d)ej-d*ej.d+ej/d,ej0d-ej1d.ej2d/ej3d0ej4d1ej5d2ej6d3ej7ej8ej9ej:ej;ejd6Z?d7Z@d8ZAd9ZBdbdcd?d@ZCdddAdBZDdedDdEZEdfdGdHZFdgdKdLZGdhdRdSZHeIdidUdVZJeIdjdXdYZKdkd\d]ZLdld_d`ZMdaS)mr&z| :param recognize_paste: When True, try to discover paste actions and turn the event into a BracketedPaste.      !"#$%&'(-.pqrstuv)wxyz{rr!r+Trecognize_pasterSr-r.cCs\d|_||_|tjrttjt |_ dSt dt j t jB|_tt|j|_ dS)NCONIN$)_fdconrrZr?isattyrr kernel32 GetStdHandlerr7osopenO_RDWRO_BINARYmsvcrt get_osfhandle)r3rr4r4r5r6s zConsoleInputReader.__init__cC|jdur t|jdSdSz Close fdcon.Nrrr^r2r4r4r5r^ zConsoleInputReader.closeIterable[KeyPress]c #sNd}td}t|}|}tjgddsdStjjt||t|t ||}fdd|D}t |}j r |rt |}|D]Q}g}|rt|jtra|jtjtjhvr||jzt|}Wn tyxd}Ynw|rt|jtra|jtjtjhvsa|rttjd|V|dur|VqLdS|EdHdS) Return a list of `KeyPress` instances. It won't return anything when there was nothing to read. (This function doesn't block.) http://msdn.microsoft.com/en-us/library/windows/desktop/ms684961(v=vs.85).aspx rtimeoutNcsg|]}|qSr4)_insert_key_data).0keyr2r4r5 sz+ConsoleInputReader.read..)rrrr7r rReadConsoleInputWrrN _get_keys_merge_paired_surrogatesr _is_pasteiter isinstancerrControlJControlMappenddatanext StopIterationrBracketedPastejoin) r3 max_countrOarrtype input_recordsall_keysgenkrr4r2r5rOsL     zConsoleInputReader.readrMcCsgSr/r4r2r4r4r5rR-rWzConsoleInputReader.flush_keys key_pressrcCs6|jr|St|jtrt|jd}nd}t|j|S)z@ Insert KeyPress data, for vt100 compatibility. r)rrrrr"getr)r3rrr4r4r5r1s   z#ConsoleInputReader._insert_key_datarOrrArray[INPUT_RECORD]Iterator[KeyPress]ccsvt|jD]2}||}|jtvr8t|jt|j}t|tr+|jr+| |EdHqt|t r8| |EdHqdS)R Generator that yields `KeyPress` objects from the input records. N) rangevalue EventTyper getattrEventrrKeyDown_event_to_key_pressesr _handle_mouse)r3rOriirevr4r4r5r?s  zConsoleInputReader._get_keys key_pressesccsd}|D]K}t|jt }|od|jkodkn}|o+d|jko)dkn}|rH|rC|j|jddd}t||}n|Vd}|rM|}q|Vq|rX|VdSdS)zm Combines consecutive KeyPresses with high and low surrogates into single characters Nuuuuz utf-16-le surrogatepass)rrrencodedecoder)rbuffered_high_surrogateris_textis_high_surrogateis_low_surrogatefullcharr4r4r5rWs*   z+ConsoleInputReader._merge_paired_surrogateskeyscCsJd}d}|D]}t|jts|d7}|jtjkr|d7}q|dko$|dkS)a< Return `True` when we should consider this list of keys as a paste event. Pasted text on windows will be turned into a `Keys.BracketedPaste` event. (It's not 100% correct, but it is probably the best possible way to detect pasting of text and handle that correctly.) rr!)rrrr)r text_count newline_countrr4r4r5rxs   zConsoleInputReader._is_pasterrcCst|tr|js Jd}|j}|jj}|dd}|dkr-|j|jvr,t |j|jd}n||j vrE|j |t j kr@}|rUt t j=d|gS|gSgS)zU For this `KEY_EVENT_RECORD`, return a list of `KeyPress` instances. Nzutf-8rr  )?rrrControlKeyStateuChar UnicodeCharrVirtualKeyCodekeycodesrmappingsrrLEFT_CTRL_PRESSEDRIGHT_CTRL_PRESSED SHIFT_PRESSEDLeftControlShiftLeftRightControlShiftRightUpControlShiftUpDownControlShiftDownHomeControlShiftHomeEndControlShiftEndInsertControlShiftInsertPageUpControlShiftPageUpPageDownControlShiftPageDownrr ControlLeft ControlRight ControlUp ControlDown ControlHome ControlEnd ControlInsertDelete ControlDelete ControlPageUpControlPageDownTabBackTab ShiftLeft ShiftRightShiftUp ShiftDown ShiftHomeShiftEnd ShiftInsert ShiftDelete ShiftPageUp ShiftPageDownr ControlSpaceEscapeLEFT_ALT_PRESSED)r3rresultcontrol_key_stateu_char ascii_charmapping meta_pressedr4r4r5rs           z(ConsoleInputReader._event_to_key_pressesrcCs|j}|j}d}tj}|t@r|dkrtj}ntj}n|tkr#tj }n|t kr*tj }|t @r1tj }|dur@|dkr=tj}ntj}d|j|jt|jjt|jjg}ttj|gS)zK Handle mouse events. Return a list of KeyPress instances. Nr;) EventFlags ButtonStaterNONE MOUSE_WHEELEDr SCROLL_UP SCROLL_DOWNFROM_LEFT_1ST_BUTTON_PRESSEDLEFTRIGHTMOST_BUTTON_PRESSEDRIGHT MOUSE_MOVED MOUSE_MOVE MOUSE_DOWNMOUSE_UPrrr\ MousePositionXYrrWindowsMouseEvent)r3r event_flags button_state event_typebuttonrr4r4r5rs4  z ConsoleInputReader._handle_mouseN)T)rrSr-r.r8r-rr_)rrr-r)rOrrrr-r)rrMr-r)rrMr-rS)rrr-rM)rrr-rM)Nr:r;r<r=rrrControlAControlBControlCControlDControlEControlFControlGControlHControlIrControlKControlLrControlNControlOControlPControlQControlRControlSControlTControlUControlVControlWControlXControlYControlZControlBackslashControlSquareCloseControlCircumflexControlUnderscore Backspacerrrrrrrrrrr F1F2F3F4F5F6F7F8F9F10F11F12rrRIGHT_ALT_PRESSEDrrrr6r^rOrRrr staticmethodrrrrr4r4r4r5r&s      !%       >     r&c@sBeZdZdZdddZdddZdd d Zdd d ZdddZdS)rDa Similar to `ConsoleInputReader`, but for usage when `ENABLE_VIRTUAL_TERMINAL_INPUT` is enabled. This assumes that Windows sends us the right vt100 escape sequences and we parse those with our vt100 parser. (Using this instead of `ConsoleInputReader` results in the "data" attribute from the `KeyPress` instances to be more correct in edge cases, because this responds to for instance the terminal being in application cursor keys mode.) r-r.csnd_g_tfdd_tjr!ttj t _ dSt dt jt jB_ttj_ dS)Ncs j|Sr/)_bufferr)rr2r4r5Ts z2Vt100ConsoleInputReader.__init__..r)rrfr$ _vt100_parserrZr?rrr rrrr7rrrrrrr2r4r2r5r6Os  z Vt100ConsoleInputReader.__init__cCrrrr2r4r4r5r^`rzVt100ConsoleInputReader.closercCsxd}td}t|}|}t|jgddsgStj|jt||t||||D]}|j |q+|j }g|_ |S)rrrr) rrrr7r rrrrrhfeedrf)r3rrOrrkey_datarr4r4r5rOeszVt100ConsoleInputReader.readrMcCs|j|j}g|_|S)zc Flush pending keys and return them. (Used for flushing the 'escape' key.) )rhflushrf)r3rr4r4r5rRs z"Vt100ConsoleInputReader.flush_keysrOrrr Iterator[str]ccs`t|jD]'}||}|jtvr-t|jt|j}t|tr-|jr-|j j }|dkr-|VqdS)rrN) rrrr rrrrrrr)r3rOrrrrrr4r4r5rs z!Vt100ConsoleInputReader._get_keysNr8r:r_)rOrrrr-rl) r:r;r<r=r6r^rOrRrr4r4r4r5rDBs   %rDc@s.eZdZdZdddZdd d Zdd dZdS)r0a Utility to keep track of which handles are connectod to which callbacks. `add_win32_handle` starts a tiny event loop in another thread which waits for the Win32 handle to become ready. When this happens, the callback will be called in the current asyncio event loop using `call_soon_threadsafe`. `remove_win32_handle` will stop this tiny event loop. NOTE: We use this technique, so that we don't have to use the `ProactorEventLoop` on Windows and we can wait for things like stdin in a `SelectorEventLoop`. This is important, because our inputhook mechanism (used by IPython), only works with the `SelectorEventLoop`. r-r.cCsi|_i|_dSr/)_handle_callbacks_remove_eventsr2r4r4r5r6s z_Win32Handles.__init__r7rcallbackrIcsvj}|dur td|t|j|<t|j|<d fdd d fdd td dS) z7 Add a Win32 handle to the event loop. NInvalid handle.r-r.c s(z WtddStdw)Nlooprr4)rorrwaitr4r5readysz-_Win32Handles.add_win32_handle..readycs2tg}|urtjdSdSr/)rr r CloseHandlecall_soon_threadsafe)r)r7rrrt remove_eventr4r5rss  z,_Win32Handles.add_win32_handle..waitrqr8)r ValueErrorremove_win32_handlerrmrrnr)r3r7ro handle_valuer4)ror7rrrtrwrsr5add_win32_handles     z_Win32Handles.add_win32_handleCallable[[], None] | NonecCsf|jdurdSz |j|j}Wn tyYnwtj|z|j|jWSty2YdSw)zt Remove a Win32 handle from the event loop. Return either the registered handler or `None`. N)rrnpopKeyErrorr rSetEventrm)r3r7eventr4r4r5rys    z!_Win32Handles.remove_win32_handleNr8)r7rrorIr-r.)r7rr-r|)r:r;r<r=r6r{ryr4r4r4r5r0s   +r0inputrorIr-Iterator[None]c cs|j}|j}|jdurtd||}|||zdVW|||r/|||dSdS|||r?|||ww)z Context manager that makes this input active in the current event loop. :param input: :class:`~prompt_toolkit.input.Input` object. :param input_ready_callback: Called when the input is ready to read. Nrpr1r7rrxryr{)rror1r7previous_callbackr4r4r5r)s"      r)c cs`|j}|j}|jdurtd||}zdVW|r$|||dSdS|r/|||ww)Nrpr)rr1r7rr4r4r5r*%s  r*c@s<eZdZdZ ddd d Zdd d ZdddZdddZdS)r'z :: with raw_mode(stdin): ''' the windows terminal is now in 'raw' mode. ''' The ``fileno`` attribute is ignored. This is to be compatible with the `raw_input` method of `.vt100_input`. NFr[ int | NonerXrSr-r.cCsttjt|_||_dSr/)rr rrrr7rX)r3r[rXr4r4r5r6As zraw_mode.__init__cCs,t}tj|jt|||_|dSr/)rr rGetConsoleModer7r original_mode_patch)r3rr4r4r5 __enter__Gs zraw_mode.__enter__cCsDd}d}d}|jj||B|B@}|jr|tO}tj|j|dSNr+rr!)rrrXENABLE_VIRTUAL_TERMINAL_INPUTr rSetConsoleModer7)r3ENABLE_ECHO_INPUTENABLE_LINE_INPUTENABLE_PROCESSED_INPUTnew_moder4r4r5rOs zraw_mode._patchaobjectcGstj|j|jdSr/)r rrr7r)r3rr4r4r5__exit__^szraw_mode.__exit__rU)r[rrXrSr-r.r8)rrr-r.)r:r;r<r=r6rrrr4r4r4r5r'6s  r'c@seZdZdZdddZdS)r(zx :: with cooked_mode(stdin): ''' The pseudo-terminal stdin is now used in cooked mode. ''' r-r.cCs0d}d}d}tj|j|jj||B|BBdSr)r rrr7rr)r3rrrr4r4r5rks zcooked_mode._patchNr8)r:r;r<r=rr4r4r4r5r(csr(rSc Csfttjt}td}tj|t|ztj|tt }|dkWtj||Stj||w)z_ Returns True when we're running Windows and VT100 escape sequences are supported. rr!) rr rrrrrrrr)hconsolerrr4r4r5rBxs rB)rr,rorIr-r)rr,r-rr`)E __future__rrrZabcrasyncior contextlibrrutilsrplatformrctypesr collections.abcr r r rrrctypes.wintypesrrtypingrprompt_toolkit.eventlooprprompt_toolkit.eventloop.win32rr(prompt_toolkit.key_binding.key_processorrprompt_toolkit.keysrprompt_toolkit.mouse_eventsrrprompt_toolkit.win32_typesrrrrr ansi_escape_sequencesr"baser# vt100_parserr$__all__r*r,r.r'rr,r%r&rDr0r)r*r'r(rBr4r4r4r5sZ             D6nX  -