a lc@sdZddlmZmZmZmZmZmZmZddl m Z m Z m Z ddl mZddlmZdgZeeeeeede ffZGddde ZdS) zA Nestedcompleter for completion of hierarchical data structures. )AnyDictIterableMappingOptionalSetUnion) CompleteEvent Completer Completion) WordCompleter)DocumentNestedCompleterNc@sheZdZdZdeeeefeddddZ eddd Z e e dd d d Z eeeed ddZdS)ra 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. TN)options ignore_casereturncCs||_||_dSNrr)selfrrr\/var/www/html/django/DPS/env/lib/python3.9/site-packages/prompt_toolkit/completion/nested.py__init__szNestedCompleter.__init__)rcCsd|jd|jdS)NzNestedCompleter(z, ignore_case=)r)rrrr__repr__#szNestedCompleter.__repr__)datarcCsi}|D]n\}}t|tr(|||<q t|trB||||<q t|trf|dd|D||<q |dusrJd||<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 Fz4NestedCompleter.from_nested_dict..N)items isinstancer dictfrom_nested_dictset)clsrrkeyvaluerrrr"&s      z NestedCompleter.from_nested_dict)documentcomplete_eventrc cs|j}t|jt|}d|vr|d}|j|}|dur|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_documentrrrr/Ms"   zNestedCompleter.get_completions)T)__name__ __module__ __qualname____doc__rstrrr boolrr classmethod NestedDictr"r r rr r/rrrrrs  ')r6typingrrrrrrrZprompt_toolkit.completionr r r Z(prompt_toolkit.completion.word_completerr Zprompt_toolkit.documentr __all__r7r:rrrrrs$