ó ±xYc@ssddlmZddlmZmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZmZdd lmZmZdd lmZdd lmZdd lmZmZdd lZdZ defd„ƒYZ!d„Z"defd„ƒYZ#defd„ƒYZ$defd„ƒYZ%defd„ƒYZ&d S(iÿÿÿÿ(tunicode_literals(t zip_longesttrange(tHasCompletionstIsDonet Conditiont to_cli_filter(tMouseEventType(tToken(t get_cwidthi(tWindowtHSplittConditionalContainert ScrollOffsets(t UIControlt UIContent(tLayoutDimension(tScrollbarMargin(tPointtCharNuCompletionsMenuuMultiColumnCompletionsMenutCompletionsMenuControlcBsweZdZdZd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „ZRS( uJ Helper for drawing the complete menu to the screen. :param scroll_offset: Number (integer) representing the preferred amount of completions to be displayed before and after the current one. When this is a very high number, the current completion will be shown in the middle most of the time. icCstjj|_dS(N(RtMenut Completionsttoken(tself((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyt__init__%scCstS(N(tFalse(Rtcli((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyt has_focus(scCsF|jj}|r>|jd|ƒ}|jd|ƒ}||SdSdS(Niôi(tcurrent_buffertcomplete_statet_get_menu_widtht_get_menu_meta_width(RRtmax_available_widthRt menu_widthtmenu_meta_width((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pytpreferred_width+s  cCs'|jj}|rt|jƒSdSdS(Ni(RRtlentcurrent_completions(RRtwidthtmax_available_heightt wrap_linesR((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pytpreferred_height5s  c sÆ|jj}|r¿|j‰|j‰ˆj||ƒ‰ˆj|ˆ|ƒ‰ˆj|ƒ‰‡‡‡‡‡‡fd†}td|dtdddˆpšdƒdt ˆƒdt d ˆj ƒƒStƒS( u= Create a UIContent object for this control. csQˆ|}|ˆk}ˆj||ˆƒ}ˆrM|ˆj||ˆƒ7}n|S(N(t_get_menu_item_tokenst_get_menu_item_meta_tokens(titctis_current_completiontresult(t completionstindexR#R"Rt show_meta(s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pytget_lineJs   R4tcursor_positiontxityt line_countt default_charu ( RRR&tcomplete_indexRR t _show_metaRRR%RR(RRR'theightRR4((R1R2R#R"RR3s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pytcreate_content<s     cCstd„|jDƒƒS(uT Return ``True`` if we need to show a column with meta information. css|]}|jVqdS(N(t display_meta(t.0R.((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pys ^s(tanyR&(RR((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR;ZscCs0t|t|jtd„|jDƒƒdƒƒS(u6 Return the width of the main column. css|]}t|jƒVqdS(N(R tdisplay(R?R.((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pys dsi(tmintmaxt MIN_WIDTHR&(Rt max_widthR((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR`scCs;|j|ƒr3t|td„|jDƒƒdƒSdSdS(u6 Return the width of the meta column. css|]}t|jƒVqdS(N(R R>(R?R.((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pys lsiiN(R;RBRCR&(RRER((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR gscCsi|r|jjj}n |jj}t|j|dƒ\}}d|d|}|d||ffgS(Niu u %s%s (Rt CompletiontCurrentt _trim_textRA(Rt completionR/R'Rttextttwtpadding((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR+qs  cCsi|r|jjj}n |jj}t|j|dƒ\}}d|d|}|d||ffgS(Niu u %s%s (RtMetaRGRHR>(RRIR/R'RRJRKRL((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR,{s  cCs”|j}|jtjkr:|j|jjƒd|_nV|jtj kre|j dddt ƒn+|jtj kr|j dddt ƒndS(u> Handle mouse events: clicking and scrolling. tcountitdisable_wrap_aroundN(Rt event_typeRtMOUSE_UPtgo_to_completiontpositionR7tNoneRt SCROLL_DOWNt complete_nexttTruet SCROLL_UPtcomplete_previous(RRt mouse_eventtb((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyt mouse_handler…s  (t__name__t __module__t__doc__RDRRR$R*R=R;RR R+R,R\(((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyRs       cCsÂt|ƒ}||kr´t|ƒ|krY|td|dƒ d| }|t|ƒfSd}x5|D]-}t||ƒ|dkrf||7}qfqfW|d7}|t|ƒfSn ||fSdS(uo Trim the text to `max_width`, append dots when the text is too long. Returns (text, width) tuple. iiu...uN(R R%RC(RJRER't trimmed_textR.((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyRH™s    tCompletionsMenucBseZddeed„ZRS(icCs§t|ƒ}t|ƒ}tt|ƒjdtdtƒdtddƒdtddd|ƒdtd |d |ƒd td |ƒgd t ƒdt ƒt ƒ@|@ƒdS(NtcontentR'RBiR<iRCtscroll_offsetsttoptbottomt right_marginstdisplay_arrowstdont_extend_widthtfilter( RtsuperRaRR RRR RRWRR(Rt max_heightt scroll_offsett extra_filterRg((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR¶s    N(R]R^RTRWRR(((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyRaµst MultiColumnCompletionMenuControlcBskeZdZdZddd„Zd„Zd„Zd„Zd„Zd„Z d „Z d „Z d „Z RS( u† Completion menu that displays all the completions in several columns. When there are more completions than space for them to be displayed, an arrow is shown on the left or right side. `min_rows` indicates how many rows will be available in any possible case. When this is langer than one, in will try to use less columns and more rows until this value is reached. Be careful passing in a too big value, if less than the given amount of rows are available, more columns would have been required, but `preferred_width` doesn't know about that and reports a too small value. This results in less completions displayed and additional scrolling. (It's a limitation of how the layout engine currently works: first the widths are calculated, then the heights.) :param suggested_max_column_width: The suggested max width of a column. The column can still be bigger than this, but if there is place for two columns of this width, we will display two columns. This to avoid that if there is one very wide completion, that it doesn't significantly reduce the amount of columns. iicCsŽt|tƒr|dks!t‚||_||_tjj|_d|_ d|_ d|_ d|_ i|_ t|_t|_d|_dS(Nii(t isinstancetinttAssertionErrortmin_rowstsuggested_max_column_widthRRRRtscrollt_rendered_rowst_rendered_columnst_total_columnst_render_pos_to_completionRt_render_left_arrowt_render_right_arrowt _render_width(RRrRs((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyRàs!         cCs d|_dS(Ni(Rt(R((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pytresetñscCstS(N(R(RR((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyRôscCs…|jj}|j|ƒ}t|tjt|jƒt|j ƒƒƒ}x-||kry|||j kry||8}qMW||j S(u{ Preferred width: prefer to use at least min_rows, but otherwise as much as possible horizontally. ( RRt_get_column_widthRptmathtceilR%R&tfloatRrt_required_margin(RRR!Rt column_widthR0((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR$÷s  /"cCs[|jj}|j|ƒ}td||j|ƒ}ttjt|j ƒt |ƒƒƒS(u^ Preferred height: as much as needed in order to display all the completions. i( RRR}RCRRpR~RR%R&R€(RRR'R(R)RR‚t column_count((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR*s cs |jj‰|jˆƒ}i|_d d„}‡‡fd†}d}ˆrt|||ƒ}||jkr„|||j}ntd||j|ƒ}t ||ˆj ƒƒ} t t | Œƒ} ˆj pÔd|} t| t|j | |dƒƒ|_ |j dk} |j t| dƒ|k} g‰x`t| ƒD]I\}}g}|t| ƒdk}| r“|tj|rƒdndfg7}nx¢t||j | ƒD]‰\}‰ˆd k r||jˆ|ˆƒ|ƒ7}xOt|ƒD]!}ˆ|j||||f ABC DEF Gxx t fillvalue(titerR(tntiterableR„targs((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pytgrouperscsˆjdk oˆˆjkS(uB Returns True when this completion is the currently selected one. N(R:RTtcurrent_completion(RI(R.R(s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR/siiiiucsˆ|S(N((R-(ttokens_for_line(s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR4hsR4R8N(RRR}RxRTRBRsRCRtlistR&tzipR:RtR%t enumerateRt ScrollbarR+RRRFtappendRuRvRwRyRzR{R(RRR'R<R‚R‰R/tHORIZONTAL_MARGIN_REQUIREDtvisible_columnstcolumns_trows_tselected_columntrender_left_arrowtrender_right_arrowt row_indextrowttokenst middle_rowt column_indexR6R4((R.RR‹s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR=sT   &%$ "!%    cCstd„|jDƒƒdS(u2 Return the width of each column. css|]}t|jƒVqdS(N(R RA(R?R.((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pys qsi(RCR&(RR((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR}mscCse|r|jjj}n |jj}t|j|ƒ\}}d||d}|d||ffgS(Nu iu %s%s(RRFRGRHRA(RRIR/R'RRJRKRL((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR+ss  cs|j‰‡‡fd†}‡‡fd†}|jtjkrI|ƒnÂ|jtjkre|ƒn¦|jtjkr |jj}|jj}|dkr±ˆj r|ƒqq |ˆj dkrÚˆj r|ƒqq ˆj j ||fƒ}|r ˆj|ƒq ndS(u0 Handle scoll and click events. cs6ˆjdˆjdtƒtdˆjdƒˆ_dS(NRNROii(RYRuRWRCRt((R[R(s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyt scroll_left„scs@ˆjdˆjdtƒtˆjˆjˆjdƒˆ_dS(NRNROi(RVRuRWRBRwRvRt((R[R(s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyt scroll_rightˆsiiN(RRPRRURXRQRSR6R7RyR{RzRxtgettapply_completion(RRRZRRžR6R7RI((R[Rs:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR\~s&          ( R]R^R_RRR|RR$R*R=R}R+R\(((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyRnÈs    \  tMultiColumnCompletionsMenucBs#eZdZddeed„ZRS(uÍ Container that displays the completions in several columns. When `show_meta` (a :class:`~prompt_toolkit.filters.CLIFilter`) evaluates to True, it shows the meta information at the bottom. iic CsÐt|ƒ}t|ƒ}tƒtƒ@|@}td„ƒ}tdtdtd|d|ƒdtddƒdtdd ƒƒd |ƒ}tdtdtƒƒd ||@|@ƒ}t t |ƒj ||gƒdS( NcSstd„|jjjDƒƒS(Ncss|]}|jVqdS(N(R>(R?R.((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pys ¶s(R@RRR&(R((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pytµsRbRrRsR'RBiR<iRi( RRRRR R RnRt_SelectedCompletionMetaControlRjR¡R( RRrRsR3Rmt full_filtertany_completion_has_metatcompletions_windowt meta_window((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR­s"   (R]R^R_RWR(((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR¡§sR£cBs2eZdZd„Zd„Zd„Zd„ZRS(uH Control that shows the meta information of the selected token. cCs;|jjr3|jj}dtd„|jDƒƒSdSdS(u? Report the width of the longest meta text as the preferred width of this control. It could be that we use less width, but this way, we're sure that the layout doesn't change when we select another completion (E.g. that completions are suddenly shown in more or fewer columns.) icss|]}t|jƒVqdS(N(R R>(R?R.((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pys ÚsiN(RRRCR&(RRR!tstate((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR$Ðs  cCsdS(Ni((RRR'R(R)((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR*Þscs=|j|ƒ‰‡fd†}td|dˆr6dndƒS(NcsˆS(N((R-(Rš(s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR4äsR4R8ii(t _get_tokensR(RRR'R<R4((Ršs:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR=áscCsQtjjj}|jj}|rM|jrM|jjrM|d|jjfgSgS(Nu %s (RRRtMultiColumnMetaRRRŠR>(RRRR¨((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR©és  (R]R^R_R$R*R=R©(((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyR£Ìs    (uCompletionsMenuuMultiColumnCompletionsMenu('t __future__Rt six.movesRRtprompt_toolkit.filtersRRRRtprompt_toolkit.mouse_eventsRtprompt_toolkit.tokenRtprompt_toolkit.utilsR t containersR R R R tcontrolsRRt dimensionRtmarginsRtscreenRRR~t__all__RRHRaRnR¡R£(((s:lib/python2.7/site-packages/prompt_toolkit/layout/menus.pyts&"" ‚ ß%