o j @sfdZddlmZddlmZddlmZmZmZddl m Z ddl m Z m Z dgZGd ddeZd S) z" Completer for a regular grammar. ) annotations)Iterable) CompleteEvent Completer Completion)Document)Match_CompiledGrammarGrammarCompleterc@s8eZdZdZddd ZdddZdddZdddZdS)r af Completer which can be used for autocompletion according to variables in the grammar. Each variable can have a different autocompleter. :param compiled_grammar: `GrammarCompleter` instance. :param completers: `dict` mapping variable names of the grammar to the `Completer` instances to be used for each variable. compiled_grammarr completersdict[str, Completer]returnNonecCs||_||_dSN)r r )selfr r r/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/contrib/regular_languages/completion.py__init__s zGrammarCompleter.__init__documentrcomplete_eventrIterable[Completion]ccs4|j|j}|r||||EdHdSdSr)r match_prefixtext_before_cursor_remove_duplicates_get_completions_for_match)rrrmrrrget_completions#s z GrammarCompleter.get_completionsmatchr c cs|D]M}|j}|j}|j|}|rR|j}|j||}t|t |} | | |D]%} |dt || j | j } t |j|| |t |j| j| jdVq,qdS)z Yield all the possible completions for this input string. (The completer assumes that the cursor position was at the end of the input string.) N)textstart_positiondisplay display_meta) end_nodesvarnamestartr getvaluer unescaperlenrr!r rescapestringr"r#) rrrmatch_variabler%r& completerr unwrapped_textr completionnew_textrrrr-s,     z+GrammarCompleter._get_completions_for_matchitemsccs@ddd}t}|D]}||}||vr|||Vq dS) z Remove duplicates, while keeping the order. (Sometimes we have duplicates, because the there several matches of the same grammar, each yielding similar completions.) r0rrtuple[str, int]cSs |j|jfSr)r r!)r0rrrhash_completionZs z.hash_completionN)r0rrr3)setadd)rr2r4yielded_so_farr0 hash_valuerrrrSs  z#GrammarCompleter._remove_duplicatesN)r r r rrr)rrrrrr)rr rrrr)r2rrr)__name__ __module__ __qualname____doc__rrrrrrrrr s   &N)r< __future__rcollections.abcrprompt_toolkit.completionrrrprompt_toolkit.documentrcompilerr r __all__r rrrrs