U ou f @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 ddddZdS)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-8milen2s/prompt_toolkit/contrib/regular_languages/completion.py__init__szGrammarCompleter.__init__rrzIterable[Completion])documentcomplete_eventrccs2|j|j}|r.||||}|EdHdSr)r Z match_prefixZtext_before_cursor_remove_duplicates_get_completions_for_match)rrrmZ completionsrrrget_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_matchzlist[Completion])itemsrcCs$g}|D]}||kr||q|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)rr)resultirrrrTs  z#GrammarCompleter._remove_duplicatesN)__name__ __module__ __qualname____doc__rrrrrrrrr s   &N)r0 __future__rtypingrZprompt_toolkit.completionrrrZprompt_toolkit.documentrcompilerr r __all__r rrrrs