o ip*#@sddlZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd l mZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZdeeddddddddddf dedeeeeeeeffdeeeeeeeffdedeedeededededed ed!ed"ed#ed$eed%ed&ef"d'd(Z dS))N)Any)Dict)Optional)Sequence)Union) Application) KeyBindings)Keys)Style)utils)DEFAULT_QUESTION_PREFIX)DEFAULT_SELECTED_POINTER)common)Choice)InquirerControl) Separator)Question)merge_styles_defaultFTmessagechoicesdefaultqmarkpointerstyle use_shortcutsuse_arrow_keys use_indicator use_jk_keysuse_emacs_keysuse_search_filter show_selectedshow_description instructionkwargsreturnc ss s | s | s td| rrtdr%| r%tdd|Dr%td|dus/t|dkr3tdrPtd d|D}|ttjkrPtd ttj|t|g}t||||| | |d fd d }tj |fi|}t j t j ddj t jdddd}rtjD]1\}}|jdur|jsstd|jt|ts|jdus|jrqfdd}|||jqfdd}fdd}rfdd}tjD] }j |dd|q߈j t jdd|r j t jdd|j t jdd|| r!j ddd|j ddd|| r8j t jdd|j t jdd|j t jddfdd} t jdd }t t!d"||d!t"#|t!j$S)#uA list of items to select **one** option from. The user can pick one option and confirm it (if you want to allow the user to select multiple options, use :meth:`questionary.checkbox` instead). Example: >>> import questionary >>> questionary.select( ... "What do you want to do?", ... choices=[ ... "Order a pizza", ... "Make a reservation", ... "Ask for opening hours" ... ]).ask() ? What do you want to do? Order a pizza 'Order a pizza' .. image:: ../images/select.gif This is just a really basic example, the prompt can be customised using the parameters. Args: message: Question text choices: Items shown in the selection, this can contain :class:`Choice` or or :class:`Separator` objects or simple items as strings. Passing :class:`Choice` objects, allows you to configure the item more (e.g. preselecting it or disabling it). default: A value corresponding to a selectable item in the choices, to initially set the pointer position to. qmark: Question prefix displayed in front of the question. By default this is a ``?``. pointer: Pointer symbol in front of the currently highlighted element. By default this is a ``»``. Use ``None`` to disable it. instruction: A hint on how to navigate the menu. It's ``(Use shortcuts)`` if only ``use_shortcuts`` is set to True, ``(Use arrow keys or shortcuts)`` if ``use_arrow_keys`` & ``use_shortcuts`` are set and ``(Use arrow keys)`` by default. style: A custom color and style for the question parts. You can configure colors as well as font types for different elements. use_indicator: Flag to enable the small indicator in front of the list highlighting the current location of the selection cursor. use_shortcuts: Allow the user to select items from the list using shortcuts. The shortcuts will be displayed in front of the list items. Arrow keys, j/k keys and shortcuts are not mutually exclusive. use_arrow_keys: Allow the user to select items from the list using arrow keys. Arrow keys, j/k keys and shortcuts are not mutually exclusive. use_jk_keys: Allow the user to select items from the list using `j` (down) and `k` (up) keys. Arrow keys, j/k keys and shortcuts are not mutually exclusive. use_emacs_keys: Allow the user to select items from the list using `Ctrl+N` (down) and `Ctrl+P` (up) keys. Arrow keys, j/k keys, emacs keys and shortcuts are not mutually exclusive. use_search_filter: Flag to enable search filtering. Typing some string will filter the choices to keep only the ones that contain the search string. Note that activating this option disables "vi-like" navigation as "j" and "k" can be part of a prefix and therefore cannot be used for navigation show_selected: Display current selection choice at the bottom of list. show_description: Display description of current selection if available. Returns: :class:`Question`: Question instance, ready to be prompted (using ``.ask()``). z^Some option to move the selection is required. Arrow keys, j/k keys, emacs keys, or shortcuts.zSCannot use j/k keys with prefix filter search, since j/k can be part of the prefix.css |] }t|dddvVqdS) shortcut_key)jkN)getattr.0cr-N/home/ubuntu/.local/lib/python3.10/site-packages/questionary/prompts/select.py szselect..zqA choice is trying to register j/k as a shortcut key when they are in use as arrow keys disable one or the other.Nrz'A list of choices needs to be provided.css|] }t|tsdVqdS)N) isinstancerr*r-r-r.r/szA list with shortcuts supports a maximum of {} choices as this is the maximum number of keyboard shortcuts that are available. You provided {} choices!)rrrr r!rinitial_choicecsdfddfg}jr6tjtr*|ddddjDf|S|djf|SrA|df|SrPrPd rJd ndd }nr_s_d rYd ndd }n d rdd ndd }|d|f|S)Nz class:qmarkzclass:questionz {} z class:answerr&cSsg|]}|dqS)r0r-)r+tokenr-r-r. sz5select..get_prompt_tokens..zclass:instructionz(Use shortcuts or arrow keysz, type to filter)z(Use shortcutsz(Use arrow keys)format is_answeredr1get_pointed_attitlelistappendjoin)tokensinstruction_msg)icr"rrrrrr-r.get_prompt_tokenss*  z!select..get_prompt_tokensTeagercSs|jjtdddS)Nzclass:aborting) exceptionr)appexitKeyboardInterrupteventr-r-r._szselect.._zc{} does not have a shortcut and arrow keys for movement are disabled. This choice is not reachable.cs j|ddfdd}dS)NTrAcs _dSN) pointed_atrG)ir?r-r. select_choices z3select.._reg_binding..select_choice)add)rLkeysrM)bindingsr?)rLr. _reg_bindings zselect.._reg_bindingc(srdSdSrJ) select_nextis_selection_validrGr?r-r.move_cursor_downz select..move_cursor_downcrRrJ)select_previousrTrGrUr-r.move_cursor_uprWzselect..move_cursor_upcs|jdjdS)Nr)add_search_character key_sequencekeyrGrUr-r. search_filterszselect..search_filterr'r(csd_|jjjddS)NT)result)r7rDrEr8valuerGrUr-r. set_answer szselect..set_answercSsdS)zDisallow inserting other text.Nr-rGr-r-r.otherszselect..other)layout key_bindingsrr-)% ValueErroranylensumr SHORTCUT_KEYSr6rrcreate_inquirer_layoutrrNr ControlQControlC enumeraterr%disabled RuntimeErrorr9r1rstring printable BackspaceDownUpControlNControlPControlMrrrr used_kwargs__init__)rrrrrrrrrrrrr r!r"r#real_len_of_choices merged_styler@rbrIrLr,rQrVrYr] characterr`rar-)rPr?r"rrrrrr.selectsf          r|)!rotypingrrrrrprompt_toolkit.applicationrprompt_toolkit.key_bindingrprompt_toolkit.keysr prompt_toolkit.stylesr questionaryr questionary.constantsr r questionary.promptsrquestionary.prompts.commonrrrquestionary.questionrquestionary.stylesrstrboolr|r-r-r-r.s