Spaces:
Sleeping
Sleeping
File size: 12,589 Bytes
2d876d1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 |
"""Module contains the class to create a rawlist prompt."""
from typing import Any, Callable, List, Optional, Tuple, Union
from InquirerPy.base import InquirerPyUIListControl
from InquirerPy.enum import INQUIRERPY_POINTER_SEQUENCE
from InquirerPy.exceptions import InvalidArgument
from InquirerPy.prompts.list import ListPrompt
from InquirerPy.separator import Separator
from InquirerPy.utils import (
InquirerPyDefault,
InquirerPyKeybindings,
InquirerPyListChoices,
InquirerPyMessage,
InquirerPySessionResult,
InquirerPyStyle,
InquirerPyValidate,
)
__all__ = ["RawlistPrompt"]
class InquirerPyRawlistControl(InquirerPyUIListControl):
"""An :class:`~prompt_toolkit.layout.UIControl` class that displays a list of choices.
Reference the parameter definition in :class:`.RawlistPrompt`.
"""
def __init__(
self,
choices: InquirerPyListChoices,
default: Any,
pointer: str,
separator: str,
marker: str,
session_result: Optional[InquirerPySessionResult],
multiselect: bool,
marker_pl: str,
) -> None:
self._pointer = pointer
self._separator = separator
self._marker = marker
self._marker_pl = marker_pl
super().__init__(
choices=choices,
default=default,
session_result=session_result,
multiselect=multiselect,
)
def _format_choices(self) -> None:
separator_count = 0
for index, choice in enumerate(self.choices):
if isinstance(choice["value"], Separator):
separator_count += 1
continue
choice["display_index"] = index + 1 - separator_count
choice["actual_index"] = index
if self.choices:
first_valid_choice_index = 0
while isinstance(
self.choices[first_valid_choice_index]["value"], Separator
):
first_valid_choice_index += 1
if self.selected_choice_index == first_valid_choice_index:
for choice in self.choices:
if isinstance(choice["value"], Separator):
continue
if choice["display_index"] == self._default:
self.selected_choice_index = choice["actual_index"]
break
def _get_hover_text(self, choice) -> List[Tuple[str, str]]:
display_choices = []
display_choices.append(("class:pointer", self._pointer))
display_choices.append(
(
"class:marker",
self._marker if choice["enabled"] else self._marker_pl,
)
)
if not isinstance(choice["value"], Separator):
display_choices.append(
(
"class:pointer",
"%s%s" % (str(choice["display_index"]), self._separator),
)
)
display_choices.append(("[SetCursorPosition]", ""))
display_choices.append(("class:pointer", choice["name"]))
return display_choices
def _get_normal_text(self, choice) -> List[Tuple[str, str]]:
display_choices = []
display_choices.append(("", len(self._pointer) * " "))
display_choices.append(
(
"class:marker",
self._marker if choice["enabled"] else self._marker_pl,
)
)
if not isinstance(choice["value"], Separator):
display_choices.append(
("", "%s%s" % (str(choice["display_index"]), self._separator))
)
display_choices.append(("", choice["name"]))
else:
display_choices.append(("class:separator", choice["name"]))
return display_choices
class RawlistPrompt(ListPrompt):
"""Create a prompt that displays a list of choices with index number as shortcuts.
A wrapper class around :class:`~prompt_toolkit.application.Application`.
Each choice will have an index number infront of them with keybinding created for that index number.
Enables user to use number to jump to different choices using number.
Args:
message: The question to ask the user.
Refer to :ref:`pages/dynamic:message` documentation for more details.
choices: List of choices to display and select.
Refer to :ref:`pages/prompts/rawlist:Choices` documentation for more details.
style: An :class:`InquirerPyStyle` instance.
Refer to :ref:`Style <pages/style:Alternate Syntax>` documentation for more details.
vi_mode: Use vim keybinding for the prompt.
Refer to :ref:`pages/kb:Keybindings` documentation for more details.
default: Set the default value of the prompt.
This will be used to determine which choice is highlighted (current selection),
The default value should be the value of one of the choices.
For :class:`.RawlistPrompt` specifically, default value can also be value between 0-9.
Refer to :ref:`pages/dynamic:default` documentation for more details.
separator: Separator symbol. Custom symbol that will be used as a separator between the choice index number and the choices.
qmark: Question mark symbol. Custom symbol that will be displayed infront of the question before its answered.
amark: Answer mark symbol. Custom symbol that will be displayed infront of the question after its answered.
pointer: Pointer symbol. Customer symbol that will be used to indicate the current choice selection.
instruction: Short instruction to display next to the question.
long_instruction: Long instructions to display at the bottom of the prompt.
validate: Add validation to user input.
The main use case for this prompt would be when `multiselect` is True, you can enforce a min/max selection.
Refer to :ref:`pages/validator:Validator` documentation for more details.
invalid_message: Error message to display when user input is invalid.
Refer to :ref:`pages/validator:Validator` documentation for more details.
transformer: A function which performs additional transformation on the value that gets printed to the terminal.
Different than `filter` parameter, this is only visual effect and won’t affect the actual value returned by :meth:`~InquirerPy.base.simple.BaseSimplePrompt.execute`.
Refer to :ref:`pages/dynamic:transformer` documentation for more details.
filter: A function which performs additional transformation on the result.
This affects the actual value returned by :meth:`~InquirerPy.base.simple.BaseSimplePrompt.execute`.
Refer to :ref:`pages/dynamic:filter` documentation for more details.
height: Preferred height of the prompt.
Refer to :ref:`pages/height:Height` documentation for more details.
max_height: Max height of the prompt.
Refer to :ref:`pages/height:Height` documentation for more details.
multiselect: Enable multi-selection on choices.
You can use `validate` parameter to control min/max selections.
Setting to True will also change the result from a single value to a list of values.
marker: Marker Symbol. Custom symbol to indicate if a choice is selected.
This will take effects when `multiselect` is True.
marker_pl: Marker place holder when the choice is not selected.
This is empty space by default.
border: Create border around the choice window.
keybindings: Customise the builtin keybindings.
Refer to :ref:`pages/kb:Keybindings` for more details.
show_cursor: Display cursor at the end of the prompt.
Set to False to hide the cursor.
cycle: Return to top item if hit bottom during navigation or vice versa.
wrap_lines: Soft wrap question lines when question exceeds the terminal width.
raise_keyboard_interrupt: Raise the :class:`KeyboardInterrupt` exception when `ctrl-c` is pressed. If false, the result
will be `None` and the question is skiped.
mandatory: Indicate if the prompt is mandatory. If True, then the question cannot be skipped.
mandatory_message: Error message to show when user attempts to skip mandatory prompt.
session_result: Used internally for :ref:`index:Classic Syntax (PyInquirer)`.
Examples:
>>> from InquirerPy import inquirer
>>> result = inquirer.rawlist(message="Select one:", choices=[1, 2, 3]).execute()
>>> print(result)
1
"""
def __init__(
self,
message: InquirerPyMessage,
choices: InquirerPyListChoices,
default: InquirerPyDefault = None,
separator: str = ") ",
style: Optional[InquirerPyStyle] = None,
vi_mode: bool = False,
qmark: str = "?",
amark: str = "?",
pointer: str = " ",
instruction: str = "",
long_instruction: str = "",
transformer: Optional[Callable[[Any], Any]] = None,
filter: Optional[Callable[[Any], Any]] = None,
height: Optional[Union[int, str]] = None,
max_height: Optional[Union[int, str]] = None,
multiselect: bool = False,
marker: str = INQUIRERPY_POINTER_SEQUENCE,
marker_pl: str = " ",
border: bool = False,
validate: Optional[InquirerPyValidate] = None,
invalid_message: str = "Invalid input",
keybindings: Optional[InquirerPyKeybindings] = None,
show_cursor: bool = True,
cycle: bool = True,
wrap_lines: bool = True,
raise_keyboard_interrupt: bool = True,
mandatory: bool = True,
mandatory_message: str = "Mandatory prompt",
session_result: Optional[InquirerPySessionResult] = None,
) -> None:
self.content_control = InquirerPyRawlistControl(
choices=choices,
default=default,
pointer=pointer,
separator=separator,
marker=marker,
session_result=session_result,
multiselect=multiselect,
marker_pl=marker_pl,
)
super().__init__(
message=message,
choices=choices,
style=style,
border=border,
vi_mode=vi_mode,
qmark=qmark,
amark=amark,
instruction=instruction,
long_instruction=long_instruction,
transformer=transformer,
filter=filter,
height=height,
max_height=max_height,
multiselect=multiselect,
validate=validate,
invalid_message=invalid_message,
keybindings=keybindings,
show_cursor=show_cursor,
cycle=cycle,
wrap_lines=wrap_lines,
raise_keyboard_interrupt=raise_keyboard_interrupt,
mandatory=mandatory,
mandatory_message=mandatory_message,
session_result=session_result,
)
def _on_rendered(self, _) -> None:
"""Override this method to apply custom keybindings.
Needs to creat these kb in the callback due to `after_render`
retrieve the choices asynchronously.
Check if fetched choices exceed the limit of 9, raise
InvalidArgument exception.
"""
def keybinding_factory(choice):
@self.register_kb(str(choice["display_index"]))
def keybinding(_) -> None:
self.content_control.selected_choice_index = int(choice["actual_index"])
return keybinding
if self.content_control.choice_count >= 10:
raise InvalidArgument("rawlist argument choices cannot exceed length of 9")
for choice in self.content_control.choices:
if not isinstance(choice["value"], Separator):
keybinding_factory(choice)
def _get_prompt_message(self) -> List[Tuple[str, str]]:
"""Return the formatted text to display in the prompt.
Overriding this method to allow multiple formatted class to be displayed.
"""
display_message = super()._get_prompt_message()
if not self.status["answered"] and self.content_control.choices:
display_message.append(
("class:input", str(self.content_control.selection["display_index"]))
)
return display_message
|