,j=ldZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZmZddlmZddlmZddlmZdd lmZdd lmZmZmZmZdd lm Z m!Z!dd l"m#Z#m$Z$dd l%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>ddl?m@Z@ddlAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZIerddlJmKZKdgZLGdde2ZMGdde4ZNy) z3Module contains the class to create a fuzzy prompt.N) TYPE_CHECKINGAnyCallableDictListOptionalTupleUnioncast) fuzzy_match) fzy_scorer substr_scorer) HAYSTACKS) Application)Buffer)IsDone)ConditionalContainerFloatContainerHSplitWindow) BufferControl DummyControl) DimensionLayoutDimension)Layout) AfterInput BeforeInput) SimpleLexer)ValidationError)Frame) FakeDocumentInquirerPyUIListControl)BaseListPrompt)InstructionWindow) MessageWindow)ValidationFloat)INQUIRERPY_POINTER_SEQUENCE)InvalidArgument) Separator)InquirerPyDefaultInquirerPyKeybindingsInquirerPyListChoicesInquirerPyMessageInquirerPySessionResultInquirerPyStyleInquirerPyValidatecalculate_height) KeyPressEvent FuzzyPromptceZdZdZdedededegefdedee de d ed e d d ffd Z ddZ d e eeeffdZd e eeeffdZd e eeeffdZded e eeeffdZed eeeffdZed efdZxZS)InquirerPyFuzzyControla(An :class:`~prompt_toolkit.layout.UIControl` class that displays a list of choices. This only displays the chocies. The actual input buffer will be handled by a separate :class:`~prompt_toolkit.layout.BufferControl`. Reference the parameter definition in :class:`.FuzzyPrompt`. choicespointermarker current_text max_linessession_result multiselect marker_pl match_exactreturnNc ||_||_||_||_|dkDr|nd|_| st nt |_t |%|d||y)Nr)r6defaultr;r<) _pointer_marker _marker_pl _current_text _max_linesr r_scorersuper__init__) selfr6r7r8r9r:r;r<r=r> __class__s Z/home/ubuntu/waydroid_script/venv/lib/python3.12/site-packages/InquirerPy/prompts/fuzzy.pyrJzInquirerPyFuzzyControl.__init__FsZ   #)'01})!)4z-  )#  cDt|jD]-\}}t|dtr t d||d<g|d</|j|_d|_t|j|j|_ |j|j z |_ y)Nvaluez:fuzzy prompt argument choices should not contain Separatorindexindicesr) enumerater6 isinstancer)r(_filtered_choices _first_lineminrG choice_count _last_line_height)rKrQchoices rM_format_choicesz&InquirerPyFuzzyControl._format_choices_s&t||4 #ME6&/95%P$F7O "F9   #"&doot/@/@A)9)99 rNcg}|jd|jf|jd|j|ddr |jn |jf|jd|ds|jd|df|St |d}t |dD]0\}}||vr|jd|f|jd|f2|S) aGet the current highlighted line of text. If in the middle of filtering, loop through the char and color indices matched char into style class `class:fuzzy_match`. Returns: FormattedText in list of tuple format. class:pointer class:markerrQenabled)z[SetCursorPosition]rRnameclass:fuzzy_match)appendrCr6rDrEsetrSrKr[display_choicesrRrQchars rM_get_hover_textz&InquirerPyFuzzyControl._get_hover_textls?@<<w0; __    :;i  " "OVF^#D E &+,G(8 D tG##**,?+FG#**OT+BC  D rNcg}|jdt|jdzf|jd|j|ddr |jn |j f|ds|jd|df|St |d}t|dD]0\}}||vr|jd |f|jd|f2|S) a<Get the line of text in `FormattedText`. If in the middle of filtering, loop through the char and color indices matched char into `class:fuzzy_match`. Calculate spaces of pointer to make the choice equally align. Returns: FormattedText in list of tuple format. r^ r_rQr`rRrarbrc)rdlenrCr6rDrErerSrfs rM_get_normal_textz'InquirerPyFuzzyControl._get_normal_textsT]]1Cc1IJK<<w0; __   i  " "Bv#7 8 &+,G(8 7 tG##**,?+FG#**B:6  7 rNcg}|jdk(r d|_|S|jdkrd|_n-|j|jk\r|jdz |_|j|jz t |j|j krWt |j|j |_|jt |j|j z |_|j|jkrD|j|_|jt |j |jz|_n_|j|jk\rF|jdz|_|jt |j |jz |_|j|jkDrC|j|_|jt |j |jz |_|jdkr9d|_|jt |j |jz|_t |j|jD]e}||jk(r"||j|j|z }n!||j|j|z }|jdg|r|j|S)a1Get all available choices in formatted text format. Overriding this method because `self.choice` will be the full choice list. Using `self.filtered_choice` to get a list of choice based on current_text. Returns: FormattedText in list of tuple format. rrA)ra ) rX_selected_choice_indexrYrVrWrZrangeselected_choice_indexrirUrmrdpop)rKrgrQs rM_get_formatted_choicesz-InquirerPyFuzzyControl._get_formatted_choicess]    !*+D '" "  & & **+D '  ( (D,=,= =*.*;*;a*?D ' OOd.. .#d6G6G2V V!$"3"3T\\BDO#T5F5F 1UUD   & &$*:*: :#::D "..T\\4CTCT1UUDO  ( (DOO ;"99A=DO#T\\4CTCT1UUD  ??T.. ."//DO#T\\4CTCT1UUD    a  D "..T\\4CTCT1UUDO4++T__= /E2224#7#78N8Nu8U#VV4#8#89O9OPU9V#WW  " ": .  /     !rN wait_timecFK|js$|jD]}g|d< |j}|Stj|d{t |jt t |jd|jd{}|S7M7w)zCall to filter choices using fzy fuzzy match. Args: wait_time: Additional time to wait before filtering the choice. Returns: Filtered choices. rRNrb)keyscorer)rFr6asynciosleepr r rrH)rKrur[r6s rM_filter_choicesz&InquirerPyFuzzyControl._filter_choicess!!#,, '$&y! 'llG-- * * *'""$Y -|| G  +s%A B!BAB!BB!B!c4|j|jS)zOverride this value since `self.choice` does not indicate the choice displayed. `self.filtered_choice` is the up to date choice displayed. Returns: A dictionary of name and value for the current pointed choice. )rUrrrKs rM selectionz InquirerPyFuzzyControl.selections%%d&@&@AArNc,t|jS)zint: Filtered choice count.)rlrUr}s rMrXz#InquirerPyFuzzyControl.choice_counts4))**rNr?N)__name__ __module__ __qualname____doc__r,strrintrr.boolrJr\rr rirmrtfloatrrr{propertyr~rX __classcell__rLs@rMr5r5=s1 &    r3w'    !!89      2 :eCHo)>>$uS#X*?@.U38_(=.`ud38n9M.B4S>BB+c++rNr5cA*eZdZdZdedddddddddeeddddddddd dddddd dfd ed ed edede e de dedede e e ge fde e e ge fdedede dededede de de dede eeefd e eeefd!e ed"ed#e ed$e d%e d&e d'e d(ed)e ed*df@fd+ Zd:d,e e d*dfd-Zd;d.Zd:d,e e d*dfd/Zd*eeeeffd0Zd*eeeeffd1Zd2Zd*efd3Zd;d4Zd;d5Zd` documentation for more details. vi_mode: Use vim keybinding for the prompt. Refer to :ref:`pages/kb:Keybindings` documentation for more details. default: Set the default value in the search buffer. Different than other list type prompts, the `default` parameter tries to replicate what fzf does and add the value in `default` to search buffer so it starts searching immediatelly. Refer to :ref:`pages/dynamic:default` documentation for more details. qmark: Question mark symbol. Custom symbol that will be displayed infront of the question before its answered. amark: Answer mark symbol. Custom symbol that will be displayed infront of the question after its answered. pointer: Pointer symbol. Customer symbol that will be used to indicate the current choice selection. instruction: Short instruction to display next to the question. long_instruction: Long instructions to display at the bottom of the prompt. validate: Add validation to user input. The main use case for this prompt would be when `multiselect` is True, you can enforce a min/max selection. Refer to :ref:`pages/validator:Validator` documentation for more details. invalid_message: Error message to display when user input is invalid. Refer to :ref:`pages/validator:Validator` documentation for more details. transformer: A function which performs additional transformation on the value that gets printed to the terminal. Different than `filter` parameter, this is only visual effect and won’t affect the actual value returned by :meth:`~InquirerPy.base.simple.BaseSimplePrompt.execute`. Refer to :ref:`pages/dynamic:transformer` documentation for more details. filter: A function which performs additional transformation on the result. This affects the actual value returned by :meth:`~InquirerPy.base.simple.BaseSimplePrompt.execute`. Refer to :ref:`pages/dynamic:filter` documentation for more details. height: Preferred height of the prompt. Refer to :ref:`pages/height:Height` documentation for more details. max_height: Max height of the prompt. Refer to :ref:`pages/height:Height` documentation for more details. multiselect: Enable multi-selection on choices. You can use `validate` parameter to control min/max selections. Setting to True will also change the result from a single value to a list of values. prompt: Input prompt symbol. Custom symbol to display infront of the input buffer to indicate for input. border: Create border around the choice window. info: Display choice information similar to fzf --info=inline next to the prompt. match_exact: Use exact sub-string match instead of using fzy fuzzy match algorithm. exact_symbol: Custom symbol to display in the info section when `info=True`. marker: Marker Symbol. Custom symbol to indicate if a choice is selected. This will take effects when `multiselect` is True. marker_pl: Marker place holder when the choice is not selected. This is empty space by default. keybindings: Customise the builtin keybindings. Refer to :ref:`pages/kb:Keybindings` for more details. cycle: Return to top item if hit bottom during navigation or vice versa. wrap_lines: Soft wrap question lines when question exceeds the terminal width. raise_keyboard_interrupt: Raise the :class:`KeyboardInterrupt` exception when `ctrl-c` is pressed. If false, the result will be `None` and the question is skiped. mandatory: Indicate if the prompt is mandatory. If True, then the question cannot be skipped. mandatory_message: Error message to show when user attempts to skip mandatory prompt. session_result: Used internally for :ref:`index:Classic Syntax (PyInquirer)`. Examples: >>> from InquirerPy import inquirer >>> result = inquirer.fuzzy(message="Select one:", choices=[1, 2, 3]).execute() >>> print(result) 1 raNF?rkTz Ez Invalid inputzMandatory promptmessager6rBr7stylevi_modeqmarkamark transformerfilter instructionlong_instructionr<promptr8r=borderinfor> exact_symbolheight max_heightvalidateinvalid_message keybindingscycle wrap_linesraise_keyboard_interrupt mandatorymandatory_messager;r?c |si}|_|_d_d_|_ddiddigddiddigggd|}t #||||||| | ||| | | |||||||d d jigi_t|ts|ntt|j_ xjd z c_t||j \__t'|||j(j$|| || _t-j._t3t5j6d t9j0t;j<t?j@gtCd} fd}!t3jD|!d_#tI| jFg}"jJr tM|"}"tOtQtItSjTdjVdtY|"t[tYt3t]t[j^ztajbt[j^zjVgtejfjht[zjVdjjg_6jljo| tqjljrjtjvjx_=y)NFrwupzc-pdownzc-n)rrtoggle toggle-exact)rrrrrrrrrrr<rrrrrrrrr;rfuncrA) height_offset) r6r7r8r9r:r;r<r=r>)on_text_changedz class:input)lexer)rcontentctjjjjdkDrjjSdS)NrrA)max preferredrW)r_dimmension_max_height_dimmension_heightcontent_controlrZr}srMz&FuzzyPrompt.__init__..sN9++--040D0D0L0Lq0P$$,,, WX, rNT)rrdont_extend_height)rrr show_cursor)r)r)rrrr)rrrleftbottom)rfloats)layoutr key_bindings editing_mode after_render)>_prompt_info_task _rendered _exact_symbolrIrJ _toggle_exactkb_func_lookuprTrr _result_default_height_offsetr1rrrr5_get_current_text_content_controlr_on_text_changed_bufferrrexactrr_generate_after_inputr_generate_before_inputrr choice_windowr_borderr rrr%_get_prompt_message _wrap_linesrrr_is_displaying_long_instructionr$_long_instructionr&_get_error_message _is_invalid _validation_window_bottom_offset_layoutfocusr_style_kb _editing_mode _after_render _application)$rKrr6rBr7rrrrrrrrr<rr8r=rrr>rrrrrrrrrrrr; input_windowchoice_height_dimmensionmain_content_windowrLs$` rMrJzFuzzyPrompt.__init__KsZDK   )4=5%.1V_uen5     #+##-#!%=/))  , .9K9K0L/MNgx0 (h(6 q ?O Jd.@.@@ <!<9O//11)## 9 d.C.CD "((+! t99: ; ;<"-0   $  $((+#  %lD4F4F%GH <<"'(;"<  %$($<$<#''+'7'7(,  -/$*H9-"<>:$*H9t/S/S#S*$($:$:$*H9t/S/S#S'+'7'70$(,(?(?#//68);#'#3#3#DD 1! #  H <('<<++++++  rNrPc|.|st|j_yt|j_y|jjtk(rtnt|j_y)zToggle matching algorithm. Switch between fzy fuzzy match or sub-string exact match. Args: value: Specify the value to toggle. N)r rrrH)rK_rPs rMrzFuzzyPrompt._toggle_exactsS  =B:D ( D (''//=@"  (rNc|jrAt|j}||j_t ||j_yy)zRender callable choices and set the buffer default text. Setting buffer default text has to be after application is rendered and choice are loaded, because `self._filter_choices` will use the event loop from `Application`. N)rrrtextrlcursor_position)rKr default_texts rM _on_renderedzFuzzyPrompt._on_rendereds; ==t}}-L ,DLL +.|+>  15. $$&rNcddddd}d}t|jjdkDr>tt j t|jjdz}|dkry ||vr||S|d d|d z zzS) a1Calculate wait time to smoother the application on big data set. Using digit of the choices lengeth to get wait time. For digit greater than 6, using formula 2^(digit - 5) * 0.3 to increase the wait_time. Returns: Desired wait time before running the filter. g?g?g?g333333?)rArrgr)rlrr6rmathlog10)rK wait_tabledigits rM_calculate_wait_timez FuzzyPrompt._calculate_wait_time<s   t##++ ,q 0 3t';';'C'C#DEFJE 19 J e$ $!}eai 011rNcz|jrd|_|j}|jr4|jjs|jj t j |jj||_|jj|jy)aHandle buffer text change event. 1. Check if there is current task running. 2. Cancel if already has task, increase wait_time 3. Create a filtered_choice task in asyncio event loop 4. Add callback 1. Run a new filter on all choices. 2. Re-calculate current selected_choice_index if it exceeds the total filtered_choice. 3. Avoid selected_choice_index less than zero, this fix the issue of cursor lose when: choice -> empty choice -> choice Don't need to create or check asyncio event loop, `prompt_toolkit` application already has a event loop running. FN) _invalidrrdonecancelry create_taskrr{add_done_callbackr)rKrrus rMrzFuzzyPrompt._on_text_changedUs$ ==!DM--/ ::djjoo/ JJ   ((  0 0 ;   $$T%:%:;rNc|jsy|jjd}|jj|d |jj|d<y)z;Handle tab event, alter the `selected` state of the choice.NrQr`)rrr~r6)rKrcurrent_selected_indexs rM_handle_toggle_choicez!FuzzyPrompt._handle_toggle_choiceqsc  !%!5!5!?!?!H$$,,-CDYO O $$%;<  rNc t|j}|jj||jrd|j d<|j s\|jjdg|j d<|jj|jjdgy|j|j d<|jj|jyd|j d<|jjd|j d<|jj|jjdy#t$r$}|jt|Yd}~yd}~wt$rYd|j d<|jsdng|j d<|jj|jsdngYywxYw)aHandle enter event. Validate the result first. In multiselect scenario, if no TAB is entered, then capture the current highlighted choice and return the value in a list. Otherwise, return all TAB choices as a list. In normal scenario, reutrn the current highlighted choice. If current UI contains no choice due to filter, return None. TansweredrbrrP)rN)r! result_value _validatorrrstatusrrr~appexit result_namer _set_errorr IndexError)rKevent fake_documentes rM _handle_enterzFuzzyPrompt._handle_enterzs} I():):;M OO $ $] 3  *. J',,-1-A-A-K-KF-S,TDKK)IINN4+?+?+I+I'+R*SNT,0,<,?e @!!89Ae B Ce Nhtn" = R8D> RT RtE#s(O'<&U38_(= '2e22<8P!IFE!7EE&%;&&&!3!rN)Orryrtypingrrrrrrr r r pfzyr pfzy.scorer r pfzy.typesr&prompt_toolkit.application.applicationrprompt_toolkit.bufferrprompt_toolkit.filters.clir prompt_toolkit.layout.containersrrrrprompt_toolkit.layout.controlsrrprompt_toolkit.layout.dimensionrrprompt_toolkit.layout.layoutr prompt_toolkit.layout.processorsrrprompt_toolkit.lexers.baserprompt_toolkit.validationrprompt_toolkit.widgets.baser InquirerPy.baser!r"InquirerPy.base.listr#!InquirerPy.containers.instructionr$InquirerPy.containers.messager% InquirerPy.containers.validationr&InquirerPy.enumr'InquirerPy.exceptionsr(InquirerPy.separatorr)InquirerPy.utilsr*r+r,r-r.r/r0r1(prompt_toolkit.key_binding.key_processorr2__all__r5r3rNrMr<s9    0 >(- GF/D25-A/?7<71*   F /C+4C+Le!.e!rN