4xj=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@ddlAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZIerddlJmKZKdgZLGdde2ZMGdde4ZNdS) 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 FuzzyPromptc\eZdZdZdedededegefdedee 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 ||_||_||_||_|dkr|nd|_| st nt |_t |d||dS)Nr)r6defaultr;r<) _pointer_marker _marker_pl _current_text _max_linesr r_scorersuper__init__) selfr6r7r8r9r:r;r<r=r> __class__s IC:\Projects\Historical\venv\Lib\site-packages\InquirerPy/prompts/fuzzy.pyrJzInquirerPyFuzzyControl.__init__Fs|   #)'01}}))!)4Gzz-  )#      c.t|jD]9\}}t|dtrt d||d<g|d<:|j|_d|_t|j|j |_ |j |jz |_ dS)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|44 # #ME6&/955 %P$F7O "F9  !%dot/@AA)99 rNcg}|d|jf|d|j|ddr|jn|jf|d|ds|d|dfnct |d}t |dD]8\}}||vr|d|f!|d|f9|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  D  " "OVF^#D E E E E&+,,G(88 D D tG###**,?+FGGGG#**OT+BCCCCrNcg}|dt|jdzf|d|j|ddr|jn|jf|ds|d|dfnct |d}t|dD]8\}}||vr|d |f!|d|f9|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]1C1Cc1IJKKK<w0;% _     i  7  " "Bv#7 8 8 8 8&+,,G(88 7 7 tG###**,?+FGGGG#**B:6666rNcg}|jdkr d|_|S|jdkrd|_n|j|jkr|jdz |_|j|jz t |j|jkrFt |j|j|_|jt |j|jz |_|j|jkr4|j|_|jt |j|jz|_nF|j|jkr6|jdz|_|jt |j|jz |_|j|jkr3|j|_|jt |j|jz |_|jdkr.d|_|jt |j|jz|_t |j|jD]i}||jkr$|||j |z }n#|| |j |z }| dj|r| |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_choicessB   ! !*+D '" "  & * **+D ' '  (D,= = =*.*;a*?D ' Od. .#d6G2V2V V V!$"3T\BBDO#T5F 1U1UUD   &$*: : :#:D ".T\4CT1U1UUDOO  (DO ; ;"9A=DO#T\4CT1U1UUD  ?T. . ."/DO#T\4CT1U1UUD   a   D ".T\4CT1U1UUDO4+T_== / /E2224#7#78Nu8U#V#VV4#8#89OPU9V#W#WW  " ": . . . .  "    ! ! !rN wait_timec$K|s|jD]}g|d<|j}nbtj|d{Vt |t t |jd|jd{V}|S)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!!lGG- ** * * * * * * *'""$$Y --| G rNc&|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=s &    r3w'    !!89           2 : : : :eCHo)>>$uS#X*?@.U38_(=....`ud38n9M.B4S>BBBXB+c+++X+++++rNr5cAeZdZdZdedddddddddeeddddddddd 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+ Zdd8Z e!d*e"ffd9 Z#e#j$d,e"d*dfd:Z#d*efd;Z%xZ&S)?r3ufCreate a prompt that lists choices while also allowing fuzzy search like fzf. A wrapper class around :class:`~prompt_toolkit.application.Application`. Fuzzy search using :func:`pfzy.match.fuzzy_match` function. Override the default keybindings for up/down as j/k cannot be bind even if `editing_mode` is vim due to the input buffer. Args: message: The question to ask the user. Refer to :ref:`pages/dynamic:message` documentation for more details. choices: List of choices to display and select. Refer to :ref:`pages/dynamic:choices` documentation for more details. style: An :class:`InquirerPyStyle` instance. Refer to :ref:`Style ` 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|n"tt|j _ xjd z c_t||j \__t'|||jj|| || _t-j_t3t5jd t9jt;jt?j gtCd} fd}!t3j"|!d_#tI| j#g}"j%rtM|"}"tOtQtItSj*dj+dtY|"t[tYt3t]t[j/ztaj1t[j/zj+gtej3j4t[zj+dj5g_6j67| tqj6j9j:j;j<_=dS)NFrwupzc-pdownzc-n)rrtoggle toggle-exact)rrrrrrrrrrr<rrrrrrrrr;rfuncrA) height_offset) r6r7r8r9r:r;r<r=r>)on_text_changedz class:input)lexer)rcontentcttjjjjdkr jjndS)NrrA)max preferredrW)r_dimmension_max_height_dimmension_heightcontent_controlrZr}srMz&FuzzyPrompt.__init__..sA9+-040D0Lq0P0P$,,VW, , , 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__KsD K   )4=5%.1V_uen5      #+##-#!%=/))    , .9K0L/MNgx00 7GG(h((66 q ?O Jd.@@ @ @ <!<9O/1)## 9 9 9 d.CDDD "(++! t9:: ;<<"-00     $ $ $ $  $(+#    %lD4F%GHH < ="'(;"<"<  %$($<#''+'7(,  -/$*HH9-"<>>:::$*HH9t/S#S*$($:$*HH9t/S#S'+'70$(,(?#/688);#'#3#D 1! ! ! # #  H <((('<+++    rNrPc||stnt|j_dS|jjtkrtnt|j_dS)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_exactsR  =B+U:: D ( ( ('/=@@ "  ( ( (rNc|jr;t|j}||j_t ||j_dSdS)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_renderedsI = =t}--L ,DL +.|+<+
>    F15. $$&&&&&rNcddddd}d}t|jjdkr;tt jt|jjdz}|dkrd S||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 0 3t';'C#D#DEEFFJE 1993 J  e$ $!}eai 011rNc\|jrd|_|}|jr2|js|jt j|j||_|j |j dS)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--// : djoo// J     (  0 0 ; ;    $$T%:;;;;;rNc|jsdS|jjd}|jj|d |jj|d<dS)z;Handle tab event, alter the `selected` state of the choice.NrQr`)rrr~r6)rKrcurrent_selected_indexs rM_handle_toggle_choicez!FuzzyPrompt._handle_toggle_choiceqs_   F!%!5!?!H$,-CDYO O $%;<    rNeventr2c4 t|j}|j||jrd|jd<|jsI|jjdg|jd<|j |jjdgdS|j |jd<|j |jdSd|jd<|jjd|jd<|j |jjddS#t$r-}| t|Yd}~dSd}~wt$rEd|jd<|jsdng|jd<|j |jsdngYdSwxYw)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)rKr fake_documentes rM _handle_enterzFuzzyPrompt._handle_enterzs I():;;M O $ $] 3 3 3 O*. J',=-1-A-KF-S,TDK)INN4+?+I'+R*SNTTTTT,0,?e @!!89Ae B Ce e e e e e Nhtn" = = = = R R8D> RT R R R RtE#s(O'<&U38_(= '''2e22222<<<<8PPPP!I!I!I!IFE!7EEEEEXE&%;&&&&&!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<s^99                       00000000 >>>>>>((((((------ GFFFFFFFFFFFFFFF//////DDDDDDDD222222555555------AAAAAAAA//////??????777777<<<<<<777777111111******                    GFFFFFF /C+C+C+C+C+4C+C+C+Le!e!e!e!e!.e!e!e!e!e!rN