o
�J�h�0 � @ 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%��� ej!d(g d)�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 �IC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\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< d4dd dd
d
d
d �d ed
e e dede e
e dedededdfdd�Zee d4dd dd
d
d
dd�d ed
e e dede e
e dedededede e deeef fdd���Zee d4dd dd
d
d
dd�d ed
e e dede e
e dededede e defdd���Ze d4dd dd
d
d
ddd �d ed
e e dede e
e dedededede e defd d��Zdedefd d �Zdedefd d!�Z e d5d
ed edede e def
d"d#��Z d$edefd%d&�Z d$edefd'd(�Z d$ed)e ddfd*d+�Z!d6d,d-�Z"edd.�de e defd/d0��Z#edd.�dede e deeef fd1d0��Z#ddd2�dede e defd3d0�Z#dS )7�
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.
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_typez*[prompt.invalid]Please enter a valid valuezA[prompt.invalid.choice]Please select one of the available optionsz: N�choices� FT��console�passwordr# �case_sensitive� show_default� show_choices�promptr& r' r( r) r* r c C sT |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 r r 6 s ��
zPromptBase.__init__)r& r' r# r( r) r* �stream�defaultr1 c
C � d S r r )
�clsr+ r& r' r# r( r) r* r2 r1 r r r �askN s zPromptBase.askc C r3 r r ) r4 r+ r& r' r# r( r) r* r1 r r r r5 _ s .)r& r' r# r( r) r* r2 r1 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.
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.
r% �r2 r1 r ) r4 r+ r& r' r# r( r) r* r2 r1 �_promptr r r r5 o 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 r2 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 r2 r+ Z_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' r1 )�input)r4 r& r+ r' r1 r r r � get_input� s zPromptBase.get_input�valuec C s> | j dusJ �| jr|�� | j v S |�� �� dd� | j D �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.
Nc S � g | ]}|� � �qS r ��lower��.0�choicer r r �
<listcomp>� � z+PromptBase.check_choice.<locals>.<listcomp>)r# r( �striprL �r rI r r r � check_choice� s zPromptBase.check_choicec C s~ |� � }z| �|�}W n ty t| j��w | jdur=| �|�s't| j��| js=| �| jdd� | jD �� |�
� � �}|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.
Nc S rJ r rK rM r r r rP � rQ z/PromptBase.process_response.<locals>.<listcomp>) rR r" �
ValueErrorr �validate_error_messager# rT �illegal_choice_messager( �indexrL )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 r[ 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�r1 c C r3 r r )r r1 r r r �__call__ s zPromptBase.__call__c C r3 r r )r r2 r1 r r r r` s r6 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.
Tr_ r$ .N)r^ rF rH r& r' rZ r r] )r r2 r1 r+ rI rY r[ r r r r` s
� ��)r$ r )r N)$r r r r r/ r" �type�__annotations__rV rW rD r# r r r r �boolr � classmethodr r r r r r5 r r
r; rF rH rT rZ r r] r^ r` 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 re 0 s re 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" rV r r r r rg < � rg c @ rf )� 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" rV r r r r rj H ri rj 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# r2 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.r8 r9 r: r, )r# r
)r r2 �yes�nor r r r; a s
*zConfirm.render_defaultrI c C s. |� � �� }|| jvrt| j��|| jd kS )zConvert choices to a bool.r )rR rL r# r rV rS r r r rZ f s
zConfirm.process_responseN)r r r r rc r" rV r# r r/ rb r r
r; rZ r r r r rl T s
rl �__main__)r\ 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 fruit)�apple�orange�pear)r# zfruit=z'What's the best Dog? (Case INSENSITIVE))zBorder TerrierZCollieZ LabradoodleF)r# r( zdoggie=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/ re rh rg rk rj rc rl r �richr\ r5 �resultr' �lenZfruitZdoggier r r r �<module> s^ ( ��� �� �