U
ʗRe', @ s d dl mZmZmZmZmZmZmZmZ ddl m
Z
ddl m Z ddl
mZmZ edZedZG dd d eZG d
d d eZG d d
d
ee ZG dd dee ZG dd dee ZG dd dee ZG dd dee Z e dkrd dl m Z e j dddrej dd dZ!e!dkr0e!d kr0q|| _|| _ || _
d S )Nr+ style) r r'
isinstancestrr
from_markupr+ r( r$ r) r* )r r+ r' r( r$ r) r* r r r r 5 s
zPromptBase.__init__)r' r( r$ r) r* stream) r+ r' r( r$ r) r* defaultr1 r c C s d S r r ) clsr+ r' r( r$ r) r* r2 r1 r r r askK s zPromptBase.ask)r+ r' r( r$ r) r* r1 r c C s d S r r )r3 r+ r' r( r$ r) r* r1 r r r r4 [ s
.)r' r( r$ r) r* r2 r1 c
C s | ||||||d} | ||dS )a 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.
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.
r& r2 r1 r )
r3 r+ r' r( r$ r) r* r2 r1 Z_promptr r r r4 j s r2 r c C s t d| ddS )zTurn the supplied default in to a Text instance.
Args:
default (DefaultType): Default value.
Returns:
Text: Text containing rendering of default value.
()prompt.default)r
)r r2 r r r render_default s zPromptBase.render_defaultc C s | j }d|_| jrJ| jrJd| j}d| d}|d ||d |dkr| jrt|t | j
fr|d | |}|| || j |S )zMake prompt text.
Args:
default (DefaultType): Default value.
Returns:
Text: Text to display in prompt.
r% /[] zprompt.choices.)
r+ copyendr* r$ joinappendr) r. r/ r# r:
prompt_suffix)r r2 r+ Z_choicesr$ _defaultr r r make_prompt s$
zPromptBase.make_prompt)r' r+ r( r1 r c C s |j |||dS )zGet input from user.
Args:
console (Console): Console instance.
prompt (TextType): Prompt text.
password (bool): Enable password entry.
Returns:
str: String from user.
)r( r1 )input)r3 r' r+ r( r1 r r r get_input s zPromptBase.get_inputvaluer c C s | j dk st| | j kS )zCheck value is in the list of valid choices.
Args:
value (str): Value entered by user.
Returns:
bool: True if choice was valid, otherwise False.
N)r$ AssertionErrorstripr rI r r r check_choice s zPromptBase.check_choicec C sX | }z| |}W n tk
r4 t| jY nX | jdk rT| |sTt| j|S )a 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.
N)rK r#
ValueErrorr validate_error_messager$ rM illegal_choice_message)r rI return_valuer r r process_response s
zPromptBase.process_response)rI errorr c C s | j | dS )zCalled to handle validation error.
Args:
value (str): String entered by user.
error (InvalidResponse): Exception instance the initiated the error.
N)r' print)r rI rS r r r on_validate_error s z PromptBase.on_validate_errorr c C s dS )z,Hook to display something before the prompt.Nr r r r r
pre_prompt s zPromptBase.pre_promptr1 )r1 r c C s d S r r )r r1 r r r __call__ s zPromptBase.__call__)r2 r1 r c C s d S r r )r r2 r1 r r r rX s r5 c
C s | | |}| j| j|| j|d}|dkr<|dkr<|S z| |}W n6 tk
r } z| || W Y q W 5 d}~X Y q X |S q dS )zRun the prompt loop.
Args:
default (Any, optional): Optional default value.
Returns:
PromptType: Processed value.
rW r% .N)rV rE rG r' r( rR r rU )r r2 r1 r+ rI rQ rS r r r rX s
)r% )r% )r% )r% )N)$r r r r r/ r# type__annotations__rO rP rC r$ r r r r boolr classmethodr r r r r r4 r r
r: rE rG rM rR r rU rV rX r r r r r" s
$
r" c @ s e Zd ZdZeZdS )PromptzbA prompt that returns a str.
Example:
>>> name = Prompt.ask("Enter your name")
N)r r r r r/ r# r r r r r] s r] c @ s e Zd ZdZeZdZdS ) IntPromptzA prompt that returns an integer.
Example:
>>> burrito_count = IntPrompt.ask("How many burritos do you want to order")
z3[prompt.invalid]Please enter a valid integer numberN)r r r r intr# rO r r r r r^ * s r^ c @ s e Zd ZdZeZdZdS ) FloatPromptzyA prompt that returns a float.
Example:
>>> temperature = FloatPrompt.ask("Enter desired temperature")
z%[prompt.invalid]Please enter a numberN)r r r r floatr# rO r r r r r` 6 s r` c @ sN e Zd ZU dZeZdZddgZee e
d< e e dddZ
e ed d
d Zd S )
ConfirmzuA yes / no confirmation prompt.
Example:
>>> if Confirm.ask("Continue"):
run_job()
z#[prompt.invalid]Please enter Y or Nynr$ r6 c C s. | j \}}t|r d| dn
d| dddS )z8Render the default as (y) or (n) rather than True/False.r7 r8 r9 r, )r$ r
)r r2 yesnor r r r: O s
zConfirm.render_defaultrH c C s. | }|| jkr t| j|| jd kS )zConvert choices to a bool.r )rK lowerr$ r rO rL r r r rR T s
zConfirm.process_responseN)r r r r r[ r# rO r$ r r/ rZ r r
r: rR r r r r rb B s
rb __main__)rT zRun [i]prompt[/i] tests?T)r2 z6:rocket: Enter a number between [b]1[/b] and [b]10[/b]
z=:pile_of_poo: [prompt.invalid]Number must be between 1 and 10znumber=z=Please enter a password [cyan](must be at least 5 characters))r( z"[prompt.invalid]password too shortz password=z
Enter a fruitappleZorangepear)r$ zfruit=z[b]OK :loudly_crying_face:N)%typingr r r r r r r r r% r r' r textr
r r r Exceptionr r r" r/ r] r_ r^ r` r[ rb r pip._vendor.richrT r4 resultr( lenZfruitr r r r sH (