o jD@sdZddlmZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZdd lmZd dlmZmZe r\ddlmZddlmZgdZGdddZ e dddZ!GdddZ"GdddZ#dS)a% An :class:`~.KeyProcessor` receives callbacks for the keystrokes parsed from the input in the :class:`~prompt_toolkit.inputstream.InputStream` instance. The `KeyProcessor` will according to the implemented keybindings call the correct callbacks when new key presses are feed through `feed`. ) annotationsN)Tasksleep)deque) Generator) TYPE_CHECKINGAny)get_app) EditingMode)vi_navigation_mode)Keys)Event)BindingKeyBindingsBase) Application)Buffer) KeyProcessorKeyPress KeyPressEventc@s0eZdZdZddd d Zdd d ZdddZdS)rz :param key: A `Keys` instance or text (one character). :param data: The received string on stdin. (Often vt100 escape codes.) Nkey Keys | strdata str | NonereturnNonecCsHt|ts t|dks J|durt|tr|j}n|}||_||_dS)Nr) isinstancer lenvaluerr)selfrrr /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/key_binding/key_processor.py__init__+s  zKeyPress.__init__strcCs|jjd|jd|jdS)Nz(key=z, data=)) __class____name__rrrr r r!__repr__7zKeyPress.__repr__otherobjectboolcCs&t|tsdS|j|jko|j|jkS)NF)rrrr)rr*r r r!__eq__:s zKeyPress.__eq__N)rrrrrrrr#)r*r+rr,)r& __module__ __qualname____doc__r"r(r-r r r r!r%s   r?_Flush)rc@seZdZdZd1ddZd2dd Zd3d dZd4ddZd5ddZd6d7ddZ d6d8ddZ d2ddZ d9dd Z d:d$d%Z d;d(d)Zd;d*d+Zd2d,d-Zd2d.d/Zd0S)z,KeyProcessor._get_matches..cSg|]}|r|qSr filterrIbr r r! z-KeyProcessor._get_matches..)tupler6get_bindings_for_keys)rrCkeysr r r! _get_matcheszszKeyProcessor._get_matchesr,cCs:tdd|D}dd|j|D}tdd|DS)z For a list of :class:`KeyPress` instances. Return True if there is any handler that is bound to a suffix of this keys. csrFr.rGrHr r r!rKrLz:KeyProcessor._is_prefix_of_longer_match..cSsh|]}|jqSr rNrPr r r! sz:KeyProcessor._is_prefix_of_longer_match..css|]}|VqdSr.r )rIfr r r!rKrL)rTr6get_bindings_starting_with_keysany)rrCrVfiltersr r r!_is_prefix_of_longer_matchs  z'KeyProcessor._is_prefix_of_longer_matchGenerator[None, KeyPress, None]c cs|j}d} d}|rd}ndV}|turd}n|||r||}|r)d}n||}dd|D}|r;|}d}|sR|rR|j|d|ddd|dd=n;|s|sd}d}tt|ddD]#} ||d| }|r|j|d|d| d|d| =d}nqb|s|dd =q) z Coroutine implementing the key match algorithm. Key strokes are sent into this generator, and it calls the appropriate handlers. FTNcSrMr )eager)rImr r r!rRrSz)KeyProcessor._process..) key_sequencerr)r>r4appendrWr] _call_handlerranger) rbufferretryflushrmatchesis_prefix_of_longer_match eager_matchesfoundir r r!r@sJ      zKeyProcessor._processF key_pressrfirstcCs$|r |j|dS|j|dS)z Add a new :class:`KeyPress` to the input queue. (Don't forget to call `process_keys` in order to process the queue.) :param first: If true, insert before everything else. N)r= appendleftrc)rrnror r r!feedszKeyProcessor.feedcCs(|r |jt|dS|j|dS)zG :param first: If true, insert before everything else. N)r= extendleftreversedextend)rrCror r r! feed_multipleszKeyProcessor.feed_multiplecstd fdd }d fdd }d}|rS|}|tu}|jtjk}|s.|s.jzj|Wnt yF  w|sP|sPj |s|s[ d Sd S) a, Process all the keys in the `input_queue`. (To be called after `feed`.) Note: because of the `feed`/`process_keys` separation, it is possible to call `feed` from inside a key binding. This function keeps looping until the queue is empty. rr,cs$jr tddjDStjS)Ncss |] }|jtjkr|VqdSr.rr CPRResponserHr r r!rKsz?KeyProcessor.process_keys..not_empty..)is_doner[r=r,r apprr r! not_emptys z,KeyProcessor.process_keys..not_emptyrcs4jrddjDd}j||SjS)NcSsg|] }|jtjkr|qSr rvrHr r r!rRz?KeyProcessor.process_keys..get_next..r)rxr=removepopleft)cprryr r!get_nexts   z+KeyProcessor.process_keys..get_nextFNrr,)rr)r r4rr rwr7firerArB Exceptionr: empty_queuer8_start_timeout)rr{ris_flushrnis_cprr ryr! process_keyss.        zKeyProcessor.process_keyscCs&t|j}|jdd|D}|S)zF Empty the input queue. Return the unprocessed input. cSsg|] }|jtjkr|qSr rvrHr r r!rR)r|z,KeyProcessor.empty_queue..)listr=clear)rrCr r r!r!s  zKeyProcessor.empty_queuehandlerrrbc Cs&t}|jj}t|jj}|jj}|j}d|_tt ||||j ||j kd}| |r2|jjddlm} z ||||Wn| yR|jYnw|rZ||||_ ||_ |r|jjrw|rw|jj} | durw| ||jjr|r|D]} |jj| j7_qdSdSdSdS)N)r?rbprevious_key_sequence is_repeatr)EditReadOnlyBuffer)r emacs_state is_recordingr,vi_staterecording_registertemporary_navigation_moder?rweakrefrefr;r< save_beforerzcurrent_buffersave_to_undo_stackprompt_toolkit.bufferrcall_fix_vi_cursor_positionoutputbell_leave_vi_temp_navigation_moderecord_in_macrocurrent_recordingrtr) rrrbrzwas_recording_emacswas_recording_viwas_temporary_navigation_moder?eventr recordingrJr r r!rd,sJ         zKeyProcessor._call_handlerrrcCsT|j}|j}|j}tr$|jjr&t|jjdkr(|jd8_||_dSdSdSdS)z After every command, make sure that if we are in Vi navigation mode, we never put the cursor after the last character of a line. (Unless it's an empty line.) rrN) rzrpreferred_columnr documentis_cursor_at_the_end_of_liner current_linecursor_position)rrrzbuffrr r r!r_s z$KeyProcessor._fix_vi_cursor_positioncCs@|j}|jtjkr|jjdur|jdurd|j_dSdSdSdS)z If we're in Vi temporary navigation (normal) mode, return to insert/replace mode after executing one action. NF)rz editing_moder VIr operator_funcr?r)rrrzr r r!rts   z+KeyProcessor._leave_vi_temp_navigation_modecsZt}|jdur dSdfdd }dfdd jr$j||_dS) a% Start auto flush timeout. Similar to Vim's `timeoutlen` option. Start a background coroutine with a timer. When this timeout expires and no key was pressed in the meantime, we flush all data in the queue and call the appropriate key binding handlers. Nrrcs,tIdHtjdkrdSdS)zWait for timeout.Nr)rrr>r  flush_keysrtimeoutr r!waits  z)KeyProcessor._start_timeout..waitcstdS)z Flush keys.N)rqr4rr r'r r!rs  z/KeyProcessor._start_timeout..flush_keysrr)r timeoutlenr9cancelcreate_background_task)rrzrr rr!rs  zKeyProcessor._start_timeoutcCs"|jttjddd|dS)zG Send SIGINT. Immediately call the SIGINT key handler. rGT)roN)rqrr SIGINTrr'r r r! send_sigints zKeyProcessor.send_sigintN)r5rrrr)rCrDrrE)rCrDrr,)rr^)F)rnrror,rr)rCrDror,rr)rrD)rrrbrDrr)rrrr)r&r0r1r2r"r:rWr]r@rqrurrrdrrrrr r r r!rGs     ; ; 3  !rc@seZdZdZd(d dZd)ddZed)ddZed*ddZed+ddZ ed,ddZ ed-ddZ ed.d d!Z d/d#d$Z ed+d%d&Zd'S)0rat Key press event, delivered to key bindings. :param key_processor_ref: Weak reference to the `KeyProcessor`. :param arg: Repetition argument. :param key_sequence: List of `KeyPress` instances. :param previouskey_sequence: Previous list of `KeyPress` instances. :param is_repeat: True when the previous event was delivered to the same handler. key_processor_ref#weakref.ReferenceType[KeyProcessor]r?rrbrDrrr,rrcCs*||_||_||_||_||_t|_dSr.)_key_processor_refrbrr_argr _app)rrr?rbrrr r r!r"s  zKeyPressEvent.__init__r#cCsd|jd|jd|jdS)NzKeyPressEvent(arg=z, key_sequence=z , is_repeat=r$)r?rbrr'r r r!r(r)zKeyPressEvent.__repr__cCs |jdjS)Nra)rbrr'r r r!rs zKeyPressEvent.datarcCs|}|dur td|S)Nz.KeyProcessor was lost. This should not happen.)rr)r processorr r r! key_processorszKeyPressEvent.key_processorApplication[Any]cC|jS)z3 The current `Application` object. )rr'r r r!rzszKeyPressEvent.apprcCs|jjS)z% The current buffer. )rzrr'r r r!rszKeyPressEvent.current_bufferintcCs0|jdkrdSt|jp d}t|dkrd}|S)z& Repetition argument. -rari@B)rr)rresultr r r!r?s  zKeyPressEvent.argcCs |jduS)zF True if repetition argument was explicitly provided. N)rr'r r r! arg_presents zKeyPressEvent.arg_presentrcCsZ|dvsJ|j}|dkr|dus|dksJ|}n |dur!|}n||}||j_dS)zb Add digit to the input argument. :param data: the typed digit as string z -0123456789rN)rrr?)rrcurrentrr r r!append_to_arg_counts   z!KeyPressEvent.append_to_arg_countcCr)zFor backward-compatibility.)rzr'r r r!cli szKeyPressEvent.cliN) rrr?rrbrDrrDrr,rrr/)rr)rr)rr)rrr)rr#rr)r&r0r1r2r"r(propertyrrrzrr?rrrr r r r!rs&        r)$r2 __future__rrasynciorr collectionsrcollections.abcrtypingrr"prompt_toolkit.application.currentr prompt_toolkit.enumsr prompt_toolkit.filters.appr prompt_toolkit.keysr prompt_toolkit.utilsr r5rrprompt_toolkit.applicationrrr__all__rr4rrr r r r!s0           d