File: //opt/python38/lib/python3.8/site-packages/pip/_vendor/rich/__pycache__/prompt.cpython-38.pyc
U
�{�e', � @ s� d dl mZmZmZmZmZmZmZmZ ddl m
Z
ddl m Z ddl
mZmZ ed�Zed�ZG dd � d e�ZG d
d � d e�ZG d d
� d
ee �ZG dd� dee �ZG dd� dee �ZG dd� dee �ZG dd� dee �Z e dk�r�d dl m Z e j ddd��r�ej dd d�Z!e!dk�r0e!d k�r0�q<e d � �q
e d e!� �� ej d dd!�Z"e#e"�d k�rj�qve d"� �qJe d#e"��� ej d$d%d&d'gd(�Z$e d)e$��� ne d*� d+S ),� )�Any�Generic�List�Optional�TextIO�TypeVar�Union�overload� )� get_console)�Console)�Text�TextType�
PromptType� DefaultTypec @ s e Zd ZdZdS )� PromptErrorz/Exception base class for prompt related errors.N)�__name__�
__module__� __qualname__�__doc__� r r �;/tmp/pip-unpacked-wheel-_0scjqea/pip/_vendor/rich/prompt.pyr s r c @ s. e Zd ZdZedd�dd�Zed�dd�ZdS ) �InvalidResponsez�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.
N)�message�returnc C s
|| _ d S �N�r )�selfr r r r �__init__ s zInvalidResponse.__init__�r c C s | j S r r �r r r r �__rich__ s zInvalidResponse.__rich__)r r r r r r r! r r r r r s r c @ s4 e Zd ZU dZeZeed< dZdZ dZ
dZ e e
e ed< d3dd dd
d
d �ee e ee e
e eedd �d
d�Zeed4dd dd
d
dd�ee e ee e
e eeee e eeef d� dd���Zeed5dd dd
d
dd�ee e ee e
e eee e ed�dd���Zed6dd dd
d
ddd�ee e ee e
e eeee e ed� dd��Zeed�dd�Zeed�dd �Z ed7eeee e ed �d d ��Z eed �d!d"�Z eed �d#d$�Z ee dd%�d&d'�Z!dd(�d)d*�Z"edd+�e e ed,�d-d.��Z#edd+�ee e eeef d/�d0d.��Z#ddd1�ee e ed/�d2d.�Z#dS )8�
PromptBasea� 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.
show_default (bool, optional): Show default in prompt. Defaults to True.
show_choices (bool, optional): Show choices in prompt. Defaults to True.
�
response_typez*[prompt.invalid]Please enter a valid valuezA[prompt.invalid.choice]Please select one of the available optionsz: N�choices� FT��console�passwordr$ � show_default� show_choices)�promptr' r( r$ r) r* r c C sN |pt � | _t|t�r$tj|dd�n|| _|| _|d k r>|| _|| _ || _
d S )Nr+ ��style) r r' �
isinstance�strr
Z 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* �defaultr0 r c C s d S r r ) �clsr+ r' r( r$ r) r* r1 r0 r r r �askK s zPromptBase.ask)r+ r' r( r$ r) r* r0 r c C s d S r r )r2 r+ r' r( r$ r) r* r0 r r r r3 [ s
.)r' r( r$ r) r* r1 r0 c
C s | ||||||d�} | ||d�S )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& �r1 r0 r )
r2 r+ r' r( r$ r) r* r1 r0 Z_promptr r r r3 j s ��r1 r c C s t d|� d�d�S )z�Turn 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 r1 r r r �render_default� s zPromptBase.render_defaultc C s� | j �� }d|_| jrJ| jrJd�| j�}d|� d�}|�d� |�|d� |dkr�| jr�t|t | j
f�r�|�d� | � |�}|�|� |�| j � |S )z�Make prompt text.
Args:
default (DefaultType): Default value.
Returns:
Text: Text to display in prompt.
r% �/�[�]� zprompt.choices.)
r+ �copy�endr* r$ �join�appendr) r. r/ r# r9 �
prompt_suffix)r r1 r+ Z_choicesr$ �_defaultr r r � make_prompt� s$
���
zPromptBase.make_prompt)r' r+ r( r0 r c C s |j |||d�S )z�Get input from user.
Args:
console (Console): Console instance.
prompt (TextType): Prompt text.
password (bool): Enable password entry.
Returns:
str: String from user.
)r( r0 )�input)r2 r' r+ r( r0 r r r � get_input� s zPromptBase.get_input��valuer c C s | j dk st�|�� | j kS )z�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.
N)r$ �AssertionError�strip�r rH r r r � check_choice� s zPromptBase.check_choicec C sX |� � }z| �|�}W n tk
r4 t| j��Y 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)rJ r# �
ValueErrorr �validate_error_messager$ rL �illegal_choice_message)r rH � return_valuer r r �process_response� s
zPromptBase.process_response)rH �errorr c C s | j �|� dS )z�Called to handle validation error.
Args:
value (str): String entered by user.
error (InvalidResponse): Exception instance the initiated the error.
N)r' �print)r rH rR 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_prompt�r0 )r0 r c C s d S r r )r r0 r r r �__call__� s zPromptBase.__call__)r1 r0 r c C s d S r r )r r1 r0 r r r rW s r4 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 )z�Run the prompt loop.
Args:
default (Any, optional): Optional default value.
Returns:
PromptType: Processed value.
rV r% .N)rU rD rF r' r( rQ r rT )r r1 r0 r+ rH rP rR r r r rW s
)r% )r% )r% )r% )N)$r r r r r/ r# �type�__annotations__rN rO rB r$ r r r r �boolr � classmethodr r r r r r3 r r
r9 rD rF rL rQ r rT rU rW 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 )� IntPromptz�A 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# rN 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# rN r r r r r_ 6 s r_ c @ sN e Zd ZU dZeZdZddgZee e
d< e e d�dd�Z
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 N�y�nr$ r5 c C s. | j \}}t|r d|� d�n
d|� d�dd�S )z8Render the default as (y) or (n) rather than True/False.r6 r7 r8 r, )r$ r
)r r1 �yes�nor r r r9 O s
zConfirm.render_defaultrG c C s. |� � �� }|| jkr t| j��|| jd kS )zConvert choices to a bool.r )rJ �lowerr$ r rN rK r r r rQ T s
zConfirm.process_responseN)r r r r rZ r# rN r$ r r/ rY r r
r9 rQ r r r r ra B s
ra �__main__)rS zRun [i]prompt[/i] tests?T)r1 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 fruitZappleZorangeZpear)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_ rZ ra r Zpip._vendor.richrS r3 �resultr( �lenZfruitr r r r �<module> sH (
� �