tniJ(dZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZeGd dZGddeZdS)zContains the interface class :class:`.BaseComplexPrompt` for more complex prompts and the mocked document class :class:`.FakeDocument`.N) dataclass)AnyCallableListOptionalTupleUnion) Application) EditingMode) Condition FilterOrBool)KeyHandlerCallable)Keys)BaseSimplePrompt)INQUIRERPY_KEYBOARD_INTERRUPT)InquirerPySessionResultInquirerPyStyleInquirerPyValidatec,eZdZUdZeed<dZeed<dS) FakeDocumentaA fake `prompt_toolkit` document class. Work around to allow non-buffer type :class:`~prompt_toolkit.layout.UIControl` to use :class:`~prompt_toolkit.validation.Validator`. Args: text: Content to be validated. cursor_position: Fake cursor position. textrcursor_positionN)__name__ __module__ __qualname____doc__str__annotations__rintT/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/InquirerPy/base/complex.pyrrs7 IIIOSr!rc%eZdZdZ d2d eeeegeffd ee d e d e d ededededeee ge fdeee ge fdee dede de de dedeeddf$fd Z d3dZdddeeefdedeegefffdZd3d Zd!eeddfd"Zd eddfd#Zdeeeeffd$Zd%eeddfd&Zdeeeefffd' Zde fd(Zd3d)Zedefd*Zej d+eddfd,Zede!fd-Z"ede!fd.Z#ede!fd/Z$ede!fd0Z%ede!fd1Z&xZ'S)4BaseComplexPromptaA base class to create a more complex prompt that will involve :class:`~prompt_toolkit.application.Application`. Note: This class does not create :class:`~prompt_toolkit.layout.Layout` nor :class:`~prompt_toolkit.application.Application`, it only contains the necessary attributes and helper functions to be consumed. Note: Use :class:`~InquirerPy.base.BaseListPrompt` to create a complex list prompt which involves multiple choices. It has more methods and helper function implemented. See Also: :class:`~InquirerPy.base.BaseListPrompt` :class:`~InquirerPy.prompts.fuzzy.FuzzyPrompt` NF? Invalid inputTMandatory promptmessagestylebordervi_modeqmarkamark instructionlong_instruction transformerfiltervalidateinvalid_message wrap_linesraise_keyboard_interrupt mandatorymandatory_messagesession_resultreturnct||||||| | | | | ||||| _d_d_d_|_|_d_jrxjdz c_jrxjdz c_jsdnd_ j rxj j z c_ tfd_ tfd_tfd_dS) N)r)r*r,r-r.r/r1r2r4r3r5r6r7r8r9Frc.jtjkSN) _editing_moder VIselfsr"z,BaseComplexPrompt.__init__..lsd.@KN.Rr!cjSr?_invalidrBsr"rDz,BaseComplexPrompt.__init__..ms T]r!cjdkS)Nr&)_long_instructionrBsr"rDz,BaseComplexPrompt.__init__..osD*b0r!)super__init___invalid_message _renderedrG_loadingrI_border_height_offset _validation_window_bottom_offset _wrap_lines!extra_long_instruction_line_countr _is_vim_edit _is_invalid_is_displaying_long_instruction)rCr)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9 __class__s` r"rKzBaseComplexPrompt.__init__5sa( ##+!%=/)    "!0   &!1  < %   1 $    ! %   1 $  9=9O0VUV-     1 16  1 1&&R&R&R&RSS$%:%:%:%:;;/8 0 0 0 00 0 ,,,r!c8|jdS)zRedraw the application UI.N) _application invalidaterBs r"_redrawzBaseComplexPrompt._redrawrs $$&&&&&r!)r2keyschtj|d|idtdtffd }|S)zDecorate keybinding registration function. Ensure that the `invalid` state is cleared on next keybinding entered. r2funcr:c&fd}|S)Nc:jrd_|dS)NFrF)eventr^rCs r" executablezDBaseComplexPrompt.register_kb..decorator..executables%=*$)DMU r!r )r^rbkb_decrCs` r" decoratorz0BaseComplexPrompt.register_kb..decorators4      V   r!)rJ register_kbr)rCr2r\rdrcrWs` @r"rezBaseComplexPrompt.register_kbvs]%$d:6:: . 3E       r!cd|jd<t|jd<d|jd<|j|ddS)zSet exception handler for the event loop. Skip the question and raise exception. Args: loop: Current event loop. context: Exception context. Tansweredresultskipped exception)rjN)_statusrrYexit)rC_contexts r"_exception_handlerz$BaseComplexPrompt._exception_handlersM$( Z !> X"& Y )=>>>>>r!appcx|js2d|_|||dSdS)amRun after the :class:`~prompt_toolkit.application.Application` is rendered/updated. Since this function is fired up on each render, adding a check on `self._rendered` to process logics that should only run once. Set event loop exception handler here, since its guaranteed that the event loop is running in `_after_render`. TN)rM_keybinding_factory _on_rendered)rCrps r" _after_renderzBaseComplexPrompt._after_rendersK~ #!DN  $ $ & & &   c " " " " "  # #r!c"||_d|_dS)zoSet error message and set invalid state. Args: message: Error message to display. TN)rLrG)rCr)s r" _set_errorzBaseComplexPrompt._set_errors !( r!cd|jfgS)zsObtain the error message dynamically. Returns: FormattedText in list of tuple format. zclass:validation-toolbar)rLrBs r"_get_error_messagez$BaseComplexPrompt._get_error_messages+%   r!rmcdS)zARun once after the UI is rendered. Acts like `ComponentDidMount`.Nr )rCrms r"rszBaseComplexPrompt._on_rendereds r!cd|jr d|jzndf}dd|jdzf}t||S)zqGet the prompt message to display. Returns: Formatted text in list of tuple format. zclass:instructionz %s  z class:answerz %srh)r/statusrJ_get_prompt_message)rC pre_answer post_answerrWs r"r}z%BaseComplexPrompt._get_prompt_messagesY )-)9 BFT% % %s &ut{8/D'DE ww**:{CCCr!c4|jS)zRun the application.) applicationrunrBs r"_runzBaseComplexPrompt._runs##%%%r!cDK|jd{VS)z#Run the application asynchronously.N)r run_asyncrBs r" _run_asynczBaseComplexPrompt._run_asyncs-%//111111111r!c,|jst|jS)adGet the application. :class:`.BaseComplexPrompt` requires :attr:`.BaseComplexPrompt._application` to be defined since this class doesn't implement :class:`~prompt_toolkit.layout.Layout` and :class:`~prompt_toolkit.application.Application`. Raises: NotImplementedError: When `self._application` is not defined. )rYNotImplementedErrorrBs r"rzBaseComplexPrompt.applications  &% %  r!valuec||_dSr?)rY)rCrs r"rzBaseComplexPrompt.applications!r!c<|js|jS|j|jzS)zint: Height offset to apply.)rRrPextra_line_countrBs r" height_offsetzBaseComplexPrompt.height_offsets) '& &$t':::r!cd}|jr|t|jz }|dz }|tt|jz }|dz }|tt|jz }|jr|dz }|S)z!int: Total length of the message.rr=)_qmarklenr_message _instruction)rCtotal_message_lengths r"rz&BaseComplexPrompt.total_message_lengths ! ; & C $4$4 4 A % C $6$6 7 77!C(9$:$: ; ;;   & A % ##r!cHtj\}}|jdz |zS)aint: Get the extra lines created caused by line wrapping. Minus 1 on the totoal message length as we only want the extra line. 24 // 24 will equal to 1 however we only want the value to be 1 when we have 25 char which will create an extra line. r=)shutilget_terminal_sizerrC term_widthrms r"extra_message_line_countz*BaseComplexPrompt.extra_message_line_counts)022 A)A-*<%CDDE; (;  ;  ;  ; ; ; ; hucz23; 3%*-.; -.; ; ; #'; !; "#; $!!89%; & '; ; ; ; ; ; z'''' ?C49%/; %&(:: ;( ? ? ? ? #+!6 #4 # # # ##$  DsCx$9      h{3      DT%S/%: D D D D D D&c&&&&2222 ![ ! ! !X !"""""";s;;;X;  $c $ $ $X $=#===X= 3   X #Xr!r$) rr dataclassesrtypingrrrrrr prompt_toolkit.applicationr prompt_toolkit.enumsr prompt_toolkit.filters.baser r 'prompt_toolkit.key_binding.key_bindingsrprompt_toolkit.keysrInquirerPy.base.simplerInquirerPy.enumrInquirerPy.utilsrrrrr$r r!r"rsNN !!!!!!>>>>>>>>>>>>>>>>222222,,,,,,????????FFFFFF$$$$$$333333999999          AAAAA(AAAAAr!