o i&@sddlZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd l m Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$Gddde Z%deddddddej&f de'de'de'ded ee d!eed"e(d#eegee'fd$eee'ge(fd%ed&ed'e"fd(d)Z)dS)*N)Any)Callable)Iterable)List)Optional)Tuple) CompleteEvent) Completion) PathCompleter) Completer)Document) KeyBindings) KeyPressEvent)Keys) SimpleLexer) CompleteStyle) PromptSession)Style)DEFAULT_QUESTION_PREFIX)build_validator)Question)merge_styles_defaultc seZdZdZ     ddedeegeefdeeegefde d ed df fd d Z d e de d e effdd ZZS)GreatUXPathCompleterzWraps :class:`prompt_toolkit.completion.PathCompleter`. Makes sure completions for directories end with a path separator. Also make sure the right path separator is used. Checks if `get_paths` returns list of existing directories. FNronly_directories get_paths file_filter min_input_len expanduserreturncsN|pdd}|D]}tj|std|dq tj|||||ddS)a*Adds validation of 'get_paths' to :class:`prompt_toolkit.completion.PathCompleter`. Args: only_directories (bool): If True, only directories will be returned, but no files. Defaults to False. get_paths (Callable[[], List[str]], optional): Callable which returns a list of directories to look into when the user enters a relative path. If None, set to (lambda: ["."]). Defaults to None. file_filter (Callable[[str], bool], optional): Callable which takes a filename and returns whether this file should show up in the completion. ``None`` when no filtering has to be done. Defaults to None. min_input_len (int): Don't do autocompletion when the input string is shorter. Defaults to 0. expanduser (bool): If True, tilde (~) is expanded. Defaults to False. Raises: ValueError: If any of the by `get_paths` returned directories does not exist. cSsdgS)N.r r r L/home/ubuntu/.local/lib/python3.10/site-packages/questionary/prompts/path.pyAsz/GreatUXPathCompleter.__init__..zS Completer for file paths 'get_paths' must return only existing directories, but 'z' does not exist.)rrrrrN)ospathisdir ValueErrorsuper__init__)selfrrrrr current_path __class__r r!r($s"    zGreatUXPathCompleter.__init__documentcomplete_eventc#s~tt|||}|D]0}|jd}|dddkr9|dddtjj}|d|f|jd<|jtjj7_|Vq dS)zGet completions. Wraps :class:`prompt_toolkit.completion.PathCompleter`. Makes sure completions for directories end with a path separator. Also make sure the right path separator is used. r/N)r'rget_completionsdisplayr#r$septext)r)r-r. completions completionstyled_display display_textr+r r!r2Ts  z$GreatUXPathCompleter.get_completions)FNNrF)__name__ __module__ __qualname____doc__boolrrrstrintr(r rrr r2 __classcell__r r r+r!rs6 0rFmessagedefaultqmarkvalidate completerstylerrrcomplete_stylekwargsrc  st|g} dttttfffdd } t|} |p"t|||dd}t}|jtj dddt fdd }|jt j j dddt fd d }t| ftd | || | |d | }|jt|t|jS)a A text input for a file or directory path with autocompletion enabled. Example: >>> import questionary >>> questionary.path( >>> "What's the path to the projects version file?" >>> ).ask() ? What's the path to the projects version file? ./pyproject.toml './pyproject.toml' .. image:: ../images/path.gif This is just a really basic example, the prompt can be customized using the parameters. Args: message: Question text. default: Default return value (single value). qmark: Question prefix displayed in front of the question. By default this is a ``?``. complete_style: How autocomplete menu would be shown, it could be ``COLUMN`` ``MULTI_COLUMN`` or ``READLINE_LIKE`` from :class:`prompt_toolkit.shortcuts.CompleteStyle`. validate: Require the entered value to pass a validation. The value can not be submitted until the validator accepts it (e.g. to check minimum password length). This can either be a function accepting the input and returning a boolean, or an class reference to a subclass of the prompt toolkit Validator class. completer: A custom completer to use in the prompt. For more information, see `this `_. style: A custom color and style for the question parts. You can configure colors as well as font types for different elements. only_directories: Only show directories in auto completion. This option does not do anything if a custom ``completer`` is passed. get_paths: Set a callable to generate paths to traverse for suggestions. This option does not do anything if a custom ``completer`` is passed. file_filter: Optional callable to filter suggested paths. Only paths where the passed callable evaluates to ``True`` will show up in the suggested paths. This does not validate the typed path, e.g. it is still possible for the user to enter a path manually, even though this filter evaluates to ``False``. If in addition to filtering suggestions you also want to validate the result, use ``validate`` in combination with the ``file_filter``. Returns: :class:`Question`: Question instance, ready to be prompted (using ``.ask()``). rcsdfddfgS)Nz class:qmarkzclass:questionz {} )formatr rCrEr r!get_prompt_tokensszpath..get_prompt_tokensT)rrrr)eagereventcSsp|jjdur d|j_dS|jjjddr6|jjjj}|tjj r'|dd}|jj |d|jj dSdS)NT) set_cursorr0)result) current_buffercomplete_stateapprFr-r5endswithr#r$r4exitappend_to_history)rO result_pathr r r! set_answers    zpath..set_answercSsH|jj}|jr d|_|jj}|tjjs| tjj|j dddS)NF) select_first) rTrRrSr-r5rUr#r$r4 insert_textstart_completion)rObr*r r r! next_segmentszpath..next_segmentz class:answer)lexerrHrG validatorrI key_bindings)rrrr?rrr addrControlMrr#r$r4rrdefault_bufferresetr rrT)rCrDrErFrGrHrrrrIrJ merged_stylerMr`bindingsrYr^pr rLr!r$ts8 I    r$)*r#typingrrrrrrprompt_toolkit.completionrr r prompt_toolkit.completion.baser prompt_toolkit.documentr prompt_toolkit.key_bindingr (prompt_toolkit.key_binding.key_processorrprompt_toolkit.keysrprompt_toolkit.lexersrprompt_toolkit.shortcuts.promptrrprompt_toolkit.stylesrquestionary.constantsrquestionary.prompts.commonrquestionary.questionrquestionary.stylesrr MULTI_COLUMNr?r>r$r r r r!st                      Z