o j@sdZddlmZddlmZmZddlmZddlm Z m Z m Z ddl m Z ddlmZdgZeeeeeBd Be BfZGd dde Zd S) zA Nestedcompleter for completion of hierarchical data structures. ) annotations)IterableMapping)Any) CompleteEvent Completer Completion) WordCompleter)DocumentNestedCompleterNc@s@eZdZdZ ddd d Zdd d ZedddZdddZdS)r a Completer which wraps around several other completers, and calls any the one that corresponds with the first word of the input. By combining multiple `NestedCompleter` instances, we can achieve multiple hierarchical levels of autocompletion. This is useful when `WordCompleter` is not sufficient. If you need multiple levels, check out the `from_nested_dict` classmethod. Toptionsdict[str, Completer | None] ignore_caseboolreturnNonecCs||_||_dS)Nr r)selfr rr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/completion/nested.py__init__ s zNestedCompleter.__init__strcCsd|jd|jdS)NzNestedCompleter(z, ignore_case=)r)rrrr__repr__&szNestedCompleter.__repr__data NestedDictcCsi}|D]5\}}t|tr|||<qt|tr!||||<qt|tr1|t|||<q|dus7Jd||<q||S)a Create a `NestedCompleter`, starting from a nested dictionary data structure, like this: .. code:: data = { 'show': { 'version': None, 'interfaces': None, 'clock': None, 'ip': {'interface': {'brief'}} }, 'exit': None 'enable': None } The value should be `None` if there is no further completion at some point. If all values in the dictionary are None, it is also possible to use a set instead. Values in this data structure can be a completers as well. N)items isinstancerdictfrom_nested_dictsetfromkeys)clsrr keyvaluerrrr)s      z NestedCompleter.from_nested_dictdocumentr complete_eventrIterable[Completion]c cs|j}t|jt|}d|vrM|d}|j|}|durK|t|d}t|t||}t||j|d} || |EdHdSdSt t |j |j d}|||EdHdS)N r)cursor_position)r) text_before_cursorlstriplensplitr getr r)get_completionsr listkeysr) rr%r&text stripped_len first_term completerremaining_text move_cursor new_documentrrrr/Ps&    zNestedCompleter.get_completionsN)T)r r rrrr)rr)rrrr )r%r r&rrr') __name__ __module__ __qualname____doc__rr classmethodrr/rrrrr s  &)r< __future__rcollections.abcrrtypingrprompt_toolkit.completionrrr(prompt_toolkit.completion.word_completerr prompt_toolkit.documentr __all__rr rr rrrrs