o Chz@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_handlesselfr4l/var/www/html/myvaluetrips/my_value_trip_new/venv/lib/python3.10/site-packages/prompt_toolkit/input/win32.py__init__As z_Win32InputBase.__init__rcCsdSr/r4r2r4r4r5handleDsz_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__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))r3rHr4r4r5attachZ 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/)listrEreadr2r4r4r5 read_keyshzWin32Input.read_keyscCs |jSr/)rE flush_keysr2r4r4r5rSks zWin32Input.flush_keysboolcCdSNFr4r2r4r4r5closednzWin32Input.closedcCs t|jdS)N) use_win10_virtual_terminal_input)r'rCr2r4r4r5r'rszWin32Input.raw_modecCstSr/)r(r2r4r4r5r(wszWin32Input.cooked_modeintcCs tjSr/)sysr?filenor2r4r4r5r\zrLzWin32Input.filenostrcCrU)Nz win32-inputr4r2r4r4r5typeahead_hashszWin32Input.typeahead_hashcCs|jdSr/)rEcloser2r4r4r5r_rRzWin32Input.closercCs|jjSr/)rEr7r2r4r4r5r7szWin32Input.handler/)r?r@r-r.)rHrIr-rJ)r-rJr-rNr-rT)r-rZ)r-r]r8r9)r:r;r<r=r6rKrMrQrSr>rWr'r(r\r^r_r7 __classcell__r4r4rFr5r%Js          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_pasterTr-r.cCs\d|_||_|tjrttjt |_ dSt dt j t jB|_tt|j|_ dS)NCONIN$)_fdconrr[r?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..)r rrr7r rReadConsoleInputWr rO _get_keys_merge_paired_surrogatesr _is_pasteiter isinstancerrControlJControlMappenddatanext StopIterationrBracketedPastejoin) r3 max_countrParrtype input_recordsall_keysgenkrr4r2r5rPsL     zConsoleInputReader.readrNcCsgSr/r4r2r4r4r5rS*rXzConsoleInputReader.flush_keys 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_datarPr 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)r3rPriirevr4r4r5r<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_surrogatefullcharr4r4r5rTs*   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_countrr4r4r5rus   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)rrTr-r.r8r-rr`)rrr-r)rPr rrr-r)rrNr-r)rrNr-rT)rrr-rN)rrr-rN)Nr:r;r<r=rrrControlAControlBControlCControlDControlEControlFControlGControlHControlIrControlKControlLrControlNControlOControlPControlQControlRControlSControlTControlUControlVControlWControlXControlYControlZControlBackslashControlSquareCloseControlCircumflexControlUnderscore Backspacerrrrrrrrrrr F1F2F3F4F5F6F7F8F9F10F11F12rrRIGHT_ALT_PRESSEDrrrr6r_rPrSrr 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)rr2r4r5Qs z2Vt100ConsoleInputReader.__init__..r)rrgr$ _vt100_parserr[r?rrr rrrr7rrrrrrr2r4r2r5r6Ls  z Vt100ConsoleInputReader.__init__cCrrrr2r4r4r5r_]rzVt100ConsoleInputReader.closercCsxd}td}t|}|}t|jgddsgStj|jt||t||||D]}|j |q+|j }g|_ |S)rrrr) r rrr7r rrr rrifeedrg)r3rrPrrkey_datarr4r4r5rPbszVt100ConsoleInputReader.readrNcCs|j|j}g|_|S)zc Flush pending keys and return them. (Used for flushing the 'escape' key.) )riflushrg)r3rr4r4r5rSs z"Vt100ConsoleInputReader.flush_keysrPr rr Iterator[str]ccs`t|jD]'}||}|jtvr-t|jt|j}t|tr-|jr-|j j }|dkr-|VqdS)rrN) rrrr rrrrrrr)r3rPrrrrr r4r4r5rs z!Vt100ConsoleInputReader._get_keysNr8r;r`)rPr rrr-rm) r:r;r<r=r6r_rPrSrr4r4r4r5rD?s   %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)rprswaitr4r5readysz-_Win32Handles.add_win32_handle..readycs2tg}|urtjdSdSr/)rr r CloseHandlecall_soon_threadsafe)r)r7rsru remove_eventr4r5rts  z,_Win32Handles.add_win32_handle..waitrrr8)r ValueErrorremove_win32_handlerrnrror)r3r7rp handle_valuer4)rpr7rsrurxrtr5add_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)rropopKeyErrorr rSetEventrn)r3r7eventr4r4r5rzs    z!_Win32Handles.remove_win32_handleNr8)r7rrprIr-r.)r7rr-r})r:r;r<r=r6r|rzr4r4r4r5r0s   +r0inputrprIr-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. Nrqr1r7rryrzr|)rrpr1r7previous_callbackr4r4r5r)s"      r)c cs`|j}|j}|jdurtd||}zdVW|r$|||dSdS|r/|||ww)Nrqr)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 | NonerYrTr-r.cCsttjt|_||_dSr/)rr rrrr7rY)r3r\rYr4r4r5r6>s zraw_mode.__init__cCs,t}tj|jt|||_|dSr/)r r rGetConsoleModer7r original_mode_patch)r3rr4r4r5 __enter__Ds zraw_mode.__enter__cCsDd}d}d}|jj||B|B@}|jr|tO}tj|j|dSNr+rr!)rrrYENABLE_VIRTUAL_TERMINAL_INPUTr rSetConsoleModer7)r3ENABLE_ECHO_INPUTENABLE_LINE_INPUTENABLE_PROCESSED_INPUTnew_moder4r4r5rLs zraw_mode._patchaobjectcGstj|j|jdSr/)r rrr7r)r3rr4r4r5__exit__[szraw_mode.__exit__rV)r\rrYrTr-r.r8)rrr-r.)r:r;r<r=r6rrrr4r4r4r5r'3s  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)r3rrrr4r4r5rhs zcooked_mode._patchNr8)r:r;r<r=rr4r4r4r5r(`sr(rTc 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)hconsolerrr4r4r5rBus rB)rr,rprIr-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&rDr0r)r*r'r(rBr4r4r4r5sX             B6nX  -