o ˷e(@sdZddlmZmZddlmZmZmZddlm Z m Z ddl m Z m Z mZddlmZddlmZer:dd lmZgd ZGd d d ed ZGdddeZGdddeZGdddeZGdddeZdS)zQ Margin implementations for a :class:`~prompt_toolkit.layout.containers.Window`. )ABCMetaabstractmethod) TYPE_CHECKINGCallableOptional) FilterOrBool to_filter)StyleAndTextTuplesfragment_list_to_textto_formatted_text) get_cwidth) UIContent)WindowRenderInfo)MarginNumberedMarginScrollbarMarginConditionalMargin PromptMarginc @sLeZdZdZedegefdefddZeddded ede fd d Z d S) rz& Base interface for a margin. get_ui_contentreturncCdS)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. rselfrrrT/var/www/ideatree/venv/lib/python3.10/site-packages/prompt_toolkit/layout/margins.py get_width"s zMargin.get_widthwindow_render_inforwidthheightcCsgS)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`.) rrrrrrrr create_margin-szMargin.create_marginN) __name__ __module__ __qualname____doc__rrrintrr r!rrrrrs r) metaclassc@s^eZdZdZ ddededdfddZd egefdefd d Z d d dedede fddZ dS)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. Frelativedisplay_tildesrNcCst||_t||_dSN)rr(r))rr(r)rrr__init__Ls zNumberedMargin.__init__rcCs|j}tdtd|dS)Nz%sr ) line_countmaxlen)rrr-rrrrSszNumberedMargin.get_widthrrrrc 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) rrrrr(style style_currentcurrent_linenoresult last_linenor5linenorrrr!Ws6     zNumberedMargin.create_margin)FF) r"r#r$r%rr+rrr&rr r!rrrrrBs(  rc@sZeZdZdZdededdfddZdegefde fd d Z d d d e de de fddZ dS)rzJ Wrapper around other :class:`.Margin` classes to show/hide them. marginfilterrNcCs||_t||_dSr*)rBrrC)rrBrCrrrr+szConditionalMargin.__init__rcCs|r |j|SdS)Nr)rCrBrrrrrrs zConditionalMargin.get_widthrrrrcCs |r|r|j|||SgSr*)rCrBr!r rrrr!s zConditionalMargin.create_margin) r"r#r$r%rrr+rrr&rr r!rrrrrsrc @sfeZdZdZ   ddedededd fd d Zd egefde fd dZ ddde de de fddZ d S)rzc Margin displaying a scrollbar. :param display_arrows: Display scroll up/down arrows. F^vdisplay_arrowsup_arrow_symboldown_arrow_symbolrNcCst||_||_||_dSr*)rrFrGrH)rrFrGrHrrrr+s  zScrollbarMargin.__init__rcCr)Nr rrrrrrszScrollbarMargin.get_widthrrrrcsR|j}|j}|}|r|d8}z$t|jt|}|jt|}tt|t d||t||Wn t y?gYSwdtdt ffdd } g} |rZ| d|j fdgd } d } d } d }t|D]6}| |r| |dsz| |d fn| | d fn| |dr| | d fn| | d f| dqf|r| d|jf| S)Nr rowrcs|ko kSS)z/True if we should display a button on this row.r)rJscrollbar_height scrollbar_toprris_scroll_buttonsz7ScrollbarMargin.create_margin..is_scroll_buttonzclass:scrollbar.arrow)zclass:scrollbarr2zclass:scrollbar.backgroundz*class:scrollbar.background,scrollbar.startzclass:scrollbar.buttonz$class:scrollbar.button,scrollbar.end r0)content_heightr;rFr/r7floatvertical_scrollr&minr.ZeroDivisionErrorboolextendrGranger8rH)rrrrrPr;rFfraction_visiblefraction_aboverNr?scrollbar_backgroundscrollbar_background_startscrollbar_buttonscrollbar_button_endirrKrr!sR      zScrollbarMargin.create_margin)FrDrE) r"r#r$r%rstrr+rrr&rr r!rrrrrs0  rc @sxeZdZdZ ddegefdeeeeegefddfddZ dege fdefd d Z d d d ededefddZ dS)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_promptget_continuationrcCs||_||_dSr*)r`ra)rr`rarrrr+ s zPromptMargin.__init__rcCst|}t|S)z Width to report to the `Window`.)r r`r )rrtextrrrrs zPromptMargin.get_widthrrrrc Csd|j}g}|t||r0d}|jddD]}|d|t|||||k|}q|S)Nr r0)rarVr r`r7r8)rrrrrar?last_yr5rrrr!s zPromptMargin.create_marginr*) r"r#r$r%rr rr&rUr+rrr!rrrrrs,   rN)r%abcrrtypingrrrprompt_toolkit.filtersrrprompt_toolkit.formatted_textr r r prompt_toolkit.utilsr controlsr containersr__all__rrrrrrrrrs    %EW