o Rh0@sddlmZmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZmZedZedZGdd d eZGd d d eZGd d d eeZGdddeeZGdddeeZGdddeeZGdddeeZedkrddlm Z ej!dddr ej!dddZ"e"dkre"dkrne dqe de" ej!d dd!Z#e$e#dkrne d"qe d#e#ej!d$gd%d&Z%e d'e%ej!d(gd)d*d+Z&e d,e&d.Se d-d.Sd.S)/)AnyGenericListOptionalTextIOTypeVarUnionoverload) get_console)Console)TextTextType PromptType DefaultTypec@seZdZdZdS) PromptErrorz/Exception base class for prompt related errors.N)__name__ __module__ __qualname____doc__rrM/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/rich/prompt.pyr src@s0eZdZdZdeddfddZdefddZdS) 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. messagereturnNcCs ||_dSNr)selfrrrr__init__s zInvalidResponse.__init__cCs|jSrrrrrr__rich__szInvalidResponse.__rich__)rrrrrrr rrrrrsrc@seZdZUdZeZeed<dZdZ dZ dZ e e eed< d4dd dd d d d d ed e edede e ededededdfddZee d4dd dd d d ddd ed e edede e ededededede edeeeffddZee d4dd dd d d ddd ed e edede e edededede edefddZe d4dd dd d d dddd ed e edede e ededededede edefddZdedefddZdedefd d!Ze d5d ed edede edef d"d#Zd$edefd%d&Zd$edefd'd(Zd$ed)e ddfd*d+Z!d6d,d-Z"edd.de edefd/d0Z#edd.dede edeeeffd1d0Z#ddd2dede edefd3d0Z#dS)7 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. 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: NchoicesFTconsolepasswordr#case_sensitive show_default show_choicespromptr&r'r(r)r*rcCsT|pt|_t|trtj|ddn||_||_|dur||_||_ ||_ ||_ 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*streamdefaultr1c CdSrr) clsr+r&r'r#r(r)r*r2r1rrraskNszPromptBase.askc Cr3rr) r4r+r&r'r#r(r)r*r1rrrr5_s.)r&r'r#r(r)r*r2r1c 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%r2r1r) r4r+r&r'r#r(r)r*r2r1_promptrrrr5os cCstd|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 )rr2rrrrender_defaults zPromptBase.render_defaultcCs|j}d|_|jr%|jr%d|j}d|d}|d||d|dkrC|jrCt|t |j frC|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"r; prompt_suffix)rr2r+_choicesr#_defaultrrr make_prompts"          zPromptBase.make_promptcCs|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'r1)input)r4r&r+r'r1rrr get_inputszPromptBase.get_inputvaluecCs>|jdusJ|jr||jvS|dd|jDvS)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. NcSg|]}|qSrlower.0choicerrr z+PromptBase.check_choice..)r#r(striprMrrJrrr check_choices zPromptBase.check_choicecCs~|}z||}Wn tyt|jw|jdur=||s't|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. NcSrKrrLrNrrrrQrRz/PromptBase.process_response..) rSr" ValueErrorrvalidate_error_messager#rUillegal_choice_messager(indexrM)rrJ return_valuerrrprocess_responses       zPromptBase.process_responseerrorcCs|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_errorcCsdS)z,Hook to display something before the prompt.Nrrrrr pre_prompt szPromptBase.pre_promptr1cCr3rr)rr1rrr__call__szPromptBase.__call__cCr3rr)rr2r1rrrrasr6c Cs |||}|j|j||j|d}|dkr|dkr|Sz||}W|Sty?}z |||WYd}~qd}~ww)zRun the prompt loop. Args: default (Any, optional): Optional default value. Returns: PromptType: Processed value. Tr`r$.N)r_rGrIr&r'r[rr^)rr2r1r+rJrZr\rrrras     )r$r)rN)$rrrrr/r"type__annotations__rWrXrDr#rrrr boolr classmethodr rrrrr5rr r;rGrIrUr[rr^r_rarrrrr!s<                   '     &r!c@seZdZdZeZdS)PromptzbA prompt that returns a str. Example: >>> name = Prompt.ask("Enter your name") N)rrrrr/r"rrrrrf0srfc@eZdZdZeZdZdS) 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"rWrrrrrh<rhc@rg) FloatPromptzyA prompt that returns a float. Example: >>> temperature = FloatPrompt.ask("Enter desired temperature") z%[prompt.invalid]Please enter a numberN)rrrrfloatr"rWrrrrrkHrjrkc@sReZdZUdZeZdZddgZee e d<de de fdd Z d e defd d Zd S)ConfirmzuA yes / no confirmation prompt. Example: >>> if Confirm.ask("Continue"): run_job() z#[prompt.invalid]Please enter Y or Nynr#r2rcCs4|j\}}t|rd|dddSd|dddS)z8Render the default as (y) or (n) rather than True/False.r8r9r:r,)r#r )rr2yesnorrrr;as *zConfirm.render_defaultrJcCs.|}||jvrt|j||jdkS)zConvert choices to a bool.r)rSrMr#rrWrTrrrr[fs   zConfirm.process_responseN)rrrrrdr"rWr#rr/rcrr r;r[rrrrrmTs rm__main__)r]zRun [i]prompt[/i] tests?T)r2z6: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)'typingrrrrrrrr r$r r&r textr rrr Exceptionrrr!r/rfrirhrlrkrdrmrrichr]r5resultr'lenfruitdoggierrrrs^(