File: //usr/lib/python3/dist-packages/pip/_vendor/rich/__pycache__/prompt.cpython-310.pyc
o
@%Ne+, � @ 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 dkr�d dl m Z e j ddd�r� ej dd d�Z!e!dkr�e!d kr�ne d � q�e d e!� �� ej d dd!�Z"e#e"�d kr�ne d"� q�e d#e"��� ej d$g d%�d&�Z$e d'e$��� d)S e d(� d)S 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 �9/usr/lib/python3/dist-packages/pip/_vendor/rich/prompt.pyr s r c @ s0 e Zd ZdZdeddfdd�Zdefdd�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.
�message�returnNc C s
|| _ d S �N�r )�selfr r r r �__init__ s
zInvalidResponse.__init__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 @ s� e Zd ZU dZeZeed< dZdZ dZ
dZ e e
e ed< d3dd dd
d
d �d ed
e e dede e
e dededdfdd�Zee d3dd dd
d
dd�d ed
e e dede e
e dededede e deeef fdd���Zee d3dd dd
d
dd�d ed
e e dede e
e dedede e defdd���Ze d3dd dd
d
ddd�d ed
e e dede e
e dededede e defd d��Zdedefd d �Zdedefd d �Z e d4d
ed edede e def
d!d"��Z d#edefd$d%�Z d#edefd&d'�Z d#ed(e ddfd)d*�Z!d5d+d,�Z"edd-�de e defd.d/��Z#edd-�dede e deeef fd0d/��Z#ddd1�dede e defd2d/�Z#dS )6�
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 c C sN |pt � | _t|t�rtj|dd�n|| _|| _|d ur || _|| _ || _
d S )Nr* ��style) r r& �
isinstance�strr
� 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�defaultr0 c C � d S r r ) �clsr* r& r' r# r( r) r1 r0 r r r �askK s zPromptBase.askc C r2 r r )r3 r* r& r' r# r( r) r0 r r r r4 [ 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 )
r3 r* r& r' r# r( r) r1 r0 �_promptr r r r4 j s � 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|_| jr%| jr%d�| j�}d|� d�}|�d� |�|d� |dkrC| jrCt|t | j
f�rC|�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" r: �
prompt_suffix)r r1 r* �_choicesr# �_defaultr r r � make_prompt� s"
��
zPromptBase.make_promptc 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)r3 r& r* r' r0 r r r � get_input� s zPromptBase.get_input�valuec C s | j dusJ �|�� | j v S )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# �strip�r rI r r r � check_choice� s zPromptBase.check_choicec C sR |� � }z| �|�}W n ty t| j��w | jdur'| �|�s't| 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 rI � return_valuer r r �process_response� s
�
zPromptBase.process_response�errorc 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 rI rR r r r �on_validate_error� s z PromptBase.on_validate_errorc C s dS )z,Hook to display something before the prompt.Nr r r r r �
pre_prompt� s zPromptBase.pre_prompt�r0 c C r2 r r )r r0 r r r �__call__� s zPromptBase.__call__c C r2 r r )r r1 r0 r r r rW s r5 c
C s� | � � | �|�}| j| j|| j|d�}|dkr |dkr |S z| �|�}W |S ty? } z | �||� W Y d}~q d}~ww )z�Run the prompt loop.
Args:
default (Any, optional): Optional default value.
Returns:
PromptType: Processed value.
TrV r$ .N)rU rF rH r& r' rQ r rT )r r1 r0 r* rI rP rR r r r rW s
� ��)r$ r )r N)$r r r r r. r" �type�__annotations__rN rO rC r# r r r r �boolr � classmethodr r r r r r4 r r
r: rF rH 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 @ � 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^ * � r^ c @ r] )� 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 ra 6 r` ra c @ sR e Zd ZU dZeZdZddgZee e
d< de de fdd �Z
d
e defd 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# r1 r c C s4 | j \}}t|rd|� d�dd�S d|� d�dd�S )z8Render the default as (y) or (n) rather than True/False.r7 r8 r9 r+ )r# r
)r r1 �yes�nor r r r: O s
*zConfirm.render_defaultrI c C s. |� � �� }|| jvrt| 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
r: rQ r r r r rc B s
rc �__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 fruit)�apple�orange�pear)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^ ra rZ rc r �pip._vendor.richrS r4 �resultr' �len�fruitr r r r �<module> sR ( ��� � �