U NSh0@sddlmZmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZmZedZedZGdd d eZGd d d eZGd d d eeZGdddeeZGdddeeZGdddeeZGdddeeZedkrddlm Z ej!dddrej!dddZ"e"dkr0e"dkr0q||_||_ ||_ ||_ dS)Nr,style) r r' isinstancestrr from_markupr,r(r$r)r*r+)rr,r'r(r$r)r*r+rrrr6s zPromptBase.__init__)r'r(r$r)r*r+stream) r,r'r(r$r)r*r+defaultr2rc CsdSrr) clsr,r'r(r$r)r*r+r3r2rrraskNszPromptBase.ask) r,r'r(r$r)r*r+r2rc CsdSrr) r4r,r'r(r$r)r*r+r2rrrr5_s.)r'r(r$r)r*r+r3r2c 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. 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&r3r2r) r4r,r'r(r$r)r*r+r3r2Z_promptrrrr5os r3rcCstd|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 )rr3rrrrender_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/r0r#r; prompt_suffix)rr3r,Z_choicesr$_defaultrrr make_prompts$          zPromptBase.make_prompt)r'r,r(r2rcCs|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(r2)input)r4r'r,r(r2rrr get_inputszPromptBase.get_inputvaluercCs>|jdk st|jr"||jkS|dd|jDkS)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. NcSsg|] }|qSrlower.0choicerrr sz+PromptBase.check_choice..)r$AssertionErrorr)striprLrrJrrr check_choices zPromptBase.check_choicecCs|}z||}Wntk r4t|jYnX|jdk r||sTt|j|js||jdd|jD | }|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. NcSsg|] }|qSrrKrMrrrrPsz/PromptBase.process_response..) rRr# ValueErrorrvalidate_error_messager$rTillegal_choice_messager)indexrL)rrJ return_valuerrrprocess_responses    zPromptBase.process_response)rJerrorrcCs|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)rrJr[rrron_validate_errorszPromptBase.on_validate_errorrcCsdS)z,Hook to display something before the prompt.Nrr rrr pre_prompt szPromptBase.pre_promptr2)r2rcCsdSrr)rr2rrr__call__szPromptBase.__call__)r3r2rcCsdSrr)rr3r2rrrr`sr6c 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. r_r%.N)r^rFrHr'r(rZrr])rr3r2r,rJrYr[rrrr`s   )r%)r%)r%)r%)N)$rrrrr0r#type__annotations__rVrWrDr$rrrr boolr classmethodr rrrrr5rr r;rFrHrTrZrr]r^r`rrrrr"s        '   r"c@seZdZdZeZdS)PromptzbA prompt that returns a str. Example: >>> name = Prompt.ask("Enter your name") N)rrrrr0r#rrrrre0srec@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#rVrrrrrf<srfc@seZdZdZeZdZdS) FloatPromptzyA prompt that returns a float. Example: >>> temperature = FloatPrompt.ask("Enter desired temperature") z%[prompt.invalid]Please enter a numberN)rrrrfloatr#rVrrrrrhHsrhc@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$r7cCs.|j\}}t|rd|dn d|dddS)z8Render the default as (y) or (n) rather than True/False.r8r9r:r-)r$r )rr3yesnorrrr;as zConfirm.render_defaultrIcCs.|}||jkr t|j||jdkS)zConvert choices to a bool.r)rRrLr$rrVrSrrrrZfs   zConfirm.process_responseN)rrrrrcr#rVr$rr0rbrr r;rZrrrrrjTs rj__main__)r\zRun [i]prompt[/i] tests?T)r3z6: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'What's the best Dog? (Case INSENSITIVE)zBorder TerrierZCollieZ LabradoodleF)r$r)zdoggie=z[b]OK :loudly_crying_face:N)'typingrrrrrrrr r%r r'r textr rrr Exceptionrrr"r0rergrfrirhrcrjrpip._vendor.richr\r5resultr(lenZfruitZdoggierrrrsT(