U ou f @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@sXeZdZdZdddddddZd d d d Zed ddddZddddddZdS)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. Tzdict[str, Completer | None]boolNone)options ignore_casereturncCs||_||_dSNrr)selfrrrD/tmp/pip-unpacked-wheel-8milen2s/prompt_toolkit/completion/nested.py__init__szNestedCompleter.__init__str)rcCsd|jd|jdS)NzNestedCompleter(z, ignore_case=)r)rrrr__repr__$szNestedCompleter.__repr__ NestedDict)datarcCsi}|D]n\}}t|tr(|||<q t|trB||||<q t|trf|dd|D||<q |dksrtd||<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 Gsz4NestedCompleter.from_nested_dict..N)items isinstancer dictfrom_nested_dictsetAssertionError)clsrrkeyvaluerrrr$'s      z NestedCompleter.from_nested_dictr rzIterable[Completion])documentcomplete_eventrc cs|j}t|jt|}d|kr|d}|j|}|dk r|t|d}t|t||}t||j|d} || |EdHn*t t |j |j d}|||EdHdS)N r)cursor_position)r) Ztext_before_cursorlstriplensplitrgetr r-get_completionsr listkeysr) rr*r+textZ stripped_lenZ first_termZ completerZremaining_textZ move_cursorZ new_documentrrrr2Ns$    zNestedCompleter.get_completionsN)T) __name__ __module__ __qualname____doc__rr classmethodr$r2rrrrr s &)r9 __future__rtypingrrrrrZprompt_toolkit.completionrr r Z(prompt_toolkit.completion.word_completerr Zprompt_toolkit.documentr __all__rrr rrrrs