)L i d Z ddlmZ ddlmZmZ ddlmZmZ ddl m
Z
ddl m Z ddl
mZmZ erdd lmZ g d
Z G d d Z G d
de Z G d de Z G d de Z G d de Z G d de Z G d de Zy)a)
`Fish-style `_ like auto-suggestion.
While a user types input in a certain buffer, suggestions are generated
(asynchronously.) Usually, they are displayed after the input. When the cursor
presses the right arrow and the cursor is at the end of the input, the
suggestion will be inserted.
If you want the auto suggestions to be asynchronous (in a background thread),
because they take too much time, and could potentially block the event loop,
then wrap the :class:`.AutoSuggest` instance into a
:class:`.ThreadedAutoSuggest`.
) annotations)ABCMetaabstractmethod)
TYPE_CHECKINGCallable run_in_executor_with_context )Document)Filter to_filter)Buffer)
Suggestion AutoSuggestThreadedAutoSuggestDummyAutoSuggestAutoSuggestFromHistoryConditionalAutoSuggestDynamicAutoSuggestc e Zd ZdZddZddZy)r zb
Suggestion returned by an auto-suggest algorithm.
:param text: The suggestion text.
c || _ y Ntext)selfr s a/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/prompt_toolkit/auto_suggest.py__init__zSuggestion.__init__. s c " d| j dS )Nz Suggestion()r )r s r __repr__zSuggestion.__repr__1 s TYYKq))r N)r strreturnNone)r# r" )__name__
__module__ __qualname____doc__r r! r r r r ' s *r r c 6 e Zd ZdZedd Z ddZy)r z9
Base class for auto suggestion implementations.
c y)a
Return `None` or a :class:`.Suggestion` instance.
We receive both :class:`~prompt_toolkit.buffer.Buffer` and
:class:`~prompt_toolkit.document.Document`. The reason is that auto
suggestions are retrieved asynchronously. (Like completions.) The
buffer text could be changed in the meantime, but ``document`` contains
the buffer document like it was at the start of the auto suggestion
call. So, from here, don't access ``buffer.text``, but use
``document.text`` instead.
:param buffer: The :class:`~prompt_toolkit.buffer.Buffer` instance.
:param document: The :class:`~prompt_toolkit.document.Document` instance.
Nr) r bufferdocuments r get_suggestionzAutoSuggest.get_suggestion: s r c . K | j || S w)z
Return a :class:`.Future` which is set when the suggestions are ready.
This function can be overloaded in order to provide an asynchronous
implementation.
r/ r buffr. s r get_suggestion_asyncz AutoSuggest.get_suggestion_asyncK s ""422s Nr- r r. r r# Suggestion | Noner3 r r. r r# r6 )r% r& r' r( r r/ r4 r) r r r r 5 s8 33&.3 3r r ) metaclassc 4 e Zd ZdZddZddZ ddZy)r z
Wrapper that runs auto suggestions in a thread.
(Use this to prevent the user interface from becoming unresponsive if the
generation of suggestions takes too much time.)
c || _ y r ) auto_suggest)r r; s r r z ThreadedAutoSuggest.__init__] s
(r c : | j j || S r )r; r/ r2 s r r/ z"ThreadedAutoSuggest.get_suggestion` s //h??r c J K d fd }t | d{ S 7 w)z@
Run the `get_suggestion` function in a thread.
c ( j S r r1 )r3 r. r s r run_get_suggestion_threadzKThreadedAutoSuggest.get_suggestion_async..run_get_suggestion_threadj s &&tX6 6r N)r# r6 r )r r3 r. r? s ``` r r4 z(ThreadedAutoSuggest.get_suggestion_asyncc s 7 22KLLLLs #!#N)r; r r# r$ r7 r% r& r' r( r r/ r4 r) r r r r V s3 )@
M
M&.
M
Mr r c e Zd ZdZddZy)r z?
AutoSuggest class that doesn't return any suggestion.
c y r r) r, s r r/ z DummyAutoSuggest.get_suggestionu s r Nr5 r% r& r' r( r/ r) r r r r p s r r c e Zd ZdZddZy)r z=
Give suggestions based on the lines in the history.
c \ |j }|j j dd d }|j rqt t |j
D ]L }t |j D ]. }|j | st |t | d c c S N y )N
r
) historyr rsplitstripreversedlist get_strings
splitlines
startswithr len)r r- r. rH r stringlines r r/ z%AutoSuggestFromHistory.get_suggestion~ s .. }}##D!,R0 ::<"4(;(;(=#>?
= $V%6%6%8 9 =Dt, )$s4y{*; <<=
=
r Nr5 rC r) r r r r y s r r c e Zd ZdZddZddZy)r zU
Auto suggest that can be turned on and of according to a certain condition.
c 2 || _ t | | _ y r )r; r
filter)r r; rU s r r z ConditionalAutoSuggest.__init__ s ( ' r c \ | j r | j j || S y r )rU r; r/ r, s r r/ z%ConditionalAutoSuggest.get_suggestion s' ;;=$$33FHE Er N)r; r rU z
bool | Filterr# r$ r5 )r% r& r' r( r r/ r) r r r r s (r r c 4 e Zd ZdZddZddZ ddZy)r z
Validator class that can dynamically returns any Validator.
:param get_validator: Callable that returns a :class:`.Validator` instance.
c || _ y r )get_auto_suggest)r rY s r r zDynamicAutoSuggest.__init__ s
0 r c ^ | j xs
t }|j || S r )rY r r/ r r3 r. r; s r r/ z!DynamicAutoSuggest.get_suggestion s, ,,.D2B2D **4::r c z K | j xs
t }|j || d { S 7 wr )rY r r4 r[ s r r4 z'DynamicAutoSuggest.get_suggestion_async s8 ,,.D2B2D !66tXFFFFs 2;9;N)rY z Callable[[], AutoSuggest | None]r# r$ r7 r@ r) r r r r s2 1;GG&.G Gr r N)r(
__future__r abcr r typingr r prompt_toolkit.eventloopr r. r filtersr r
r- r __all__r r r r r r r r) r r rc s # ' * A & * * 3G 3BM+ M4{ [ ,
[
G Gr