o j(@sdZddlmZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZmZmZddlmZd d lmZe rBd d lmZgd ZGd ddedZGdddeZGdddeZGdddeZGdddeZdS)zQ Margin implementations for a :class:`~prompt_toolkit.layout.containers.Window`. ) annotations)ABCMetaabstractmethod)Callable) TYPE_CHECKING) FilterOrBool to_filter)StyleAndTextTuplesfragment_list_to_textto_formatted_text) get_cwidth) UIContent)WindowRenderInfo)MarginNumberedMarginScrollbarMarginConditionalMargin PromptMarginc@s,eZdZdZedddZedd dZdS)rz& Base interface for a margin. get_ui_contentCallable[[], UIContent]returnintcCdS)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. rselfrrr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/layout/margins.py get_width&s zMargin.get_widthwindow_render_inforwidthheightr cCsgS)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`.) rrrr r!rrr create_margin1szMargin.create_marginNrrrrrrr rr!rrr )__name__ __module__ __qualname____doc__rrr#rrrrr!s   r) metaclassc@s2eZdZdZ dddd Zdd dZdddZdS)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. Frelativerdisplay_tildesrNonecCst||_t||_dSN)rr+r,)rr+r,rrr__init__Ps zNumberedMargin.__init__rrrcCs|j}tdt|dS)Nr ) line_countmaxlen)rrr1rrrrVszNumberedMargin.get_widthrrr r!r c Cs|}d}d}|jjj}g}d} t|jD]M\} } | | kr\| dur#n9| |krD|r5||d| dfn'||d| d|fn|rNt| |d} ||d| d|f| } |dq| r{| |j kr{|d| d7} | |j ksm|S)Nzclass:line-numberzclass:line-number.currentz%ir z%i  )z class:tildez~ ) r+ ui_contentcursor_positiony enumeratedisplayed_linesappendrjustabsr, window_height) rrr r!r+style style_currentcurrent_linenoresult last_linenor9linenorrrr#Zs6     zNumberedMargin.create_marginN)FF)r+rr,rrr-r$r%r&r'r(r)r/rr#rrrrrFs  rc@s.eZdZdZddd Zdd dZdddZdS)rzJ Wrapper around other :class:`.Margin` classes to show/hide them. marginrfilterrrr-cCs||_t||_dSr.)rGrrH)rrGrHrrrr/szConditionalMargin.__init__rrrcCs|r |j|SdS)Nr)rHrGrrrrrrs zConditionalMargin.get_widthrrr r!r cCs |r|r|j|||SgSr.)rHrGr#r"rrrr#s zConditionalMargin.create_marginN)rGrrHrrr-r$r%rFrrrrrs   rc@s6eZdZdZ   ddd d ZdddZdddZdS)rzc Margin displaying a scrollbar. :param display_arrows: Display scroll up/down arrows. F^vdisplay_arrowsrup_arrow_symbolstrdown_arrow_symbolrr-cCst||_||_||_dSr.)rrKrLrN)rrKrLrNrrrr/s  zScrollbarMargin.__init__rrrcCr)Nr rrrrrrszScrollbarMargin.get_widthrrr r!r csJ|j}|j}|}|r|d8}z$t|jt|}|jt|}tt|t d||t||Wn t y?gYSwdfdd } g} |rV| d |j fd gd } d } d } d}t |D]6}| |r~| |dsv| |dfn| | dfn| |dr| | dfn| | df| dqb|r| d |jf| S)Nr rowrrboolcs|ko kSS)z/True if we should display a button on this row.r)rPscrollbar_height 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)rPrrrQ)content_heightr?rKr3r;floatvertical_scrollrminr2ZeroDivisionErrorextendrLranger<rN)rrr r!rWr?rKfraction_visiblefraction_aboverUrCscrollbar_backgroundscrollbar_background_startscrollbar_buttonscrollbar_button_endirrRrr#sR      zScrollbarMargin.create_marginN)FrIrJ)rKrrLrMrNrMrr-r$r%rFrrrrrs rc@s2eZdZdZ ddd d ZdddZdddZdS)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. N get_prompt Callable[[], StyleAndTextTuples]get_continuation5None | Callable[[int, int, bool], StyleAndTextTuples]rr-cCs||_||_dSr.)rerg)rrergrrrr/s zPromptMargin.__init__rrrcCst|}t|S)z Width to report to the `Window`.)r rer )rrtextrrrrs zPromptMargin.get_widthrrr r!r c Csd|j}g}|t||r0d}|jddD]}|d|t|||||k|}q|S)Nr r4)rgr\r rer;r<)rrr r!rgrClast_yr9rrrr#s zPromptMargin.create_marginr.)rerfrgrhrr-r$r%rFrrrrrs  rN)r) __future__rabcrrcollections.abcrtypingrprompt_toolkit.filtersrrprompt_toolkit.formatted_textr r r prompt_toolkit.utilsr controlsr containersr__all__rrrrrrrrrs"       %DV