o ˷e @sbdZddlmZmZmZddlmZmZmZddl m Z ddl m Z m Z dgZGdddeZd S) z" Completer for a regular grammar. )DictIterableList) CompleteEvent Completer Completion)Document)Match_CompiledGrammarGrammarCompleterc@s|eZdZdZdedeeefddfddZde d e de e fd d Z d ed e de e fd dZde e dee fddZdS)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_grammar completersreturnNcCs||_||_dSN)r r)selfr rrj/var/www/ideatree/venv/lib/python3.10/site-packages/prompt_toolkit/contrib/regular_languages/completion.py__init__s zGrammarCompleter.__init__documentcomplete_eventccs8|j|j}|r||||}|EdHdSdSr)r match_prefixtext_before_cursor_remove_duplicates_get_completions_for_match)rrrm completionsrrrget_completions!s z GrammarCompleter.get_completionsmatchc 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_nodesvarnamestartrgetvaluer unescaperlenrr rrescapestringr!r") rrrmatch_variabler$r% completerrunwrapped_textr completionnew_textrrrr-s,     z+GrammarCompleter._get_completions_for_matchitemscCs$g}|D] }||vr||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)rr1resultirrrrSs  z#GrammarCompleter._remove_duplicates)__name__ __module__ __qualname____doc__r rstrrrrrrrrr rrrrrrrr s0      &N)r8typingrrrprompt_toolkit.completionrrrprompt_toolkit.documentrcompilerr r __all__r rrrrs