U bh @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@sVeZdZdZddddddZdd d d d d Zdd d dddZd d dddZdS)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. r zdict[str, Completer]None)compiled_grammar completersreturncCs||_||_dSN)r r)selfr rrW/tmp/pip-unpacked-wheel-9s95tbmv/prompt_toolkit/contrib/regular_languages/completion.py__init__szGrammarCompleter.__init__rrzIterable[Completion])documentcomplete_eventrccs.|j|j}|r*||||EdHdSr)r Z match_prefixZtext_before_cursor_remove_duplicates_get_completions_for_match)rrrmrrrget_completions#s  z GrammarCompleter.get_completionsr )matchrrc cs|D]}|j}|j}|j|}|r|j}|j||}t|t |} | | |D]J} |dt || j | j } t |j|| |t |j| j| jdVqVqdS)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)Z end_nodesvarnamestartrgetvaluer unescaperlenrrrrescapestringrr) rrrZmatch_variabler r!Z completerrZunwrapped_textr completionZnew_textrrrr-s&    z+GrammarCompleter._get_completions_for_match)itemsrccsDddddd}t}|D]$}||}||kr|||VqdS)z Remove duplicates, while keeping the order. (Sometimes we have duplicates, because the there several matches of the same grammar, each yielding similar completions.) rztuple[str, int])r(rcSs |j|jfSr)rr)r(rrrhash_completionZsz.hash_completionN)setadd)rr)r*Zyielded_so_farr(Z hash_valuerrrrSs z#GrammarCompleter._remove_duplicatesN)__name__ __module__ __qualname____doc__rrrrrrrrr s   &N)r0 __future__rtypingrZprompt_toolkit.completionrrrZprompt_toolkit.documentrcompilerr r __all__r rrrrs