U 祡c,@sddlmZmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZmZedZedZGdd d eZGd d d eZGd d d eeZGdddeeZGdddeeZGdddeeZGdddeeZedkrddlmZej dddrej dddZ!e!dkr0e!dkr0q||_||_ ||_ dS)Nr+style) r r' isinstancestrr Z 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*defaultr0rc CsdSrr) clsr+r'r(r$r)r*r1r0rrraskKszPromptBase.ask)r+r'r(r$r)r*r0rcCsdSrr)r2r+r'r(r$r)r*r0rrrr3[s .)r'r(r$r)r*r1r0c 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&r1r0r) r2r+r'r(r$r)r*r1r0Z_promptrrrr3jsr1rcCstd|ddS)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 )rr1rrrrender_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/r#r9 prompt_suffix)rr1r+Z_choicesr$_defaultrrr make_prompts$          zPromptBase.make_prompt)r'r+r(r0rcCs|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(r0)input)r2r'r+r(r0rrr get_inputszPromptBase.get_inputvaluercCs|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$AssertionErrorstriprrHrrr check_choices zPromptBase.check_choicecCsX|}z||}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)rJr# ValueErrorrvalidate_error_messager$rLillegal_choice_message)rrH return_valuerrrprocess_responses  zPromptBase.process_response)rHerrorrcCs|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)rrHrRrrron_validate_errorszPromptBase.on_validate_errorrcCsdS)z,Hook to display something before the prompt.Nrr rrr pre_promptszPromptBase.pre_promptr0)r0rcCsdSrr)rr0rrr__call__szPromptBase.__call__)r1r0rcCsdSrr)rr1r0rrrrWsr4c Cs|||}|j|j||j|d}|dkr<|dkr<|Sz||}Wn6tk r}z|||WYqW5d}~XYqX|SqdS)zRun the prompt loop. Args: default (Any, optional): Optional default value. Returns: PromptType: Processed value. rVr%.N)rUrDrFr'r(rQrrT)rr1r0r+rHrPrRrrrrWs   )r%)r%)r%)r%)N)$rrrrr/r#type__annotations__rNrOrBr$rrrr boolr classmethodr rrrrr3rr r9rDrFrLrQrrTrUrWrrrrr"s         $    r"c@seZdZdZeZdS)PromptzbA prompt that returns a str. Example: >>> name = Prompt.ask("Enter your name") N)rrrrr/r#rrrrr\sr\c@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)rrrrintr#rNrrrrr]*sr]c@seZdZdZeZdZdS) FloatPromptzyA prompt that returns a float. Example: >>> temperature = FloatPrompt.ask("Enter desired temperature") z%[prompt.invalid]Please enter a numberN)rrrrfloatr#rNrrrrr_6sr_c@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$r5cCs.|j\}}t|rd|dn d|dddS)z8Render the default as (y) or (n) rather than True/False.r6r7r8r,)r$r )rr1yesnorrrr9Os zConfirm.render_defaultrGcCs.|}||jkr t|j||jdkS)zConvert choices to a bool.r)rJlowerr$rrNrKrrrrQTs   zConfirm.process_responseN)rrrrrZr#rNr$rr/rYrr r9rQrrrrraBs ra__main__)rSzRun [i]prompt[/i] tests?T)r1z6: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 fruitZappleZorangeZpear)r$zfruit=z[b]OK :loudly_crying_face:N)%typingrrrrrrrr r%r r'r textr rrr Exceptionrrr"r/r\r^r]r_rZrarZrichrSr3resultr(lenZfruitrrrrsH(