tni4dZddlZddlZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZmZddlmZmZddlmZdd lmZdd lmZdd lmZdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'erddl(m)Z)GddeZ*dS)z3Contains the base class :class:`.BaseSimplePrompt`.N)ABCabstractmethod) TYPE_CHECKINGAnyCallableDictListOptionalTupleUnioncast) EditingMode) Condition FilterOrBool) KeyBindingsKeyHandlerCallable)Keys)Style) Validator)INQUIRERPY_KEYBOARD_INTERRUPT)RequiredKeyNotFound)InquirerPyMessageInquirerPySessionResultInquirerPyStyleInquirerPyValidate get_style) KeyPressEventc#eZdZdZ d4d ed eed ed ed ededee dedee e ge fdee e ge fde dededededee ddf"dZ dZed eddfdZdedddfdZdedddfd Zededddfd!Zedeee ffd"Zejd5d#Zdd$d%eeefdede egeffd&Zed'eeefd(eeefdeeeeffd)Zede fd*Z ede fd+Z!d6deede fd,Z"d5d-Z#edefd.Z$edeee ffd/Z%e%jd0eee fddfd1Z%edeee ffd2Z&e&jd0eee fddfd3Z&dS)7BaseSimplePrompta The base class to create a simple terminal input prompt. Note: No actual :class:`~prompt_toolkit.application.Application` is created by this class. This class only creates some common interface and attributes that can be easily used by `prompt_toolkit`. To have a functional prompt, you'll at least have to implement the :meth:`.BaseSimplePrompt._run` and :meth:`.BaseSimplePrompt._get_prompt_message`. See Also: :class:`~InquirerPy.prompts.input.InputPrompt` NF? Invalid inputTMandatory promptmessagestylevi_modeqmarkamark instructionvalidateinvalid_message transformerfilterdefault wrap_linesraise_keyboard_interrupt mandatorymandatory_messagesession_resultreturnc*|_|_|pi_t|ts|n"t t|j_|_t| ts| n| j_tj |r|j ntj _ |_|_dddd_t#_d_| _| _| _|s"t/t1jddr t4jn t4j_t|t<r|_n!t=j |r|nd|d_t1jd|  _!tEfd _#d tHj%igd j#d d j#d gd did j#d gd_&dj'igdj(igdj)igd_*dS)NF)answeredresultskippedz class:inputINQUIRERPY_VI_MODEcdSNT)_s S/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/InquirerPy/base/simple.pyz+BaseSimplePrompt.__init__..fsDT)move_cursor_to_endINQUIRERPY_NO_RAISE_KBIcjSN) _raise_kbiselfsr>r?z+BaseSimplePrompt.__init__..ms r@keyzc-c)rHr-zc-dzc-z)answer interruptskipfunc)+ _mandatory_mandatory_message_result isinstancerr _message _instruction_defaultr from_dictdictr_style_qmark_amark_statusr_kb_lexer _transformer_filter _wrap_linesboolosgetenvrVIEMACS _editing_moder _validator from_callablerEr_is_rasing_kbirEnter_kb_maps _handle_enter_handle_interrupt _handle_skip_kb_func_lookup)rGr$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3s` r>__init__zBaseSimplePrompt.__init__5sZ&$"3%+ gx00 7GG(h((66 (%gx88 SGGggdl>S>S oE&Oejjy{{?OPP   $)TeLL ==# ' % #ry)=uEEFF #KNN"  h * * &DOO'5$8..#'DO !i %+C'C   ((?(?(?(?@@tz*+)<==$*=)=>>U^Ut?R>R%S%ST     234!4#9:;d/01  r@c fd}jD]W\}}t|ts|g}|D]7}||d|dt d|8XdS)aRegister all keybindings in `self._kb_maps`. It's required to call this function at the end of prompt constructor if it inherits from :class:`~InquirerPy.base.simple.BaseSimplePrompt` or :class:`~InquirerPy.base.complex.BaseComplexPrompt`. cjvrtddt|ts|g}j|d|ifd}dS)Nzkeybinding action z not foundr-cnjD]%}|d|g|dgR&dS)NrLargs)kb_func_lookupget)eventmethodactionrGs r>r=zABaseSimplePrompt._keybinding_factory.._factory.._sU"1&9CCF"F6N5B6::fb+A+ABBBBBCCr@)rsrrPlist register_kb)keysr-rwr=rGs ` r>_factoryz6BaseSimplePrompt._keybinding_factory.._factorysT000)*Qv*Q*Q*QRRRdD)) v T t 3F 3 3 C C C C C4 3 C C Cr@rHr-cdSr;r<r<r@r>r?z6BaseSimplePrompt._keybinding_factory..str@N)kb_mapsitemsrPrxrtr)rGr{rHitemkbs` r>_keybinding_factoryz$BaseSimplePrompt._keybinding_factory}s C C C C C++-- T TICdD)) v T TEBFF8Y||5L5L$M$MsSSSS T T Tr@cdS)zhSet the error message for the prompt. Args: message: Error message to set. Nr<)rGr$s r> _set_errorzBaseSimplePrompt._set_error  r@rurc|js?d|jd<d|jd<d|jd<|r|jddSdS||jdS)zHandle the event when attempting to skip a prompt. Skip the prompt if the `_mandatory` field is False, otherwise show an error message that the prompt cannot be skipped. Tr6r8Nr7r7)r$)rMstatusappexitrrNrGrus r>rlzBaseSimplePrompt._handle_skips  =&*DK #%)DK "$(DK ! , d+++++ , , OOD$;O < < < < rkz"BaseSimplePrompt._handle_interruptsX"& J = H!% I  A INN"?N @ @ @ @ @ A Ar@cdS)z:Handle the event when user attempt to answer the question.Nr<rs r>rjzBaseSimplePrompt._handle_enters  r@c|jS)a Dict[str, Any]: Get current prompt status. The status contains 3 keys: "answered" and "result". answered: If the current prompt is answered. result: The result of the user answer. skipped: If the prompt is skipped. rYrFs r>rzBaseSimplePrompt.statuss |r@c||_dSrDrrGvalues r>rzBaseSimplePrompt.statuss  r@)r-rzchtjddtdtffd }|S)aKeybinding registration decorator. This decorator wraps around the :meth:`prompt_toolkit.key_binding.KeyBindings.add` with added feature to process `alt` realted keybindings. By default, `prompt_toolkit` doesn't process `alt` related keybindings, it requires `alt-ANY` to `escape` + `ANY`. Args: keys: The keys to bind that can trigger the function. filter: :class:`~prompt_toolkit.filter.Condition` to indicate if this keybinding should be active. Returns: A decorator that should be applied to the function thats intended to be active when the keys are pressed. Examples: >>> @self.register_kb("alt-j") ... def test(event): ... pass z ^alt-(.*)rLr4c,g}D]l}|}|r>|d||dW||m jj|didfd }|S)Nescaper-r4c|dSrDr<)rurLs r> executablezCBaseSimplePrompt.register_kb..decorator..executablesU r@r4N)matchappendgrouprZadd) rLformatted_keysrHrr alt_patternr-rzkwargsrGs ` r> decoratorz/BaseSimplePrompt.register_kb..decoratorsN / /#))#../"))(333"))%++a..9999"))#.... TX\> C& CF C C     D C  r@)recompiler)rGr-rzrrrs```` @r>ryzBaseSimplePrompt.register_kbsb0j..  . 3E           r@ pre_answer post_answercg}|jdrG|d|jf|d|jrdnd|jdfn|jdr|d|jf|d|jrdnd|jf||js|n$dd ||jd zfnZ|d |jf|d |jrdnd|jf|||S) aGet the question message in formatted text form to display in the prompt. This function is mainly used to render the question message dynamically based on the current status (answered or not answered) of the prompt. Note: The function requires implementation when inheriting :class:`.BaseSimplePrompt`. You should call `super()._get_prompt_message(pre_answer, post_answer)` in the implemented `_get_prompt_message`. Args: pre_answer: The message to display before the question is answered. post_answer: The information to display after the question is answered. Returns: Formatted text in list of tuple format. r8z class:skipped r!r6zclass:answermarkzclass:answered_questionz class:answerz %sr7zclass:questionmarkzclass:question)rrrWrQrXr\)rGrrdisplay_messages r>_get_prompt_messagez$BaseSimplePrompt._get_prompt_messages* ;y ! /  " "OT[#A B B B  " " DK-GSSR-G"WX    [ $ /  " "$6 #D E E E  " "-%)[8ccb8$--H     " "( #D--dk(.CDDD      " "$8$+#F G G G  " "$%)[8ccb8$--H     " ": . . .r@cdS)zAbstractmethod to enforce a run function is implemented. All prompt instance requires a `_run` call to initialise and run an instance of `PromptSession` or `Application`. Nr<rFs r>_runzBaseSimplePrompt._run,rr@c KdS)zAbstractmethod to enforce a run function is implemented. All prompt instance requires a `_run_async` call to initialise and run an instance of `PromptSession` or `Application`. Nr<rFs r> _run_asynczBaseSimplePrompt._run_async5s  r@c|}|tjd|  |_|tkrt |js|S||S)axRun the prompt and get the result. Args: raise_keyboard_interrupt: **Deprecated**. Set this parameter on the prompt initialisation instead. Returns: Value of the user answer. Types varies depending on the prompt. Raises: KeyboardInterrupt: When `ctrl-c` is pressed and `raise_keyboard_interrupt` is True. NrB)rr`rarErKeyboardInterruptr])rGr0r7s r>executezBaseSimplePrompt.execute>sm # /"$))/G+G##DO 2 2 2# #| M||F###r@cK|d{V}|tkrt|js|S||S)a Run the prompt asynchronously and get the result. Returns: Value of the user answer. Types varies depending on the prompt. Raises: KeyboardInterrupt: When `ctrl-c` is pressed and `raise_keyboard_interrupt` is True. N)rrrr])rGr7s r> execute_asynczBaseSimplePrompt.execute_asyncUs\(((((((( 2 2 2# #| M||F###r@c|jS)z-str: Instruction to display next to question.)rRrFs r>r)zBaseSimplePrompt.instructiones   r@c|jS)z$Dict[str, Any]: Keybinding mappings.rirFs r>r}zBaseSimplePrompt.kb_mapsjs }r@rc&i|j||_dSrDrrs r>r}zBaseSimplePrompt.kb_mapsos24=2E2 r@c|jS)z5Dict[str, Any]: Keybinding function lookup mappings..rmrFs r>rszBaseSimplePrompt.kb_func_lookupss ##r@c&i|j||_dSrDrrs r>rszBaseSimplePrompt.kb_func_lookupxs@$"6@%@r@)NFr r r!Nr"NNr!TTTr#NrrD)'__name__ __module__ __qualname____doc__rr rr_strrrrrrnrrrrlrkrjpropertyrrsetterr rrrryr r rrrrrr)r}rsr<r@r>rr&s  ",015.6:15)-!3<@#F F "F (F  F  F  F F -.F F hucz23F 3%*-.F F F #'F F !F "!!89#F $ %F F F F PTTT2 # $   ^  =(?"; = = = = =Ax'@ATAAAA 8O#<    ^ S#XX ]]?C***49%*/;* %&(:: ;****X3S/38=c3h3 eCHo 333^3j c   ^  #   ^ $$$#$$$$.$$$$ !S!!!X!c3hX ^3T#s(^3333^3$S#X$$$X$ADcNAtAAAAAAr@r)+rr`rabcrrtypingrrrrr r r r r prompt_toolkit.enumsrprompt_toolkit.filters.baserr'prompt_toolkit.key_binding.key_bindingsrrprompt_toolkit.keysrprompt_toolkit.styles.stylerprompt_toolkit.validationrInquirerPy.enumrInquirerPy.exceptionsrInquirerPy.utilsrrrrr(prompt_toolkit.key_binding.key_processorrrr<r@r>rs99 ########                      -,,,,,????????SSSSSSSS$$$$$$------//////999999555555GFFFFFFTATATATATAsTATATATATAr@