U Z+d(@sdZddlmZddlmZmZddlmZmZm Z ddl m Z m Z ddl mZmZmZddlmZdd lmZerdd lmZd d d ddgZGdd d edZGdd d eZGdddeZGdd d eZGdddeZdS)zQ Margin implementations for a :class:`~prompt_toolkit.layout.containers.Window`. ) annotations)ABCMetaabstractmethod) TYPE_CHECKINGCallableOptional) FilterOrBool to_filter)StyleAndTextTuplesfragment_list_to_textto_formatted_text) get_cwidth) UIContent)WindowRenderInfoMarginNumberedMarginScrollbarMarginConditionalMargin PromptMarginc@s<eZdZdZedddddZeddddd d d Zd S) rz& Base interface for a margin. Callable[[], UIContent]intget_ui_contentreturncCsdS)a  Return the width that this margin is going to consume. :param get_ui_content: Callable that asks the user control to create a :class:`.UIContent` instance. This can be used for instance to obtain the number of lines. rselfrrrA/tmp/pip-unpacked-wheel-blk8czbf/prompt_toolkit/layout/margins.py get_width$s zMargin.get_widthrr window_render_infowidthheightrcCsgS)a Creates a margin. This should return a list of (style_str, text) tuples. :param window_render_info: :class:`~prompt_toolkit.layout.containers.WindowRenderInfo` instance, generated after rendering and copying the visible part of the :class:`~prompt_toolkit.layout.controls.UIControl` into the :class:`~prompt_toolkit.layout.containers.Window`. :param width: The width that's available for this margin. (As reported by :meth:`.get_width`.) :param height: The height that's available for this margin. (The height of the :class:`~prompt_toolkit.layout.containers.Window`.) rrr!r"r#rrr create_margin/szMargin.create_marginN)__name__ __module__ __qualname____doc__rrr%rrrrrs  ) metaclassc@sHeZdZdZdddddddZdd d d d Zd d d ddddZdS)ra Margin that displays the line numbers. :param relative: Number relative to the cursor position. Similar to the Vi 'relativenumber' option. :param display_tildes: Display tildes after the end of the document, just like Vi does. FrNone)relativedisplay_tildesrcCst||_t||_dSN)r r,r-)rr,r-rrr__init__Ns zNumberedMargin.__init__rrrcCs|j}tdtd|dS)Nz%sr) line_countmaxlen)rrr1rrrrTszNumberedMargin.get_widthrr r c Cs|}d}d}|jjj}g}d} t|jD]\} } | | kr| dkrFnr| |kr|rj||d| dfq||d| d|fn0|rt| |d} ||d| d|f| } |dq,| r| |j kr|d| d7} q|S)Nzclass:line-numberzclass:line-number.currentz%irz%i  )z class:tildez~ ) r,Z ui_contentZcursor_positiony enumeratedisplayed_linesappendrjustabsr- window_height) rr!r"r#r,styleZ style_currentZcurrent_linenoresultZ last_linenor7linenorrrr%Xs4     zNumberedMargin.create_marginN)FFr&r'r(r)r/rr%rrrrrDs  c@sFeZdZdZddddddZdd d d d Zd d d ddddZdS)rzJ Wrapper around other :class:`.Margin` classes to show/hide them. rrr+)marginfilterrcCs||_t||_dSr.)rBr rC)rrBrCrrrr/szConditionalMargin.__init__rrrcCs|r|j|SdSdS)Nr)rCrBrrrrrrs zConditionalMargin.get_widthrr r cCs$|r|r|j|||SgSdSr.)rCrBr%r$rrrr%s zConditionalMargin.create_marginNrArrrrrsc@sJeZdZdZddddddd d Zd d d ddZdd d ddddZdS)rzc Margin displaying a scrollbar. :param display_arrows: Display scroll up/down arrows. F^vrstrr+)display_arrowsup_arrow_symboldown_arrow_symbolrcCst||_||_||_dSr.)r rGrHrI)rrGrHrIrrrr/s zScrollbarMargin.__init__rrrcCsdS)NrrrrrrrszScrollbarMargin.get_widthrr r cs\|j}|j}|}|r |d8}zHt|jt|}|jt|}tt|t d||t||Wnt k rgYSXdddfdd } g} |r| d|j fd gd } d } d } d }t |D]p}| |r| |ds| |dfn| | dfn,| |dr$| | dfn| | df| dq|rT| d|jf| SdS)Nrrbool)rowrcs|kokSS)z/True if we should display a button on this row.r)rLZscrollbar_heightZ scrollbar_toprris_scroll_buttonsz7ScrollbarMargin.create_margin..is_scroll_buttonzclass:scrollbar.arrow)zclass:scrollbarr6zclass:scrollbar.backgroundz*class:scrollbar.background,scrollbar.startzclass:scrollbar.buttonz$class:scrollbar.button,scrollbar.end r4)content_heightr=rGr3r9floatZvertical_scrollrminr2ZeroDivisionErrorextendrHranger:rI)rr!r"r#rPr=rGZfraction_visibleZfraction_aboverNr?Zscrollbar_backgroundZscrollbar_background_startZscrollbar_buttonZscrollbar_button_endirrMrr%sP      zScrollbarMargin.create_marginN)FrDrErArrrrrs c@sHeZdZdZdddddddZd d d d d Zdd d ddddZdS)ra [Deprecated] Create margin that displays a prompt. This can display one prompt at the first line, and a continuation prompt (e.g, just dots) on all the following lines. This `PromptMargin` implementation has been largely superseded in favor of the `get_line_prefix` attribute of `Window`. The reason is that a margin is always a fixed width, while `get_line_prefix` can return a variable width prefix in front of every line, making it more powerful, especially for line continuations. :param get_prompt: Callable returns formatted text or a list of `(style_str, type)` tuples to be shown as the prompt at the first line. :param get_continuation: Callable that takes three inputs. The width (int), line_number (int), and is_soft_wrap (bool). It should return formatted text or a list of `(style_str, type)` tuples for the next lines of the input. Nz Callable[[], StyleAndTextTuples]z5None | Callable[[int, int, bool], StyleAndTextTuples]r+) get_promptget_continuationrcCs||_||_dSr.)rWrX)rrWrXrrrr/ szPromptMargin.__init__rrrcCst|}t|S)z Width to report to the `Window`.)r rWr )rrtextrrrrs zPromptMargin.get_widthrr r c Csd|j}g}|t||r`d}|jddD],}|d|t|||||k|}q2|S)Nrr4)rXrTr rWr9r:)rr!r"r#rXr?Zlast_yr7rrrr%s zPromptMargin.create_margin)NrArrrrrs  N)r) __future__rabcrrtypingrrrZprompt_toolkit.filtersrr Zprompt_toolkit.formatted_textr r r Zprompt_toolkit.utilsr controlsrZ containersr__all__rrrrrrrrrs(     %DV