a lcc}@sdZddlmZddlmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZmZddlmZmZddlmZmZddlmZdd lmZmZmZmZmZmZdd l m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:m;Z;mm?Z?ddl>m@ZAddl>mBZBddlCmDZDmEZEmFZFddlGmHZHmIZImJZJmKZKmLZLddlMmNZNmOZOddlPmQZQmRZRddlSmTZTddlUmVZVmWZWddlXmYZYgdZZe,Z[GdddZ\Gd d!d!Z]Gd"d#d#Z^Gd$d%d%Z_Gd&d'd'Z`Gd(d)d)ZaGd*d+d+Zbe d,ZcGd-d.d.eecZdGd/d0d0edecZeGd1d2d2edecZfGd3d4d4efegZhGd5d6d6ZiGd7d8d8ZjGd9d:d:Zkd;S)z#TextArea.__init__..csjSr{)rdr\r|r\r]r~rcs tjSr{)rrer\r|r\r]r~rcsjSr{)rfr\r|r\r]r~rcsjSr{)rcr\r|r\r]r~r) documentr`rlrdrerfrcrgrhrycsjSr{)rbr\r|r\r]r~r) processorfilterzclass:text-area.promptrt)bufferrbrxZsearch_buffer_controlrvrirjTZdisplay_arrowsrCzclass:text-area mincs tjSr{)rrkr\r|r\r]r~r) rnrmrorpcontentrtrk left_margins right_marginsrr) isinstancerDcontrolrdrerbrcrlrkrfrrrrrAr rr,r;r8r6rrr9rr7r5r4Dexactr*window)r}r_r`rarbrcrdrerfrgrhrirjrkrlrmrnrorprqrrrsrtrurvrwrxryZsearch_controlrrr\r|r]__init__s          zTextArea.__init__rzcCs|jjS)z$ The `Buffer` text. )rr_r|r\r\r]r_sz TextArea.textvaluerzcCst|d|_dSNr)rrr}rr\r\r]r_$scCs|jjS)zA The `Buffer` document (text + cursor position). )rrr|r\r\r]r(szTextArea.documentcCs|jj|dddS)NT)Zbypass_readonly)rZ set_documentrr\r\r]r/scCs|jjS)zM The accept handler. Called when the user accepts the input. rrgr|r\r\r]rg3szTextArea.accept_handlercCs ||j_dSr{rrr\r\r]rg:scCs|jSr{rr|r\r\r]__pt_container__>szTextArea.__pt_container__)r^TFNNNTNNNTFTFNNFFFNFr^NTr^Nr^)rRrSrTrUstrrrr<r rrBrrr/boolr.rDrrr:rpropertyr_setterrrrgr$rr\r\r\r]rEms>  srEc @sZeZdZdZddddejdfeeee e e ee geffe ddddZ ed d d ZdS) rFa Widget that displays the given text. It is not editable or focusable. :param text: Text to display. Can be multiline. All value types accepted by :class:`prompt_toolkit.layout.FormattedTextControl` are allowed, including a callable. :param style: A style string. :param width: When given, use this width, rather than calculating it from the text size. :param dont_extend_width: When `True`, don't take up more width than preferred, i.e. the length of the longest line of the text, or value of `width` parameter, if given. `True` by default :param dont_extend_height: When `True`, don't take up more width than the preferred height, i.e. the number of lines of the text. `False` by default. r^NTF)r_rtrmrorpalignrkrzc sX|_tdfdd }tfddd_tj|tddd |||||d _dS) NrcsRdurJtj}t|}|r6tdd|D}n tddSt|dSSdS)Ncss|]}t|VqdSr{r?).0liner\r\r] irz4Label.__init__..get_width..r preferred)rr_rmax splitlinesr)Ztext_fragmentsr_Z longest_liner}rmr\r] get_widthds   z!Label.__init__..get_widthcsjSr{r_r\r|r\r]r~prz Label.__init__..rrCrz class:label )rrmrnrtrorprrk)r_r/r-Zformatted_text_controlr*rr) r}r_rtrmrorprrkrr\rr]rUs  zLabel.__init__rcCs|jSr{rr|r\r\r]r}szLabel.__pt_container__)rRrSrTrUr+LEFTrrr/rr rrrr$rr\r\r\r]rFBs$ (rFc@s`eZdZdZdeeegdfeeeddddZe d d d Z e d d d Z e d ddZdS)rGaS Clickable button. :param text: The caption for the button. :param handler: `None` or callable. Called when the button is clicked. No parameters are passed to this callable. Use for instance Python's `functools.partial` to pass parameters to this callable if needed. :param width: Width of the button. N <>)r_handlerrm left_symbol right_symbolrzc sf|_|_|_|_|_tjdd_t dfdd }t jt j d||ddd_ dS) NT key_bindingsrircstjrdSdSdS)Nzclass:button.focusedz class:button)r layoutrr\r|r\r] get_stylesz"Button.__init__..get_stylerCF)rrnrmrtrpro)r_rrrrmr-_get_text_fragments_get_key_bindingsrrr*r+ZCENTERr)r}r_rrmrrrr\r|r]rs(  zButton.__init__rcsdjtjtj}d|dj}tddfdd }dj|fdd||fdj|fgS) Nz{:^} mouse_eventrzcs"jdur|jtjkrdSr{)r event_typer>MOUSE_UPrr|r\r]rs  z+Button._get_text_fragments..handlerzclass:button.arrowz[SetCursorPosition]r^zclass:button.text)rmr@rrformatr_r=)r}rmr_rr\r|r]rs  zButton._get_text_fragmentscs2t}|d|dtddfdd }|S)zKey bindings for the Button. enterNeventrzcsjdurdSr{)rrr|r\r]_s z#Button._get_key_bindings.._)raddE)r}kbrr\r|r]rs zButton._get_key_bindingscCs|jSr{rr|r\r\r]rszButton.__pt_container__)Nrrr)rRrSrTrUrrrintrrrrrr$rr\r\r\r]rGs   * rGc @s@eZdZdZd eeeeeee e ddddZ e dd d Z dS) rHa Draw a border around any container, optionally with a title text. Changing the title and body of the frame is possible at runtime by assigning to the `body` and `title` attributes of this class. :param body: Another container object. :param title: Text to be displayed in the top of the frame (can be formatted text). :param style: Style string to be applied to this widget. r^NF)bodytitlertrmrnrmodalrzc s`|_|_ttdd}d|}t|ddtjd|tjd|ddddtfdd d d d |dddd|tjd|ddtj dgdd } t|ddtjd|tjd|ddtj dgdd } t t dfdd } t t | | dt | | dt|dtjdtfdd |dtjdgddt|ddtjd|tjd|ddtjdgdd g|||||d_dS)Nzclass:frame.borderrz class:frame rC)rmrnchar)r|cstdjS)Nz {} )rrrr\r|r\r]r~rz Frame.__init__..zclass:frame.labelT)rtrprnrcs tjSr{)rrr\r|r\r] has_titlesz!Frame.__init__..has_title)rrrmrcsjSr{)rr\r|r\r]r~rrpadding)rmrnrtrr)rrrr*r)rQrXrVrFrYrrr(r#rWr%rZr[ container) r}rrrtrmrnrrfillZtop_row_with_titleZtop_row_without_titlerr\r|r]rsh               zFrame.__init__rcCs|jSr{rr|r\r\r]r6szFrame.__pt_container__)r^r^NNNF)rRrSrTrUr"rrr/rrrrr$rr\r\r\r]rHs$ PrHc@s.eZdZdZeddddZedddZdS) rIz Draw a shadow underneath/behind this container. (This applies `class:shadow` the the cells under the shadow. The Style should define the colors for the shadow.) :param body: Another container object. N)rrzc CsBt|tdddddtdddtdddddtdddgd|_dS) NrCTz class:shadowr)bottomrnleftright transparentr)rtoprmrrrrZfloats)r'r&r*r)r}rr\r\r]rCs(zShadow.__init__rcCs|jSr{rr|r\r\r]rZszShadow.__pt_container__)rRrSrTrUr"rr$rr\r\r\r]rI:srIc@s\eZdZdZd eeeeeeeeeedeegeffe e e dd ddZ e dd d ZdS) rJa@ Add padding around a container. This also makes sure that the parent can provide more space than required by the child. This is very useful when wrapping a small element with a fixed size into a ``VSplit`` or ``HSplit`` object. The ``HSplit`` and ``VSplit`` try to make sure to adapt respectively the width and height, possibly shrinking other elements. Wrapping something in a ``Box`` makes it flexible. :param body: Another container object. :param padding: The margin to be used around the body. This can be overridden by `padding_left`, padding_right`, `padding_top` and `padding_bottom`. :param style: A style string. :param char: Character to be used for filling the space around the body. (This is supposed to be a character with a terminal width of 1.) Nr^F) rr padding_left padding_right padding_toppadding_bottomrmrnrtrrrrzc  sdurtddttdfdd } | ||_| ||_| ||_| ||_||_tt|j| dt t|j| d|t|j| dgt|j| dg||| | dd|_ dS) Nrrrcs|dur }t|Sr{r1)rrr\r]getszBox.__init__..get)rnrr)rmrnrtrr) rr/rrrrrr(r*r)r)r}rrrrrrrmrnrtrrrrr\rr]rqs0          z Box.__init__rcCs|jSr{rr|r\r\r]rszBox.__pt_container__) NNNNNNNr^NFN)rRrSrTrUr"r/rr rrrrrr$rr\r\r\r]rJ^s8 1rJ_Tc@seZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ eed<d Z e ed <d Ze ed <deeeefeeed dddZd dddZedddZedddZd S) _DialogListz9 Common code for `RadioList` and `CheckboxList`. r^open_characterclose_charactercontainer_style default_styleselected_style checked_styleFmultiple_selectionTshow_scrollbarN)valuesdefault_valuesrzc st|dksJ|pg}|_dd|Dfdd|D_t|r\|dvr\|dn |dd_tjdkrjd_nd_t}|dtddfdd }|d tddfd d }|d tddfd d }|dtddfdd }|d|dtddfdd }|t j tddfdd } t j |dd_ tj jttddtfdddgdd_dS)NrcSsg|] \}}|qSr\r\)rrrr\r\r] rz(_DialogList.__init__..csg|]}|vr|qSr\r\)rr)keysr\r]rsuprcstdjd_dS)NrrC)r_selected_indexrr|r\r]_upsz!_DialogList.__init__.._upZdowncs ttjdjd_dSNrC)rlenrrrr|r\r]_downsz#_DialogList.__init__.._downZpageupcs.|jjj}|jr*tdjt|jj_dSr)apprcurrent_window render_inforrrdisplayed_linesrwr|r\r]_pageups  z%_DialogList.__init__.._pageupZpagedowncs8|jjj}|jr4ttjdjt|jj_dSr) rrrrrrrrrrr|r\r] _pagedowns   z'_DialogList.__init__.._pagedownrrcs dSr{) _handle_enterrr|r\r]_clicksz$_DialogList.__init__.._clickcsbtj}|jdd|D]<}tt|d}||jr j|_dSq dSr) listrrrrlower startswithdataindex)rrrr_r|r\r]_finds  z#_DialogList.__init__.._findTrrcsjSr{)rr\r|r\r]r~ rz&_DialogList.__init__..)marginr)rrtrro)rrcurrent_values current_valuerrrrrr!Anyr-rrr*rr3r5rr) r}rrrrrrrrrr\)rr}r]rsV     z_DialogList.__init__rcCsR|jr<|j|jd}||jvr.|j|qN|j|n|j|jd|_dSr)rrrrremoveappendr)r}valr\r\r]rs  z_DialogList._handle_entercsBtddfdd }g}tjD]\}}jr@|djv}n|djk}|jk}d}|rn|dj7}|r|dj7}| |j f|r| d|r| |dfn| |df| |j f| j df| t|d j d | d q"tt|D]$}||d||d |f||<q||S) Nrcs"|jtjkr|jj_dS)z Set `_selected_index` and `current_value` according to the y position of the mouse click event. N)rr>rpositionyrrrr|r\r] mouse_handlers  z6_DialogList._get_text_fragments..mouse_handlerrr^rr*rCr)r^ )r= enumeraterrrrrrrr rrrextendrrangerpop)r}r resultircheckedselectedrtr\r|r]rs4    "z_DialogList._get_text_fragmentscCs|jSr{rr|r\r\r]rGsz_DialogList.__pt_container__)N)rRrSrTrUrr__annotations__rrrrrrrrrrrrrrrrrr$rr\r\r\r]rs$           X .rcsXeZdZdZdZdZdZdZdZdZ dZ d e e e efee d d fd d ZZS)rMz} List of radio buttons. Only one can be checked at the same time. :param values: List of (value, label) tuples. ()zclass:radio-listz class:radiozclass:radio-selectedzclass:radio-checkedFN)rdefaultrzcs(|durd}n|g}tj||ddS)N)r)superr)r}rrr __class__r\r]rZszRadioList.__init__)N)rRrSrTrUrrrrrrrrrrrrr __classcell__r\r\rr]rMKsrMc@s,eZdZdZdZdZdZdZdZdZ dZ d S) rNz List of checkbox buttons. Several can be checked at the same time. :param values: List of (value, label) tuples. []zclass:checkbox-listzclass:checkboxzclass:checkbox-selectedzclass:checkbox-checkedTN) rRrSrTrUrrrrrrrr\r\r\r]rNgsrNcsXeZdZdZdZd eeddfdd Zeedd d Z e j edd d d Z Z S)rOz[Backward compatibility util: creates a 1-sized CheckboxList :param text: the text Fr^N)r_rrzcs"d|fg}tj|d||_dS)Nr)r)rrr)r}r_rrrr\r]rs zCheckbox.__init__rcCs d|jvSNrrr|r\r\r]rszCheckbox.checkedrcCs|rdg|_ng|_dSr"r#rr\r\r]rs )r^F) rRrSrTrUrrrrrrrrr\r\rr]rOwsrOc@s,eZdZdZddddZedddZdS)rKz3 A simple vertical line with a width of 1. NrcCsttjddd|_dS)Nzclass:line,vertical-linerC)rrtrm)r*rQrWrr|r\r\r]rszVerticalLine.__init__cCs|jSr{rr|r\r\r]rszVerticalLine.__pt_container__rRrSrTrUrr$rr\r\r\r]rKsrKc@s,eZdZdZddddZedddZdS)rLz6 A simple horizontal line with a height of 1. NrcCsttjddd|_dS)Nzclass:line,horizontal-linerC)rrtrn)r*rQrVrr|r\r\r]rszHorizontalLine.__init__cCs|jSr{rr|r\r\r]rszHorizontalLine.__pt_container__r$r\r\r\r]rLsrLc@sPeZdZddddZeedddZejeddddZedd d Z dS) rPNrcsnd_td_ttddtjdddtddddttdfdd d td fd d d gd gd_dS)N<z60%rCrr)rrrzclass:progress-bar.usedcsttjdS)Nweightrr _percentager\r|r\r]r~rz&ProgressBar.__init__..)rtrmzclass:progress-barcsttdjdS)Ndr&r(r\r|r\r]r~r)rrrrrr)r)rFlabelr'r*r&r)rr|r\r|r]rs0   zProgressBar.__init__cCs|jSr{)r)r|r\r\r] percentageszProgressBar.percentagercCs||_|d|j_dS)N%)r)r+r_rr\r\r]r,scCs|jSr{rr|r\r\r]rszProgressBar.__pt_container__) rRrSrTrrrr,rr$rr\r\r\r]rPs !rPN)lrU functoolsrtypingrrrrrrr r Z"prompt_toolkit.application.currentr Zprompt_toolkit.auto_suggestr r Zprompt_toolkit.bufferrrZprompt_toolkit.completionrrZprompt_toolkit.documentrZprompt_toolkit.filtersrrrrrrZprompt_toolkit.formatted_textrrrrZ#prompt_toolkit.formatted_text.utilsrZprompt_toolkit.historyrZ'prompt_toolkit.key_binding.key_bindingsrZ(prompt_toolkit.key_binding.key_processorr Zprompt_toolkit.keysr!Z prompt_toolkit.layout.containersr"r#r$r%r&r'r(r)r*r+Zprompt_toolkit.layout.controlsr,r-r.Zprompt_toolkit.layout.dimensionr/r0rr2Zprompt_toolkit.layout.marginsr3r4r5Z prompt_toolkit.layout.processorsr6r7r8r9r:Zprompt_toolkit.lexersr;r<Zprompt_toolkit.mouse_eventsr=r>Zprompt_toolkit.utilsr@Zprompt_toolkit.validationrArBZtoolbarsrD__all__rrQrErFrGrHrIrJrrrMrNrrOrKrLrPr\r\r\r]sZ (        0       V?Y`$H#