h0ddlmZmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZmZedZedZGdd eZGd d eZGd d eeZGddeeZGddeeZGddeeZGddeeZedk(rddlm Z ejCddr ejCddZ"e"dk\re"dkrn e d'e de" ejCd d!Z#e$e#dk\rn e d"(e d#e#ejCd$gd%&Z%e d'e%ejCd(gd)d*+Z&e d,e&y.e d-y.y.)/)AnyGenericListOptionalTextIOTypeVarUnionoverload) get_console)Console)TextTextType PromptType DefaultTypeceZdZdZy) PromptErrorz/Exception base class for prompt related errors.N)__name__ __module__ __qualname____doc__U/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pip/_vendor/rich/prompt.pyrr s9rrc,eZdZdZdeddfdZdefdZy)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||_yNr)selfrs r__init__zInvalidResponse.__init__s  rc|jSr r!r"s r__rich__zInvalidResponse.__rich__s ||r)rrrrrr#r&rrrrrs&T(rrceZdZUdZeZeed<dZdZ dZ dZ e e eed< d)dddd d d d d ed e ed ede e ededededdfdZee d)dddd d d ddd ed e ed ede e ededededede edeeeffdZee d)dddd d d ddd ed e ed ede e edededede edefdZe d)dddd d d dddd ed e ed ede e ededededede edefdZdedefdZdedefdZe d*d ed ed ede edef dZdedefdZdedefd Zded!e ddfd"Z!d+d#Z"edd$de edefd%Z#edd$dede edeeeffd&Z#ddd'dede edefd(Z#y), 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/r0rc|xs t|_t|trt j |dn||_||_|||_||_ ||_ ||_ y)Nr1style) r r, isinstancestrr from_markupr1r-r*r.r/r0)r"r1r,r-r*r.r/r0s rr#zPromptBase.__init__6sh/+- &#&   V8 4 !  "DL,((r)r,r-r*r.r/r0streamdefaultr8cyr r) clsr1r,r-r*r.r/r0r9r8s raskzPromptBase.askNs rcyr r) r;r1r,r-r*r.r/r0r8s rr<zPromptBase.ask_s r.)r,r-r*r.r/r0r9r8c 4||||||||} | || S)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+r9r8r) r;r1r,r-r*r.r/r0r9r8_prompts rr<zPromptBase.askos2: )%% wv66rc"td|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)r"r9s rrender_defaultzPromptBase.render_defaultsayN$455rc|jj}d|_|jrP|jrDdj |j}d|d}|j d|j |d|dk7r[|jrOt|t|jfr3|j d|j|}|j ||j |j|S)zMake prompt text. Args: default (DefaultType): Default value. Returns: Text: Text to display in prompt. /[] zprompt.choices.) r1copyendr0r*joinappendr/r5r6r)rE prompt_suffix)r"r9r1_choicesr*_defaults r make_promptzPromptBase.make_prompts!!#   xx -H(1oG MM#  MM'#3 4 sN!!7S$*<*<$=> MM# **73H MM( # d(() rc*|j|||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-r8)input)r;r,r1r-r8s r get_inputzPromptBase.get_inputs$}}Vhv}FFrvaluec|jJ|jr|j|jvS|jj|jDcgc]}|jc}vScc}w)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. )r*r.striplower)r"rWchoices r check_choicezPromptBase.check_choicesd||'''   ;;=DLL0 0{{}""$dll(SF(SSS(Ss"A=c|j} |j|}|j |j |st|j|jsa|j|j |j Dcgc]}|jc}j|j}|S#t$rt|jwxYwcc}w)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. ) rYr) ValueErrorrvalidate_error_messager*r\illegal_choice_messager.rZindex)r"rW return_valuer[s rprocess_responsezPromptBase.process_responses  ?'+'9'9%'@L << #$$U+%d&A&ABB&&#11LL6:llCFCII%++-X   ?!$"="=> > ?DsCC& C#errorc:|jj|y)zCalled to handle validation error. Args: value (str): String entered by user. error (InvalidResponse): Exception instance the initiated the error. N)r,print)r"rWrds ron_validate_errorzPromptBase.on_validate_errors 5!rcy)z,Hook to display something before the prompt.Nrr%s r pre_promptzPromptBase.pre_prompt srr8cyr r)r"r8s r__call__zPromptBase.__call__s rcyr r)r"r9r8s rrlzPromptBase.__call__s rr?c( |j|j|}|j|j||j|}|dk(r|dk7r|S |j |}|S#t $r}|j||Yd}~d}~wwxYw)zRun the prompt loop. Args: default (Any, optional): Optional default value. Returns: PromptType: Processed value. rjrG.N)rirSrVr,r-rcrrg)r"r9r8r1rWrbrds rrlzPromptBase.__call__s OO %%g.FNN4<<vNVE{w#~ $#44U; $# # &&ue4 sA,, B5B  B)rGr )rN)$rrrrr6r)type__annotations__r_r`rPr*rrrr boolr# classmethodr rrr rr<rrrErSrVr\rcrrgrirlrrrr(r(s M4IKM#'GXd3i ')&*'+#!!))'" )  ) $s)$ )))) )0   &*'+#!!#'    '"    $s)$              {J& '       &*'+#!!#'    '"    $s)$                %7&*'+#!!#'%7%7'" %7  %7 $s)$ %7%7%7%7%7 %7 %7%7N 6k 6d 6;4> $( GGG G  G GG& T# T$ Tcj>"s"?"t";59 (6"2 j  BF % /7/? z;& '  *-$3$hv6F$RU$rr(ceZdZdZeZy)PromptzbA prompt that returns a str. Example: >>> name = Prompt.ask("Enter your name") N)rrrrr6r)rrrrtrt0sMrrtceZdZdZeZdZy) 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)r_rrrrvrv<sMRrrvceZdZdZeZdZy) FloatPromptzyA prompt that returns a float. Example: >>> temperature = FloatPrompt.ask("Enter desired temperature") z%[prompt.invalid]Please enter a numberN)rrrrfloatr)r_rrrryryHsMDrrycReZdZUdZeZdZddgZee e d<de de fdZ d e defd Zy ) ConfirmzuA yes / no confirmation prompt. Example: >>> if Confirm.ask("Continue"): run_job() z#[prompt.invalid]Please enter Y or Nynr*r9rc^|j\}}t|r d|ddSd|ddS)z8Render the default as (y) or (n) rather than True/False.rBrCrDr3)r*r)r"r9yesnos rrEzConfirm.render_defaultas:,,R'auAJ@PQQ2$ay@PQQrrWc|jj}||jvrt|j||jdk(S)zConvert choices to a bool.r)rYrZr*rr_)r"rWs rrczConfirm.process_responsefsF ##%  $!$"="=> > Q''rN)rrrrrqr)r_r*rr6rprrrErcrrrr|r|TsOMBsGT#Y#RkRdR (c(d(rr|__main__)rfzRun [i]prompt[/i] tests?T)r9z6: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)'typingrrrrrrr r rGr r,r textrrrr Exceptionrrr(r6rtrwrvrzryrqr|rpip._vendor.richrfr<resultr-lenfruitdoggierrrrsQQQ \ " m$ :): k O$$O$d Z_  S 3 S E*U# E(j(4 z&{{-t{<]]HRS#F{v| Q R  x !zzO"H8}! 6 7  (&' ?4O P ui ! 5?   z"# *+Er