golden hour
/opt/cloudlinux/venv/lib/python3.11/site-packages/pip/_vendor/rich
⬆️ Go Up
Upload
File/Folder
Size
Actions
LICENSE
1.03 KB
Del
OK
__init__.py
5.95 KB
Del
OK
__main__.py
7.71 KB
Del
OK
__pycache__
-
Del
OK
_cell_widths.py
9.97 KB
Del
OK
_emoji_codes.py
136.95 KB
Del
OK
_emoji_replace.py
1.04 KB
Del
OK
_export_format.py
2.08 KB
Del
OK
_extension.py
265 B
Del
OK
_fileno.py
799 B
Del
OK
_inspect.py
9.43 KB
Del
OK
_log_render.py
3.15 KB
Del
OK
_loop.py
1.21 KB
Del
OK
_null_file.py
1.36 KB
Del
OK
_palettes.py
6.9 KB
Del
OK
_pick.py
423 B
Del
OK
_ratio.py
5.2 KB
Del
OK
_spinners.py
19.45 KB
Del
OK
_stack.py
351 B
Del
OK
_timer.py
417 B
Del
OK
_win32_console.py
22.22 KB
Del
OK
_windows.py
1.88 KB
Del
OK
_windows_renderer.py
2.72 KB
Del
OK
_wrap.py
3.32 KB
Del
OK
abc.py
890 B
Del
OK
align.py
10.08 KB
Del
OK
ansi.py
6.76 KB
Del
OK
bar.py
3.19 KB
Del
OK
box.py
10.44 KB
Del
OK
cells.py
5.01 KB
Del
OK
color.py
17.78 KB
Del
OK
color_triplet.py
1.03 KB
Del
OK
columns.py
6.96 KB
Del
OK
console.py
98.49 KB
Del
OK
constrain.py
1.26 KB
Del
OK
containers.py
5.37 KB
Del
OK
control.py
6.33 KB
Del
OK
default_styles.py
8.06 KB
Del
OK
diagnose.py
1 KB
Del
OK
emoji.py
2.31 KB
Del
OK
errors.py
642 B
Del
OK
file_proxy.py
1.64 KB
Del
OK
filesize.py
2.43 KB
Del
OK
highlighter.py
9.36 KB
Del
OK
json.py
4.91 KB
Del
OK
jupyter.py
3.18 KB
Del
OK
layout.py
13.68 KB
Del
OK
live.py
14.82 KB
Del
OK
live_render.py
3.44 KB
Del
OK
logging.py
12.18 KB
Del
OK
markup.py
8.25 KB
Del
OK
measure.py
5.18 KB
Del
OK
padding.py
4.79 KB
Del
OK
pager.py
828 B
Del
OK
palette.py
3.32 KB
Del
OK
panel.py
10.9 KB
Del
OK
pretty.py
35.54 KB
Del
OK
progress.py
58.99 KB
Del
OK
progress_bar.py
7.97 KB
Del
OK
prompt.py
12.16 KB
Del
OK
protocol.py
1.36 KB
Del
OK
py.typed
0 B
Del
OK
region.py
166 B
Del
OK
repr.py
4.33 KB
Del
OK
rule.py
4.49 KB
Del
OK
scope.py
2.78 KB
Del
OK
screen.py
1.55 KB
Del
OK
segment.py
24.16 KB
Del
OK
spinner.py
4.12 KB
Del
OK
status.py
4.32 KB
Del
OK
style.py
26.36 KB
Del
OK
styled.py
1.23 KB
Del
OK
syntax.py
35.52 KB
Del
OK
table.py
39.11 KB
Del
OK
terminal_theme.py
3.29 KB
Del
OK
text.py
46.44 KB
Del
OK
theme.py
3.68 KB
Del
OK
themes.py
102 B
Del
OK
traceback.py
35.02 KB
Del
OK
tree.py
9.23 KB
Del
OK
Edit: prompt.py
from typing import Any, Generic, List, Optional, TextIO, TypeVar, Union, overload from . import get_console from .console import Console from .text import Text, TextType PromptType = TypeVar("PromptType") DefaultType = TypeVar("DefaultType") class PromptError(Exception): """Exception base class for prompt related errors.""" class InvalidResponse(PromptError): """Exception to indicate a response was invalid. Raise this within process_response() to indicate an error and provide an error message. Args: message (Union[str, Text]): Error message. """ def __init__(self, message: TextType) -> None: self.message = message def __rich__(self) -> TextType: return self.message class PromptBase(Generic[PromptType]): """Ask the user for input until a valid response is received. This is the base class, see one of the concrete classes for examples. Args: prompt (TextType, optional): Prompt text. Defaults to "". console (Console, optional): A Console instance or None to use global console. Defaults to None. password (bool, optional): Enable password input. Defaults to False. choices (List[str], optional): A list of valid choices. Defaults to None. case_sensitive (bool, optional): Matching of choices should be case-sensitive. Defaults to True. show_default (bool, optional): Show default in prompt. Defaults to True. show_choices (bool, optional): Show choices in prompt. Defaults to True. """ response_type: type = str validate_error_message = "[prompt.invalid]Please enter a valid value" illegal_choice_message = ( "[prompt.invalid.choice]Please select one of the available options" ) prompt_suffix = ": " choices: Optional[List[str]] = None def __init__( self, prompt: TextType = "", *, console: Optional[Console] = None, password: bool = False, choices: Optional[List[str]] = None, case_sensitive: bool = True, show_default: bool = True, show_choices: bool = True, ) -> None: self.console = console or get_console() self.prompt = ( Text.from_markup(prompt, style="prompt") if isinstance(prompt, str) else prompt ) self.password = password if choices is not None: self.choices = choices self.case_sensitive = case_sensitive self.show_default = show_default self.show_choices = show_choices @classmethod @overload def ask( cls, prompt: TextType = "", *, console: Optional[Console] = None, password: bool = False, choices: Optional[List[str]] = None, case_sensitive: bool = True, show_default: bool = True, show_choices: bool = True, default: DefaultType, stream: Optional[TextIO] = None, ) -> Union[DefaultType, PromptType]: ... @classmethod @overload def ask( cls, prompt: TextType = "", *, console: Optional[Console] = None, password: bool = False, choices: Optional[List[str]] = None, case_sensitive: bool = True, show_default: bool = True, show_choices: bool = True, stream: Optional[TextIO] = None, ) -> PromptType: ... @classmethod def ask( cls, prompt: TextType = "", *, console: Optional[Console] = None, password: bool = False, choices: Optional[List[str]] = None, case_sensitive: bool = True, show_default: bool = True, show_choices: bool = True, default: Any = ..., stream: Optional[TextIO] = None, ) -> Any: """Shortcut to construct and run a prompt loop and return the result. Example: >>> filename = Prompt.ask("Enter a filename") Args: prompt (TextType, optional): Prompt text. Defaults to "". console (Console, optional): A Console instance or None to use global console. Defaults to None. password (bool, optional): Enable password input. Defaults to False. choices (List[str], optional): A list of valid choices. Defaults to None. case_sensitive (bool, optional): Matching of choices should be case-sensitive. Defaults to True. show_default (bool, optional): Show default in prompt. Defaults to True. show_choices (bool, optional): Show choices in prompt. Defaults to True. stream (TextIO, optional): Optional text file open for reading to get input. Defaults to None. """ _prompt = cls( prompt, console=console, password=password, choices=choices, case_sensitive=case_sensitive, show_default=show_default, show_choices=show_choices, ) return _prompt(default=default, stream=stream) def render_default(self, default: DefaultType) -> Text: """Turn the supplied default in to a Text instance. Args: default (DefaultType): Default value. Returns: Text: Text containing rendering of default value. """ return Text(f"({default})", "prompt.default") def make_prompt(self, default: DefaultType) -> Text: """Make prompt text. Args: default (DefaultType): Default value. Returns: Text: Text to display in prompt. """ prompt = self.prompt.copy() prompt.end = "" if self.show_choices and self.choices: _choices = "/".join(self.choices) choices = f"[{_choices}]" prompt.append(" ") prompt.append(choices, "prompt.choices") if ( default != ... and self.show_default and isinstance(default, (str, self.response_type)) ): prompt.append(" ") _default = self.render_default(default) prompt.append(_default) prompt.append(self.prompt_suffix) return prompt @classmethod def get_input( cls, console: Console, prompt: TextType, password: bool, stream: Optional[TextIO] = None, ) -> str: """Get input from user. Args: console (Console): Console instance. prompt (TextType): Prompt text. password (bool): Enable password entry. Returns: str: String from user. """ return console.input(prompt, password=password, stream=stream) def check_choice(self, value: str) -> bool: """Check value is in the list of valid choices. Args: value (str): Value entered by user. Returns: bool: True if choice was valid, otherwise False. """ assert self.choices is not None if self.case_sensitive: return value.strip() in self.choices return value.strip().lower() in [choice.lower() for choice in self.choices] def process_response(self, value: str) -> PromptType: """Process response from user, convert to prompt type. Args: value (str): String typed by user. Raises: InvalidResponse: If ``value`` is invalid. Returns: PromptType: The value to be returned from ask method. """ value = value.strip() try: return_value: PromptType = self.response_type(value) except ValueError: raise InvalidResponse(self.validate_error_message) if self.choices is not None: if not self.check_choice(value): raise InvalidResponse(self.illegal_choice_message) if not self.case_sensitive: # return the original choice, not the lower case version return_value = self.response_type( self.choices[ [choice.lower() for choice in self.choices].index(value.lower()) ] ) return return_value def on_validate_error(self, value: str, error: InvalidResponse) -> None: """Called to handle validation error. Args: value (str): String entered by user. error (InvalidResponse): Exception instance the initiated the error. """ self.console.print(error) def pre_prompt(self) -> None: """Hook to display something before the prompt.""" @overload def __call__(self, *, stream: Optional[TextIO] = None) -> PromptType: ... @overload def __call__( self, *, default: DefaultType, stream: Optional[TextIO] = None ) -> Union[PromptType, DefaultType]: ... def __call__(self, *, default: Any = ..., stream: Optional[TextIO] = None) -> Any: """Run the prompt loop. Args: default (Any, optional): Optional default value. Returns: PromptType: Processed value. """ while True: self.pre_prompt() prompt = self.make_prompt(default) value = self.get_input(self.console, prompt, self.password, stream=stream) if value == "" and default != ...: return default try: return_value = self.process_response(value) except InvalidResponse as error: self.on_validate_error(value, error) continue else: return return_value class Prompt(PromptBase[str]): """A prompt that returns a str. Example: >>> name = Prompt.ask("Enter your name") """ response_type = str class IntPrompt(PromptBase[int]): """A prompt that returns an integer. Example: >>> burrito_count = IntPrompt.ask("How many burritos do you want to order") """ response_type = int validate_error_message = "[prompt.invalid]Please enter a valid integer number" class FloatPrompt(PromptBase[float]): """A prompt that returns a float. Example: >>> temperature = FloatPrompt.ask("Enter desired temperature") """ response_type = float validate_error_message = "[prompt.invalid]Please enter a number" class Confirm(PromptBase[bool]): """A yes / no confirmation prompt. Example: >>> if Confirm.ask("Continue"): run_job() """ response_type = bool validate_error_message = "[prompt.invalid]Please enter Y or N" choices: List[str] = ["y", "n"] def render_default(self, default: DefaultType) -> Text: """Render the default as (y) or (n) rather than True/False.""" yes, no = self.choices return Text(f"({yes})" if default else f"({no})", style="prompt.default") def process_response(self, value: str) -> bool: """Convert choices to a bool.""" value = value.strip().lower() if value not in self.choices: raise InvalidResponse(self.validate_error_message) return value == self.choices[0] if __name__ == "__main__": # pragma: no cover from pip._vendor.rich import print if Confirm.ask("Run [i]prompt[/i] tests?", default=True): while True: result = IntPrompt.ask( ":rocket: Enter a number between [b]1[/b] and [b]10[/b]", default=5 ) if result >= 1 and result <= 10: break print(":pile_of_poo: [prompt.invalid]Number must be between 1 and 10") print(f"number={result}") while True: password = Prompt.ask( "Please enter a password [cyan](must be at least 5 characters)", password=True, ) if len(password) >= 5: break print("[prompt.invalid]password too short") print(f"password={password!r}") fruit = Prompt.ask("Enter a fruit", choices=["apple", "orange", "pear"]) print(f"fruit={fruit!r}") doggie = Prompt.ask( "What's the best Dog? (Case INSENSITIVE)", choices=["Border Terrier", "Collie", "Labradoodle"], case_sensitive=False, ) print(f"doggie={doggie!r}") else: print("[b]OK :loudly_crying_face:")
Save