} d dl mZmZmZmZmZmZmZmZ ddl m
Z
ddl m Z ddl
mZmZ ed Z ed Z G d d e Z G d
d e Z G d d
ee Z G d dee Z G d dee Z G d dee Z G d dee Z e dk rd dl m Z e ! dd r e! dd Z"e"dk re"d k rn e d 0 e d e" e! d d! Z# e$e# d k rn e d" 3 e d#e# e! d$g d%& Z% e d'e% e! d(g d)d*+ Z& e d,e& d.S e d- d.S d.S )/ )AnyGenericListOptionalTextIOTypeVarUnionoverload ) get_console)Console)TextTextType
PromptType DefaultTypec e Zd ZdZdS ) PromptErrorz/Exception base class for prompt related errors.N)__name__
__module__ __qualname____doc__ w/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/pip/_vendor/rich/prompt.pyr r s 9999r r c . e Zd ZdZdeddfdZdefdZdS )InvalidResponsezException 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.
messagereturnNc || _ d S Nr )selfr s r __init__zInvalidResponse.__init__ s
r c | j S r r! r" s r __rich__zInvalidResponse.__rich__ s
|r )r r r r r r# r& r r r r r s[ T ( r r c e Zd ZU dZeZeed< dZdZ dZ
dZ e e
e ed< d*dd dd
d
d
d d ed
e e dede e
e dedededdfdZee d*dd dd
d
d
ddd ed
e e dede e
e dedededede e deeef fd Zee d*dd dd
d
d
ddd ed
e e dede e
e dededede e defd Ze d*dd dd
d
d
dddd ed
e e dede e
e dedededede e defd Zdedefd Zdedefd Z e d+d
ed edede e def
d Z d edefd Z d edefd!Z d ed"e ddfd#Z!d,d$Z"edd%de e defd& Z#edd%dede e deeef fd' Z#ddd(dede e defd)Z#dS )-
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: Nchoices FTconsolepasswordr* case_sensitive show_default show_choicespromptr- r. r/ r0 r1 r c |p
t | _ t |t rt j |d n|| _ || _ ||| _ || _ || _
|| _ d S )Nr2 style) r r-
isinstancestrr from_markupr2 r. r* r/ r0 r1 )r" r2 r- r. r* r/ r0 r1 s r r# zPromptBase.__init__6 s{ /+-- &#&&
D V8 4 4 4 4
!
"DL , ( (r )r- r. r* r/ r0 r1 streamdefaultr9 c d S r r )
clsr2 r- r. r* r/ r0 r1 r: r9 s
r askzPromptBase.askN s r c d S r r ) r< r2 r- r. r* r/ r0 r1 r9 s r r= zPromptBase.ask_ s r .)r- r. r* r/ r0 r1 r: r9 c @ | ||||||| }
|
|| 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, r: r9 r ) r< r2 r- r. r* r/ r0 r1 r: r9 _prompts r r= zPromptBase.asko sF : # )%%
wwv6666r c * t d| dd S )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" r: s r render_defaultzPromptBase.render_default s NNNN$4555r c | j }d|_ | j rR| j rKd | j }d| d}| d | |d |dk rb| j r[t |t | j
f r?| d | | }| | | | j |S )zMake prompt text.
Args:
default (DefaultType): Default value.
Returns:
Text: Text to display in prompt.
r+ /[] zprompt.choices.)
r2 copyendr1 r* joinappendr0 r6 r7 r) rF
prompt_suffix)r" r: r2 _choicesr* _defaults r make_promptzPromptBase.make_prompt s !!##
5 5xx --H%(oooG MM# MM'#3 4 4 4
sNN!
7S$*<$=>>
MM# **733H MM( # # #
d()))
r c 2 | ||| S )zGet input from user.
Args:
console (Console): Console instance.
prompt (TextType): Prompt text.
password (bool): Enable password entry.
Returns:
str: String from user.
)r. r9 )input)r< r- r2 r. r9 s r get_inputzPromptBase.get_input s $ }}Vhv}FFFr valuec | j J | j r| | j v S | d | j D v S )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.
Nc 6 g | ]}| S r lower.0choices r
z+PromptBase.check_choice.. s (S(S(SF (S(S(Sr )r* r/ stripr[ r" rW s r check_choicezPromptBase.check_choice sb |''' 1;;==DL0 0{{}}""$$(S(Sdl(S(S(SSSr c | } | | }n"# t $ r t | j w xY w| j | | st | j | j sT| | j d | j D |
}|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 6 g | ]}| S r rZ r\ s r r_ z/PromptBase.process_response.. s CCCF CCCr ) r` r)
ValueErrorr validate_error_messager* rb illegal_choice_messager/ indexr[ )r" rW return_values r process_responsezPromptBase.process_response s
?'+'9'9%'@'@LL ? ? ?!$"=>> > ? < #$$U++
C%d&ABBB&
# 1 1LCCdlCCCII%++--XX
s , A errorc : | 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" rW rk s r on_validate_errorz PromptBase.on_validate_error s
5!!!!!r c dS )z,Hook to display something before the prompt.Nr r% s r
pre_promptzPromptBase.pre_prompt s r r9 c d S r r )r" r9 s r __call__zPromptBase.__call__ s r c d S r r )r" r: r9 s r rs zPromptBase.__call__ s r r@ c B | | | }| | j || j | }|dk r|dk r|S | | }|S # t $ r }| || Y d}~d}~ww xY w)zRun the prompt loop.
Args:
default (Any, optional): Optional default value.
Returns:
PromptType: Processed value.
Trq r+ .N)rp rS rV r- r. rj r rn )r" r: r9 r2 rW ri rk s r rs zPromptBase.__call__ s $ OO %%g..FNN4< vNVVE{{w#~~
$ # 4 4U ; ;
$# #
&&ue444
s A4 4
B >BB )r+ r )r N)$r r r r r7 r) type__annotations__rf rg rP r* r r r r
boolr# classmethodr
r r r r r= r r rF rS rV rb rj r rn rp rs r r r r( r( sX M4 IK M#'GXd3i
''' ) &* '+ # ! !) ) )) '" )
) $s)$
) ) ) )
) ) ) )0
&* '+ # ! !#'
'"
$s)$
{J& '
X [
&* '+ # ! !#' '"
$s)$
X [ %7 &* '+ # ! !#'%7 %7 %7%7 '" %7
%7 $s)$
%7 %7 %7 %7 %7 %7
%7 %7 %7 [%7N 6k 6d 6 6 6 6 ; 4 > $( G GG G G
G
G G G [G& T# T$ T T T T c j >"s "? "t " " " "; ; ; ; 59 (6"2 j X BF % /7/? z;& ' X
*- $ $ $3 $hv6F $RU $ $ $ $ $ $r r( c e Zd ZdZeZdS )PromptzbA prompt that returns a str.
Example:
>>> name = Prompt.ask("Enter your name")
N)r r r r r7 r) r r r r{ r{ 0 s MMMr r{ c 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) rf r r r r} r} < s' M Rr r} c 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) rf r r r r r H s' M Dr r c Z e Zd ZU dZeZdZddgZee e
d< de de fdZ
d e defd
Zd S ) ConfirmzuA yes / no confirmation prompt.
Example:
>>> if Confirm.ask("Continue"):
run_job()
z#[prompt.invalid]Please enter Y or Nynr* r: r c P | j \ }}t |rd| dnd| dd S )z8Render the default as (y) or (n) rather than True/False.rC rD rE r4 )r* r )r" r: yesnos r rF zConfirm.render_defaulta s: ,R'8JJJJJy2yyy@PQQQQr rW c | }|| j vrt | j || j d k S )zConvert choices to a bool.r )r` r[ r* r rf ra s r rj zConfirm.process_responsef sJ
##%% $ $!$"=>> > Q''r N)r r r r rx r) rf r* r r7 rw r r rF rj r r r r r T s M B sGT#Y###Rk Rd R R R R
(c (d ( ( ( ( ( (r r __main__)rm zRun [i]prompt[/i] tests?T)r: 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)appleorangepear)r* zfruit=z'What's the best Dog? (Case INSENSITIVE))zBorder TerrierCollie 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( r7 r{ r~ r} r r rx r r pip._vendor.richrm r= resultr. lenfruitdoggier r r r s Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q
W\
"
"
gm$$ : : : : :) : : : k O$ O$ O$ O$ O$$ O$ O$ O$d Z_ S S S S S
3 S S S E E E E E*U# E E E( ( ( ( (j ( ( (4 z&&&&&&{{-t{<< , S ]]HRS # F {{v|| EQ R R R
S !!! 8 zzO " H s8}}!! E6 7 7 7 8 &(&&'''
?4O4O4O
PP
u !!! 5???
"""##### *+++++E r