관리-도구
편집 파일: prompt.cpython-38.pyc
U &?�f(, � @ 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 �E/opt/hc_python/lib/python3.8/site-packages/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 � 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�} | ||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& �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|� 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|_| 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# 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 |||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)r3 r' r+ r( r1 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 rI 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)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 )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 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_prompt�r1 )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 )z�Run 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 )� 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# 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 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$ r6 c C s. | j \}}t|r d|� d�n d|� d�dd�S )z8Render the default as (y) or (n) rather than True/False.r7 r8 r9 r, )r$ r )r r2 �yes�nor 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 fruit�appleZorange�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 r` r[ rb r �pip._vendor.richrT r4 �resultr( �lenZfruitr r r r �<module> sH ( � �