VdZddlmZddlmZmZddlmZddlm Z dZ Gdd eZ y ) z" Completer for a regular grammar. )unicode_literals) Completer Completion)Document)_CompiledGrammar)GrammarCompleterc(eZdZdZdZdZdZdZy)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. cht|tsJt|tsJ||_||_yN) isinstancerdictcompiled_grammar completers)selfrrs Flib/third_party/prompt_toolkit/contrib/regular_languages/completion.py__init__zGrammarCompleter.__init__s3*,<===*d+++ 0$c#K|jj|j}|r-|j|j ||}|D]}|yywr )r match_prefixtext_before_cursor_remove_duplicates_get_completions_for_match)rdocumentcomplete_eventm completionscs rget_completionsz GrammarCompleter.get_completions s\  ! ! . .x/J/J K 11//>BDK!! sAAc #HK|jD] }|j}|j}|jj |}|s:|j }|j j||}t|t|} |j| |D]} |dt|| jz| jz} t|j j|| |t|jz | j | j" yw)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_nodesvarnamestartrgetvaluerunescaperlenrr"r!rescapestringr#r$) rmatchrmatch_variabler&r' completerr!unwrapped_textr completionnew_texts rrz+GrammarCompleter._get_completions_for_match*s $oo/N$,,G"((E++G4I%++"&!6!6!?!?!N$NC4GH#,";";Hn"UJ-.Ts4y:;T;T/TUXbXgXggH%!2299'8L',s5<> #V!0s A D" CD"cDg}|D]}||vs|j||S)z Remove duplicates, while keeping the order. (Sometimes we have duplicates, because the there several matches of the same grammar, each yielding similar completions.) )append)ritemsresultis rrz#GrammarCompleter._remove_duplicatesJs- A a  rN)__name__ __module__ __qualname____doc__rrrrrrr r s%>@ rr N) r< __future__rprompt_toolkit.completionrrprompt_toolkit.documentrcompilerr__all__r r=rrrCs-(;,&  DyDr