U ou f(@sdZddlmZddlmZmZddlmZmZddl m Z m Z ddl m Z mZmZddlmZdd lmZer|dd lmZd d d ddgZGdd d edZGdd d eZGdddeZGdd d eZGdddeZdS)zQ Margin implementations for a :class:`~prompt_toolkit.layout.containers.Window`. ) annotations)ABCMetaabstractmethod) TYPE_CHECKINGCallable) 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-8milen2s/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)rr+r,)rr+r,rrr__init__Ns zNumberedMargin.__init__rrrcCs|j}tdtd|dS)Nz%sr ) line_countmaxlen)rrr0rrrrTszNumberedMargin.get_widthrr rc 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%ir z%i  )z class:tildez~ ) r+Z ui_contentZcursor_positiony enumeratedisplayed_linesappendrjustabsr, window_height) rr r!r"r+styleZ style_currentZcurrent_linenoresultZ last_linenor6linenorrrr$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-)rArrB)rrArBrrrr.szConditionalMargin.__init__rrrcCs|r|j|SdSdS)Nr)rBrArrrrrrs zConditionalMargin.get_widthrr rcCs$|r|r|j|||SgSdSr-)rBrAr$r#rrrr$s zConditionalMargin.create_marginNr@rrrrrsc@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-)rrFrGrH)rrFrGrHrrrr.s zScrollbarMargin.__init__rrrcCsdS)Nr rrrrrrszScrollbarMargin.get_widthrr rcs\|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)Nr rbool)rowrcs|kokSS)z/True if we should display a button on this row.r)rKZscrollbar_heightZ scrollbar_toprris_scroll_buttonsz7ScrollbarMargin.create_margin..is_scroll_buttonzclass:scrollbar.arrow)zclass:scrollbarr5zclass:scrollbar.backgroundz*class:scrollbar.background,scrollbar.startzclass:scrollbar.buttonz$class:scrollbar.button,scrollbar.end r3)content_heightr<rFr2r8floatZvertical_scrollrminr1ZeroDivisionErrorextendrGranger9rH)rr r!r"rOr<rFZfraction_visibleZfraction_aboverMr>Zscrollbar_backgroundZscrollbar_background_startZscrollbar_buttonZscrollbar_button_endirrLrr$sP      zScrollbarMargin.create_marginN)FrCrDr@rrrrrs 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-)rVrW)rrVrWrrrr. szPromptMargin.__init__rrrcCst|}t|S)z Width to report to the `Window`.)r rVr )rrtextrrrrs zPromptMargin.get_widthrr rc Csd|j}g}|t||r`d}|jddD],}|d|t|||||k|}q2|S)Nr r3)rWrSr rVr8r9)rr r!r"rWr>Zlast_yr6rrrr$s zPromptMargin.create_margin)Nr@rrrrrs  N)r( __future__rabcrrtypingrrZprompt_toolkit.filtersrrZprompt_toolkit.formatted_textr r r Zprompt_toolkit.utilsr controlsrZ containersr__all__rrrrrrrrrs(     %DV