o i=l@sdZddlZddlZddlmZmZmZmZmZm 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@ddlAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZIerddlJmKZKdgZLGddde2ZMGddde4ZNdS) 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 FuzzyPromptcseZdZdZdedededegefdedee de d ed e d d ffd d Z dddZ d e eeeffddZd e eeeffddZd e eeeffddZded e eeeffddZed eeeffddZed efddZZS)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 sP||_||_||_||_|dkr|nd|_| stnt|_tj |d||ddS)Nr)r4defaultr9r:) _pointer_marker _marker_pl _current_text _max_linesr r _scorersuper__init__) selfr4r5r6r7r8r9r:r;r< __class__U/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/InquirerPy/prompts/fuzzy.pyrGFs  zInquirerPyFuzzyControl.__init__cCsjt|jD]\}}t|dtrtd||d<g|d<q|j|_d|_t|j|j |_ |j |j|_ dS)Nvaluez:fuzzy prompt argument choices should not contain Separatorindexindicesr) enumerater4 isinstancer(r'_filtered_choices _first_lineminrD choice_count _last_line_height)rHrNchoicerKrKrL_format_choices_s z&InquirerPyFuzzyControl._format_choicescCsg}|d|jf|d|j|ddr|jn|jf|d|ds2|d|df|St|d}t|dD]\}}||vrN|d|fq>|d|fq>|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:markerrNenabled)z[SetCursorPosition]rOnameclass:fuzzy_match)appendr@r4rArBsetrPrHrXdisplay_choicesrOrNcharrKrKrL_get_hover_textls&   z&InquirerPyFuzzyControl._get_hover_textcCsg}|dt|jdf|d|j|ddr|jn|jf|ds1|d|df|St|d}t|dD]\}}||vrM|d |fq=|d|fq=|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. rZ r[rNr\rOr]r^r_)r`lenr@r4rArBrarPrbrKrKrL_get_normal_texts$  z'InquirerPyFuzzyControl._get_normal_textcCsg}|jdkr d|_|S|jdkrd|_n |j|jkr!|jd|_|j|jt|j|jkrAt|j|j|_|jt|j|j|_|j|jkrW|j|_|jt|j|j|_n|j|jkrn|jd|_|jt|j|j|_|j|jkr|j|_|jt|j|j|_|jdkrd|_|jt|j|j|_t|j|jD]!}||jkr|||j |7}n || |j |7}| dq|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. rr>)r] ) rU_selected_choice_indexrVrSrTrWrangeselected_choice_indexrerRrhr`pop)rHrcrNrKrKrL_get_formatted_choicess>           z-InquirerPyFuzzyControl._get_formatted_choices wait_timecs`|s|jD]}g|d<q|j}|St|IdHt|tt|jd|jdIdH}|S)zCall to filter choices using fzy fuzzy match. Args: wait_time: Additional time to wait before filtering the choice. Returns: Filtered choices. rONr^)keyscorer)rCr4asynciosleepr r rrE)rHrorXr4rKrKrL_filter_choicess     z&InquirerPyFuzzyControl._filter_choicescCs |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. )rRrlrHrKrKrL selections z InquirerPyFuzzyControl.selectioncCs t|jS)zint: Filtered choice count.)rgrRrurKrKrLrUs z#InquirerPyFuzzyControl.choice_countr=N)__name__ __module__ __qualname____doc__r+strrintrr-boolrGrYrrrerhrnfloatrrrtpropertyrvrU __classcell__rKrKrIrLr3=s@       0 r3cAseZdZdZdedddddddddeeddddddddd 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+d, ZdId-e e d*dfd.d/ZdJd0d1ZdId-e e d*dfd2d3Zd*eeeeffd4d5Zd*eeeeffd6d7Zd8d9Zd*efd:d;ZdJdd?ZdKdBdCZ e!d*e"ffdDdE Z#e#j$d-e"d*dfdFdEZ#d*efdGdHZ%Z&S)Lr2ufCreate 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 r]NF?rfTz Ez Invalid inputzMandatory promptmessager4r?r5stylevi_modeqmarkamark transformerfilter instructionlong_instructionr:promptr6r;borderinfor< exact_symbolheight max_heightvalidateinvalid_message keybindingscycle wrap_linesraise_keyboard_interrupt mandatorymandatory_messager9r=c # s|si}|_|_d_d_|_ddiddigddiddigggd|}tjd1id|d |d |d |d |d |d| d| d|d|d| d| d| d|d|d|d|d|d|d|ddjigi_t |t s{|nt t |j _ jd7_t||jd\__t|||jj|| ||d _tjd!_ttdtjtjtj gt!d"d#d$} fd%d&}!tj"|!d'd(_#t$| j#g}"j%rt&|"}"t't(t$t)j*d'j+d'd)t,|"t-d*t,tt.d+t-j/@d*t0j1t-j/@j+d,gt2j3j4t-@j+d-j5d.gd/_6j67| t8j6j9j:j;j) height_offset) r4r5r6r7r8r9r:r;r<)on_text_changedz class:input)lexer)rcontentcs*tjjjjdkrjjdSddS)Nrr>)max preferredrT)r_dimmension_max_height_dimmension_heightcontent_controlrWrKrurKrLsz&FuzzyPrompt.__init__..T)rrdont_extend_height)rrr show_cursor)r)r)rrrr)rrrleftbottom)rfloats)layoutr key_bindings editing_mode after_renderrK)>_prompt_info_task _rendered _exact_symbolrFrG _toggle_exactkb_func_lookuprQrr _result_default_height_offsetr0rrrr3_get_current_text_content_controlr_on_text_changed_bufferrrexactrr_generate_after_inputr_generate_before_inputrr choice_windowr_borderrrrr$_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)#rHrr4r?r5rrrrrrrrr:rr6r;rrr<rrrrrrrrrrrr9 input_windowchoice_height_dimmensionmain_content_windowrIrurLrGKs"             $ zFuzzyPrompt.__init__rMcCs8|dur|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 r rrE)rH_rMrKrKrLrs   zFuzzyPrompt._toggle_exactcCs,|jrt|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)rr|rtextrgcursor_position)rHr default_textrKrKrL _on_rendereds  zFuzzyPrompt._on_renderedcCsR|jsdS|jjD]}|jj|d}t|dtrq |r|n|d |d<q dS)zkToggle all choice `enabled` status. Args: value: Specify the value to toggle. NrNrMr\) _multiselectrrRr4rQr()rHrrMrX raw_choicerKrKrL_handle_toggle_alls zFuzzyPrompt._handle_toggle_allcCszg}|jr;|d|d|jjdt|jjf|jr-|ddt|jdf|jjt kr;|d|j f|S)z,Virtual text displayed after the user input.)r]z zclass:fuzzy_info/z ()) rr`rrUrgr4rselected_choicesrEr rrHdisplay_messagerKrKrLrs  z!FuzzyPrompt._generate_after_inputcCsg}|dd|jf|S)z8Display prompt symbol as virtual text before user input.zclass:fuzzy_promptz%s )r`rrrKrKrLr/sz"FuzzyPrompt._generate_before_inputcCs&|rdS||j_|jdS)zrrgr)rgrr4r}mathlog10)rH wait_tabledigitrKrKrL_calculate_wait_time<s z FuzzyPrompt._calculate_wait_timecCsT|jrd|_|}|jr|js|jt|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) _invalidrrdonecancelrr create_taskrrtadd_done_callbackr)rHrrorKrKrLrUs  zFuzzyPrompt._on_text_changedcCs8|jsdS|jjd}|jj|d |jj|d<dS)z;Handle tab event, alter the `selected` state of the choice.NrNr\)rrrvr4)rHrcurrent_selected_indexrKrKrL_handle_toggle_choiceqs  z!FuzzyPrompt._handle_toggle_choiceeventr1c Cs6z\t|j}|j||jrAd|jd<|js0|jjdg|jd<|j j |jjdgdWdS|j |jd<|j j |jdWdSd|jd<|jjd|jd<|j j |jjddWdSt yv}z| t|WYd}~dSd}~wtyd|jd<|jsdng|jd<|j j |jsdngdYdSw)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. Tansweredr^rrM)rN)r result_value _validatorrrstatusrrrvappexit result_namer _set_errorr| IndexError)rHr fake_documenterKrKrL _handle_enterzs*       zFuzzyPrompt._handle_entercstttjS)z2InquirerPyFuzzyControl: Override for type-hinting.)r r3rFrrurIrKrLrszFuzzyPrompt.content_controlcCs ||_dSN)r)rHrMrKrKrLrs cCs|jjS)zGet current input buffer text.)rrrurKrKrLrszFuzzyPrompt._get_current_textrrw)rr1r=N)'rxryrzr{r&r,r+r)r|rr.r~rrr r}r/r*r-rGrrrrrrrrrrrrrrr3rsetterrrrKrKrIrLr2sK      !(     #)Or{rrrtypingrrrrrrrr 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.baserInquirerPy.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.r/r0(prompt_toolkit.key_binding.key_processorr1__all__r3r2rKrKrKrLsB,                ( G