o Df@sdZddlmZddlmZmZmZmZmZddl m Z m Z m Z ddl mZddlmZdgZeeeeeede ffZGd dde ZdS) zA Nestedcompleter for completion of hierarchical data structures. ) annotations)AnyIterableMappingSetUnion) 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||_||_dSNrr)selfrrrY/home/ubuntu/webapp/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]7\}}t|tr|||<qt|tr!||||<qt|tr3|dd|D||<q|dus9Jd||<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. cSsi|]}|dqSrr).0itemrrr Hsz4NestedCompleter.from_nested_dict..N)items isinstancer dictfrom_nested_dictset)clsrrkeyvaluerrrr%(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_cursorlstriplensplitrgetr r.get_completionsr listkeysr) rr*r+text stripped_len first_term completerremaining_text move_cursor new_documentrrrr4Os&    zNestedCompleter.get_completionsN)T)rrrrrr)rr)rrrr )r*r r+rrr,) __name__ __module__ __qualname____doc__rr classmethodr%r4rrrrr s  &)rA __future__rtypingrrrrrprompt_toolkit.completionrr r (prompt_toolkit.completion.word_completerr prompt_toolkit.documentr __all__rrr rrrrs