B od',@sddlmZmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZmZedZedZGdd d eZGd d d eZGd d d eeZGdddeeZGdddeeZGdddeeZGdddeeZedkrddlmZej dddrx2ej dddZ!e!dkr0e!dkr0Pedq Wede!x,ej d dd!Z"e#e"dkrlPed"qNWed#e"ej d$d%d&d'gd(Z$ed)e$ned*d+S),)AnyGenericListOptionalTextIOTypeVarUnionoverload) get_console)Console)TextTextType PromptType DefaultTypec@seZdZdZdS) PromptErrorz/Exception base class for prompt related errors.N)__name__ __module__ __qualname____doc__rrI/var/www/html/venv/lib/python3.7/site-packages/pip/_vendor/rich/prompt.pyr src@s.eZdZdZeddddZedddZdS) 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. N)messagereturncCs ||_dS)N)r)selfrrrr__init__szInvalidResponse.__init__)rcCs|jS)N)r)rrrr__rich__szInvalidResponse.__rich__)rrrrrrrrrrrrsrc@s4eZdZUdZeZeed<dZdZ dZ dZ e e eed<d3dd dd d d ee eee e eeedd d dZeed4dd dd d ddee eee e eeeee eeeefd ddZeed5dd dd d ddee eee e eeee eedddZed6dd dd d dddee eee e eeeee eed ddZeedddZeedddZed7eeee eedddZeed d!d"Zeed d#d$Zee dd%d&d'Z!dd(d)d*Z"edd+e eed,d-d.Z#edd+ee eeeefd/d0d.Z#ddd1ee eed/d2d.Z#dS)8 PromptBaseaAsk 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: NchoicesFT)consolepasswordr show_default show_choices)promptr"r#r r$r%rcCsN|pt|_t|tr$tj|ddn||_||_|dk r>||_||_ ||_ dS)Nr&)style) r r" isinstancestrr from_markupr&r#r r$r%)rr&r"r#r r$r%rrrr5s zPromptBase.__init__)r"r#r r$r%stream) r&r"r#r r$r%defaultr+rc CsdS)Nr) clsr&r"r#r r$r%r,r+rrraskKszPromptBase.ask)r&r"r#r r$r%r+rcCsdS)Nr)r-r&r"r#r r$r%r+rrrr.[s .)r"r#r r$r%r,r+c Cs |||||||d} | ||dS)aShortcut 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"r#r r$r%)r,r+r) r-r&r"r#r r$r%r,r+Z_promptrrrr.js)r,rcCstd|ddS)zTurn the supplied default in to a Text instance. Args: default (DefaultType): Default value. Returns: Text: Text containing rendering of default value. ()zprompt.default)r )rr,rrrrender_defaults zPromptBase.render_defaultcCs|j}d|_|jrJ|jrJd|j}d|d}|d||d|dkr|jrt|t |j fr|d| |}||||j |S)zMake prompt text. Args: default (DefaultType): Default value. Returns: Text: Text to display in prompt. r!/[] zprompt.choices.) r&copyendr%r joinappendr$r(r)rr1 prompt_suffix)rr,r&Z_choicesr _defaultrrr make_prompts          zPromptBase.make_prompt)r"r&r#r+rcCs|j|||dS)zGet input from user. Args: console (Console): Console instance. prompt (TextType): Prompt text. password (bool): Enable password entry. Returns: str: String from user. )r#r+)input)r-r"r&r#r+rrr get_inputszPromptBase.get_input)valuercCs|jdk st||jkS)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. N)r AssertionErrorstrip)rr?rrr check_choices zPromptBase.check_choicecCsX|}y||}Wntk r4t|jYnX|jdk rT||sTt|j|S)aProcess 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)rAr ValueErrorrvalidate_error_messager rBillegal_choice_message)rr? return_valuerrrprocess_responses  zPromptBase.process_response)r?errorrcCs|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)rr?rHrrron_validate_errorszPromptBase.on_validate_error)rcCsdS)z,Hook to display something before the prompt.Nr)rrrr pre_promptszPromptBase.pre_prompt)r+)r+rcCsdS)Nr)rr+rrr__call__szPromptBase.__call__)r,r+rcCsdS)Nr)rr,r+rrrrLs)r,r+c Csx|||}|j|j||j|d}|dkr>|dkr>|Sy||}Wn0tk r|}z|||wWdd}~XYqX|SqWdS)zRun the prompt loop. Args: default (Any, optional): Optional default value. Returns: PromptType: Processed value. )r+r!.N)rKr<r>r"r#rGrrJ)rr,r+r&r?rFrHrrrrLs   )r!)r!)r!)r!)N)$rrrrr)rtype__annotations__rDrEr:r rrrr boolr classmethodr rrrrr.rr r1r<r>rBrGrrJrKrLrrrrrst   &$&    rc@seZdZdZeZdS)PromptzbA prompt that returns a str. Example: >>> name = Prompt.ask("Enter your name") N)rrrrr)rrrrrrQsrQc@seZdZdZeZdZdS) 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)rrrrintrrDrrrrrR*srRc@seZdZdZeZdZdS) FloatPromptzyA prompt that returns a float. Example: >>> temperature = FloatPrompt.ask("Enter desired temperature") z%[prompt.invalid]Please enter a numberN)rrrrfloatrrDrrrrrT6srTc@sNeZdZUdZeZdZddgZee e d<e e dddZ 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 Nynr )r,rcCs.|j\}}t|rd|dn d|dddS)z8Render the default as (y) or (n) rather than True/False.r/r0zprompt.default)r')r r )rr,yesnorrrr1Os zConfirm.render_default)r?rcCs.|}||jkr t|j||jdkS)zConvert choices to a bool.r)rAlowerr rrD)rr?rrrrGTs   zConfirm.process_responseN)rrrrrOrrDr rr)rNrr r1rGrrrrrVBs rV__main__)rIzRun [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 fruitappleZorangepear)r zfruit=z[b]OK :loudly_crying_face:N)%typingrrrrrrrr r!r r"r textr rrr Exceptionrrrr)rQrSrRrTrOrVrpip._vendor.richrIr.resultr#lenZfruitrrrrsF(