o h1x@sddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ej dks-Je s9ddl Z dd lmZdd lmZmZmZdd lmZmZdd lmZmZmZmZmZdd lmZddlmZm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3gdZ4dZ5dZ6dZ7dZ8dZ9Gddde1Z:Gddde:Z;GdddZe d4d)d*Z?e d5d+d,Z@Gd-d.d.ZAGd/d0d0eAZBd6d2d3ZCdS)7) annotationsN)abstractmethod)get_running_loop)contextmanager)SPHINX_AUTODOC_RUNNINGwin32)windll)Arraybyrefpointer)DWORDHANDLE)CallableContextManagerIterableIteratorTextIOrun_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\/var/www/html/optinet_system/venv/lib/python3.10/site-packages/prompt_toolkit/input/win32.py__init__As z_Win32InputBase.__init__rcCdSr/r4r2r4r4r5handleDsz_Win32InputBase.handleNr-r.r-r)__name__ __module__ __qualname____doc__r6propertyrr8r4r4r4r5r,<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__r4r5r6Os   zWin32Input.__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. )r))r3rIr4r4r5attachZ zWin32Input.attachcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )r*r2r4r4r5detachaszWin32Input.detachlist[KeyPress]cCst|jSr/)listrFreadr2r4r4r5 read_keyshzWin32Input.read_keyscCr7r/r4r2r4r4r5flushkzWin32Input.flushboolcCdSNFr4r2r4r4r5closednszWin32Input.closedcCs t|jdS)N) use_win10_virtual_terminal_input)r'rDr2r4r4r5r'rszWin32Input.raw_modecCstSr/)r(r2r4r4r5r(wszWin32Input.cooked_modeintcCs tjSr/)sysr@filenor2r4r4r5r]zrMzWin32Input.filenostrcCrW)Nz win32-inputr4r2r4r4r5typeahead_hashrUzWin32Input.typeahead_hashcCs|jdSr/)rFcloser2r4r4r5r`rSzWin32Input.closercCs|jjSr/)rFr8r2r4r4r5r8szWin32Input.handler/)r@rAr-r.)rIrJr-rK)r-rK)r-rOr9r-rV)r-r[)r-r^r:)r;r<r=r>r6rLrNrRrTr?rYr'r(r]r_r`r8 __classcell__r4r4rGr5r%Js          r%c@s>eZdZdZidejdejdejdejdej 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@d8ZAd9ZBd`dad?d@ZCdbdAdBZDdcdDdEZEdddHdIZFdedOdPZGeHdfdSdTZIeHdgdVdWZJdhdZd[ZKdid]d^ZLd_S)jr&z| :param recognize_paste: When True, try to discover paste actions and turn the event into a BracketedPaste.      !"#$%&'(-.pqrstuv)wxyz{rr!r+Trecognize_pasterVr-r.cCs\d|_||_|tjrttjt |_ dSt dt j t jB|_tt|j|_ dS)NCONIN$)_fdconrr\r@isattyrr kernel32 GetStdHandlerr8osopenO_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..)r rrr8r rReadConsoleInputWr rP _get_keys_merge_paired_surrogatesr _is_pasteiter isinstancerrControlJControlMappenddatanext StopIterationrBracketedPastejoin) r3 max_countrQarrtype input_recordsall_keysgenkrr4r2r5rQsL     zConsoleInputReader.read key_pressrcCs6|jr|St|jtrt|jd}nd}t|j|S)z@ Insert KeyPress data, for vt100 compatibility. r)rrrrr"getr)r3rrr4r4r5r*s   z#ConsoleInputReader._insert_key_datarQr rArray[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)r3rQriirevr4r4r5r8s  zConsoleInputReader._get_keys key_pressesrOccsd}|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_surrogatefullcharr4r4r5rPs*   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_countrr4r4r5rqs   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_typebuttonrr4r4r5r s4  z ConsoleInputReader._handle_mouseN)T)rrVr-r.r9r-r)rrr-r)rQr rrr-r)rrOr-r)rrOr-rV)rrr-rO)rrr-rO)Mr;r<r=r>rrrControlAControlBControlCControlDControlEControlFControlGControlHControlIrControlKControlLrControlNControlOControlPControlQControlRControlSControlTControlUControlVControlWControlXControlYControlZControlBackslashControlSquareCloseControlCircumflexControlUnderscore Backspacerrrrrrrrrrr F1F2F3F4F5F6F7F8F9F10F11F12rrRIGHT_ALT_PRESSEDrrrr6r`rQrr staticmethodrrrrr4r4r4r5r&s      !%       >    r&c@s8eZdZdZdddZdddZdd d ZdddZdS)rEa 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)rr2r4r5Ms z2Vt100ConsoleInputReader.__init__..r)rrgr$ _vt100_parserr\r@rrr rrrr8rrrrrrr2r4r2r5r6Hs  z Vt100ConsoleInputReader.__init__cCrrrr2r4r4r5r`YrzVt100ConsoleInputReader.closercCsxd}td}t|}|}t|jgddsgStj|jt||t||||D]}|j |q+|j }g|_ |S)rrrr) r rrr8r rrr rrifeedrg)r3rrQrrkey_datarr4r4r5rQ^szVt100ConsoleInputReader.readrQr rr Iterator[str]ccs`t|jD]'}||}|jtvr-t|jt|j}t|tr-|jr-|j j }|dkr-|VqdS)rrN) rrrr rrrrrrr)r3rQrrrrr r4r4r5rs z!Vt100ConsoleInputReader._get_keysNr9r;)rQr rrr-rl)r;r<r=r>r6r`rQrr4r4r4r5rE;s   %rEc@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__r8rcallbackrJcsvj}|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)r8rrrt remove_eventr4r5rss  z,_Win32Handles.add_win32_handle..waitrqr9)r ValueErrorremove_win32_handlerrmrrnr)r3r8ro handle_valuer4)ror8rrrtrwrsr5add_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)r3r8eventr4r4r5rys    z!_Win32Handles.remove_win32_handleNr9)r8rrorJr-r.)r8rr-r|)r;r<r=r>r6r{ryr4r4r4r5r0s   +r0inputrorJr-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. Nrpr1r8rrxryr{)rror1r8previous_callbackr4r4r5r)s"      r)c cs`|j}|j}|jdurtd||}zdVW|r$|||dSdS|r/|||ww)Nrpr)rr1r8rr4r4r5r*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 | NonerZrVr-r.cCsttjt|_||_dSr/)rr rrrr8rZ)r3r]rZr4r4r5r6,s zraw_mode.__init__cCs,t}tj|jt|||_|dSr/)r r rGetConsoleModer8r original_mode_patch)r3rr4r4r5 __enter__2s zraw_mode.__enter__cCsDd}d}d}|jj||B|B@}|jr|tO}tj|j|dSNr+rr!)rrrZENABLE_VIRTUAL_TERMINAL_INPUTr rSetConsoleModer8)r3ENABLE_ECHO_INPUTENABLE_LINE_INPUTENABLE_PROCESSED_INPUTnew_moder4r4r5r:s zraw_mode._patchaobjectcGstj|j|jdSr/)r rrr8r)r3rr4r4r5__exit__Iszraw_mode.__exit__rX)r]rrZrVr-r.r9)rrr-r.)r;r<r=r>r6rrrr4r4r4r5r'!s  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 rrr8rr)r3rrrr4r4r5rVs zcooked_mode._patchNr9)r;r<r=r>rr4r4r4r5r(Nsr(rVc 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 rrrr rr rr)hconsolerrr4r4r5rCcs rC)rr,rorJr-r)rr,r-rra)D __future__rrr\abcrasyncior contextlibrutilsrplatformrctypesr r r r ctypes.wintypesr rtypingrrrrrprompt_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&rEr0r)r*r'r(rCr4r4r4r5sX             B2`X  -