^iJ(dZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZeGd dZGddeZy)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<y) 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__rinta/mnt/e/genesis-system/.venvs/browser-army/lib/python3.12/site-packages/InquirerPy/base/complex.pyrrs IOSr!rc%\eZdZdZ d-deeeegeffdee de de ded ed ed ed eee ge fd eee ge fdee dede de de dedeeddf$fd Z d.dZdddeeefd edeegefffdZd.dZdeeddfdZdeddfdZdeeeeffdZd eeddfd!Zdeeeefffd" Zde fd#Zd.d$Zedefd%Zej@d&eddfd'Zede!fd(Z"ede!fd)Z#ede!fd*Z$ede!fd+Z%ede!fd,Z&xZ'S)/BaseComplexPromptaA 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` NTmessagestylebordervi_modeqmarkamark instructionlong_instruction transformerfiltervalidateinvalid_message wrap_linesraise_keyboard_interrupt mandatorymandatory_messagesession_resultreturnc t||||||| | | | | ||||| _d_d_d_|_|_d_jrxjdz c_j rxjdz c_j sdnd_ jrxjjz c_ tfd_ tfd_tfd_y) N)r%r&r(r)r*r+r-r.r0r/r1r2r3r4r5Frc>jtjk(SN) _editing_moder VIselfsr"z,BaseComplexPrompt.__init__..lsd.@.@KNN.Rr!cjSr;_invalidr>sr"r@z,BaseComplexPrompt.__init__..ms T]]r!c"jdk7S)N)_long_instructionr>sr"r@z,BaseComplexPrompt.__init__..osD**b0r!)super__init___invalid_message _renderedrC_loadingrF_border_height_offset _validation_window_bottom_offset _wrap_lines!extra_long_instruction_line_countr _is_vim_edit _is_invalid_is_displaying_long_instruction)r?r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5 __class__s` r"rHzBaseComplexPrompt.__init__5s( ##+!%=/)  "!0   !1  <<   1 $   ! !   1 $ 9=9O9OUV-     1 166  1&&RS$%:;/8 00 ,r!c8|jjy)zRedraw the application UI.N) _application invalidater>s r"_redrawzBaseComplexPrompt._redrawrs $$&r!)r.keyscPt|d|idtdtffd }|S)zDecorate keybinding registration function. Ensure that the `invalid` state is cleared on next keybinding entered. r.funcr6c fd}|S)Nc<jrd_|y)NFrB)eventr[r?s r" executablezDBaseComplexPrompt.register_kb..decorator..executables==$)DMU r!r )r[r_kb_decr?s` r" decoratorz0BaseComplexPrompt.register_kb..decorators     r!)rG register_kbr)r?r.rYrar`rTs` @r"rbzBaseComplexPrompt.register_kbvs6$d:6: . 3E r!cd|jd<t|jd<d|jd<|jj|dy)zSet exception handler for the event loop. Skip the question and raise exception. Args: loop: Current event loop. context: Exception context. Tansweredresultskipped exception)rgN)_statusrrVexit)r?_contexts r"_exception_handlerz$BaseComplexPrompt._exception_handlersJ$( Z !> X"& Y )=>r!appcn|js)d|_|j|j|yy)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)rJ_keybinding_factory _on_rendered)r?rms r" _after_renderzBaseComplexPrompt._after_renders1~~!DN  $ $ &   c " r!c ||_d|_y)zoSet error message and set invalid state. Args: message: Error message to display. TN)rIrC)r?r%s r" _set_errorzBaseComplexPrompt._set_errors !( r!c d|jfgS)zsObtain the error message dynamically. Returns: FormattedText in list of tuple format. zclass:validation-toolbar)rIr>s r"_get_error_messagez$BaseComplexPrompt._get_error_messages+%%   r!rjcy)zARun once after the UI is rendered. Acts like `ComponentDidMount`.Nr )r?rjs r"rpzBaseComplexPrompt._on_rendereds r!cd|jrd|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 %sre)r+statusrG_get_prompt_message)r? pre_answer post_answerrTs r"rzz%BaseComplexPrompt._get_prompt_messagesS )-)9)9FT%% %s &ut{{8/D'DE w*:{CCr!c6|jjS)zRun the application.) applicationrunr>s r"_runzBaseComplexPrompt._runs##%%r!cRK|jjd{S7w)z#Run the application asynchronously.N)r~ run_asyncr>s r" _run_asynczBaseComplexPrompt._run_asyncs!%%//1111s '%'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. )rVNotImplementedErrorr>s r"r~zBaseComplexPrompt.applications  % %   r!valuec||_yr;)rV)r?rs r"r~zBaseComplexPrompt.applications !r!cd|js |jS|j|jzS)zint: Height offset to apply.)rOrMextra_line_countr>s r" height_offsetzBaseComplexPrompt.height_offsets0&& &$$t':':::r!c d}|jr|t|jz }|dz }|tt|jz }|dz }|tt|jz }|jr|dz }|S)z!int: Total length of the message.rr9)_qmarklenr_message _instruction)r?total_message_lengths r"rz&BaseComplexPrompt.total_message_lengths ! ;; C $4 4 A % C $6 77!C(9(9$: ;;    A % ##r!cTtj\}}|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. r9)shutilget_terminal_sizerr? term_widthrjs r"extra_message_line_countz*BaseComplexPrompt.extra_message_line_counts,002 A))A-*<%CDDE; (;  ;  ;  ; ; ; ; hucz23; 3%*-.; -.; ; ; #'; !; "#; $!!89%; & '; z' ?C49%/; %&(:: ;( ? #+!6 #4 ##$  DsCx$9   h{3   DT%S/%: D&c&2 ![ ! !"""";s;;  $c $ $=#== 3  #r!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"rs^N !>>2,?F$39     A(Ar!