tniK@dZddlmZddlmZmZmZmZmZm Z ddl m Z m Z ddl mZddlmZddlmZmZddlmZdd lmZdd lmZmZmZmZmZmZmZgd Z eGd d Z!eGddeZ"Gdde Z#GddeZ$dS)z5Module contains the class to create an expand prompt.) dataclass)AnyCallableListOptionalTupleUnion)BaseListPromptInquirerPyUIListControl)Choice)INQUIRERPY_POINTER_SEQUENCE)InvalidArgumentRequiredKeyNotFound) ListPrompt) Separator)InquirerPyDefaultInquirerPyKeybindingsInquirerPyListChoicesInquirerPyMessageInquirerPySessionResultInquirerPyStyleInquirerPyValidate) ExpandPrompt ExpandHelp ExpandChoicec0eZdZUdZdZeed<dZeed<dS)rzHelp choice for the :class:`.ExpandPrompt`. Args: key: The key to bind to toggle the expansion of the prompt. message: The help message. hkeyHelp, list all choicesmessageN)__name__ __module__ __qualname____doc__rstr__annotations__r V/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/InquirerPy/prompts/expand.pyrrs;CNNN+GS+++++r(rc>eZdZUdZdZeeed<fdZxZ S)raChoice class for :class:`.ExpandPrompt`. See Also: :class:`~InquirerPy.base.control.Choice` Args: value: The value of the choice when user selects this choice. name: The value that should be presented to the user prior/after selection of the choice. This value is optional, if not provided, it will fallback to the string representation of `value`. enabled: Indicates if the choice should be pre-selected. This only has effects when the prompt has `multiselect` enabled. key: Char to bind to the choice. Pressing this value will jump to the choice, If this value is missing, the first char of the `str(value)` will be used as the key. Nrct|j3t|jd|_dSdS)zdAssign stringify value to name and also create key using the first char of the value if not present.Nr)super __post_init__rr%valuelower)self __class__s r)r-zExpandChoice.__post_init__8sM  8 4:q)//11DHHH  r() r!r"r#r$rrr%r&r- __classcell__r1s@r)rr%sZ  C#222222222r(rceZdZdZdededededededed ee d e d ed d ffd Z ddZ d e eeefffd Zd e eeeffdZd e eeeffdZxZS)InquirerPyExpandControlzAn :class:`~prompt_toolkit.layout.UIControl` class that displays a list of choices. Reference the parameter definition in :class:`.ExpandPrompt`. choicesdefaultpointer separator expand_helpexpand_pointermarkersession_result multiselect marker_plreturnNc ||_||_d|_||_||_| |_||_t|||| dS)NF)r6r7r=r>) _pointer _separator _expanded_expand_pointer_marker _marker_pl _expand_helpr,__init__) r0r6r7r8r9r:r;r<r=r>r?r1s r)rIz InquirerPyExpandControl.__init__Eso  #- #' )#      r(ci|_ d}d}t|j|jD]\}}t |t s9t |t s$t |tstdt |t r|dz }n7t |tr|j n|d|d<||j|d<|dz }n#t$rtdwxYw|j |j j |j |j jddt|jdz |j|j j <d}t |j|dt r+|dz }t |j|dt +|j|krQt#|jD]>\}}t |dt r!|d|jkr ||_dS=dSdS) Nrziexpand prompt argument choices requires each choice to be type of dictionary or Separator or ExpandChoicerz3expand prompt choice requires a key 'key' to existsF)rr.nameenabledr.) _key_mapszip _raw_choicesr6 isinstancedictrrrrKeyErrorrappendrHr lenselected_choice_index enumerate_default)r0countseparator_count raw_choicechoicefirst_valid_choice_indexindexs r)_format_choicesz'InquirerPyExpandControl._format_choices`s^ EO&)$*;T\&J&J  " F":t44&z9=='z<@@ *Dj)44:#q(OO&j,??/ '.5M 5:DN6%=1 % &   %E   (,*)1      14DL0A0AA0Et(,-#$ &>?H)TT * $ ) $&>?H)TT *  %)A A A!*4 771133 3 O  " "OT5I#J K K K  " "T\$"<=fEF   r(c~g}|d|jf|d|dr|jn|jft |dt s&|d|d|jf|d|d|df|S)Nra class:markerrMr.r)z[SetCursorPosition]rbrL)rTrBrFrGrQrrCr0r\rds r)_get_hover_textz'InquirerPyExpandControl._get_hover_texts?@@@ &y 1F t    &/955   " " F5MM4??"KL    :;;;@AAAr(cg}|dt|jdzf|d|dr|jn|jft |dt sD|d|d|jf|d|dfn|d|df|S) Nrb rfrMr.rrLzclass:separator)rTrUrBrFrGrQrrCrgs r)_get_normal_textz(InquirerPyExpandControl._get_normal_textsC $6$6$<=>>> &y 1F t    &/955 H  " "B&--(Q#R S S S  " "Bv#7 8 8 8 8  " "$5vf~#F G G Gr(r@N)r!r"r#r$rrr%rrrboolrIr_rrrcrhrkr2r3s@r)r5r5?sP  &          !!89          60000dU38_(="eCHo)>"$uS#X*?r(r5cC&eZdZdZdddddddddd ezdddddddedddd dd d d d d d dfd edededee de de de de de de dee de de de dee egefdee egefdeeee fdeeee fde d e d!e d"e d#eed$e d%eed&e d'e d(e d)e d*e d+e d,eed-dfBfd. Zd7d/Zd7d0Zd7d1Zed-e fd2Zd-eee e fffd3 Zd8d4ee d-dfd5Zd7fd6 ZxZS)9ruCreate a compact prompt with the ability to expand. A wrapper class around :class:`~prompt_toolkit.application.Application`. Contains a list of chocies binded to a shortcut letter. The prompt can be expanded using `h` key. 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/prompts/expand: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 of the prompt. This will be used to determine which choice is highlighted (current selection), The default value should the value of one of the choices. For :class:`.ExpandPrompt` specifically, default value can also be a `choice["key"]` which is the shortcut key for the choice. Refer to :ref:`pages/dynamic:default` documentation for more details. separator: Separator symbol. Custom symbol that will be used as a separator between the choice index number and the choices. help_msg: This parameter is DEPRECATED. Use expand_help instead. expand_help: The help configuration for the prompt. Must be an instance of :class:`.ExpandHelp`. If this value is None, the default help key will be binded to `h` and the default help message would be "Help, List all choices." expand_pointer: Pointer symbol before prompt expansion. Custom symbol that will be displayed to indicate the prompt is not expanded. 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. 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. border: Create border around the choice window. keybindings: Customise the builtin keybindings. Refer to :ref:`pages/kb:Keybindings` for more details. show_cursor: Display cursor at the end of the prompt. Set to False to hide the cursor. 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.expand(message="Select one:", choices[{"name": "1", "value": "1", "key": "a"}]).execute() >>> print(result) "1" rbNF?rjz) rz%s z Invalid inputTzMandatory promptr r6r7stylevi_modeqmarkamarkr8r9help_msgr:r; instructionlong_instruction transformerfilterheight max_heightr>r<r?bordervalidateinvalid_message keybindings show_cursorcycle wrap_linesraise_keyboard_interrupt mandatorymandatory_messager=r@c!| t| } | |_t|||| | | ||| | |_t |||||||| |||||||||||||||| dS)N)r ) r6r7r8r9r:r;r<r?r=r>)r r6rpr{rqrrrsrurvrwrxryrzr|r}r>r~rrrrrrr=)rrHr5content_controlr,rI)"r0r r6r7rprqrrrsr8r9rtr:r;rurvrwrxryrzr>r<r?r{r|r}r~rrrrrrr=r1s" r)rIzExpandPrompt.__init__sF  $X666K'8O#))# 9 9 9  #-#!+###!%=/)1      r(cfd}jjD].}t|dts||d/dS)zOverride this method to apply custom keybindings. Needs to creat these kb in the callback due to `after_render` retrieve the choices asynchronously. crdfd }|S)Nr@cjjkrjj j_dSjjj_dSN)rHrrrDrNrV)_rr0s r) keybindingzIExpandPrompt._on_rendered..keybinding_factory..keybindingesO$+///9=9M9W5WD(222,6s;(>>>r(rl) register_kbr/)rrr0s` r)keybinding_factoryz5ExpandPrompt._on_rendered..keybinding_factorydsP   ciikk * *      + *  r(r.rN)rr6rQr)r0rrr\s` r) _on_renderedzExpandPrompt._on_rendered]si     *2 2 2FfWoy99 2""6%=111 2 2r(c$|jjsdS tj||}t |jjdt s't |jjdtsdS|r|js| |dS)zpHandle the event when user attempt to move up. Overriding this method to skip the help choice. NTr.) rrDr _handle_uprQ selectionrr_cycle _handle_downr0eventcaps r)rzExpandPrompt._handle_upts #-  F  +D%88C$.w7  !5!?!H*UU t{%%e,,,E r(c|jjsdS tj||}t |jjdt s't |jjdtsdSt |jjdtr|js| |dS|r|js| |dS)zrHandle the event when user attempt to move down. Overriding this method to skip the help choice. NTr.) rrDr rrQrrrrrrs r)rzExpandPrompt._handle_downs #-  F  -dE::C$.w7  !5!?!H*UU 4/9'BJOO   &&&t{OOE***E r(c|js4dd|jjzn|jS)zConstruct the instruction behind the question. If _instruction exists, use that. :return: The instruction text. z(%s)rb) _instructionjoinrrNkeys)r0s r)ruzExpandPrompt.instructionsB$ #FRWWT1;@@BBCC C C" r(ct}|jds'|d|jjdf|S)zReturn the formatted text to display in the prompt. Overriding this method to allow multiple formatted class to be displayed. answeredz class:inputr)r,_get_prompt_messagestatusrTrr)r0display_messager1s r)rz ExpandPrompt._get_prompt_messages\  ''5577{:&   " " 4 >u EF   r(r.c|jjsdS|jjD]I}t|dtst|dt r9|r|n|d |d<JdS)zgOverride this method to ignore `ExpandHelp`. :param value: Specify a value to toggle. Nr.rM)rrDr6rQrr)r0rr.r\s r)_handle_toggle_allzExpandPrompt._handle_toggle_alls #-  F*2 J JF&/955 w:: ). Iy8I4IF9    J Jr(cf|jjsdSt|dS)z:Override this method to ignore keypress when not expanded.N)rrDr,_handle_toggle_choice)r0rr1s r)rz"ExpandPrompt._handle_toggle_choices4#-  F %%e,,,,,r(rlr) r!r"r#r$r rrrrrrmr%rrrr intrrrrIrrrpropertyrurrrrrr2r3s@r)rrsdHH\&(+/0,0#&AA "6:15,004!115.7; )-!3<@CK K "K 'K # K ( K  K K K K K K j)K K K K hucz23!K "3%*-.#K $sCx)%K &U38_-'K ()K *+K ,-K ./K 0-.1K 23K 4345K 67K 89K :;K <#'=K >?K @AK B!!89CK D EK K K K K K Z2222.$0  S    X   T%S/%:       J J8D> JT J J J J----------r(rN)%r$ dataclassesrtypingrrrrrr InquirerPy.baser r InquirerPy.base.controlr InquirerPy.enumr InquirerPy.exceptionsrrInquirerPy.prompts.listrInquirerPy.separatorrInquirerPy.utilsrrrrrrr__all__rrr5rr'r(r)rs;;!!!!!!>>>>>>>>>>>>>>>>CCCCCCCC******777777FFFFFFFF......****** 9 8 8  , , , , , , ,  , 22222622 22CCCCC5CCCLE-E-E-E-E-:E-E-E-E-E-r(