U cc @sbdZddlmZmZmZddlmZmZmZddl m Z ddl m Z m Z dgZGdddeZd S) z" Completer for a regular grammar. )DictIterableList) CompleteEvent Completer Completion)Document)Match_CompiledGrammarGrammarCompleterc@sneZdZdZeeeefddddZe e e e dddZ ee e e d d d Ze e ee d d dZdS)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. N)compiled_grammar completersreturncCs||_||_dSN)r r)selfr rrm/var/www/html/project/venv/lib/python3.8/site-packages/prompt_toolkit/contrib/regular_languages/completion.py__init__szGrammarCompleter.__init__)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_completions)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 completionnew_textrrrr-s&    z+GrammarCompleter._get_completions_for_match)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*resultirrrrSs  z#GrammarCompleter._remove_duplicates)__name__ __module__ __qualname____doc__r rstrrrrrrrrr rrrrrrrr s     &N)r1typingrrrZprompt_toolkit.completionrrrZprompt_toolkit.documentrcompilerr r __all__r rrrrs