"dZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZd d lmZd ZGd d eeeZGddeZGddeZGddeZGddeZy)zQ Margin implementations for a :class:`~prompt_toolkit.layout.containers.Window`. )unicode_literals)ABCMetaabstractmethod)with_metaclass)range) to_cli_filter)Token) get_cwidth)token_list_to_text)MarginNumberredMarginScrollbarMarginConditionalMargin PromptMarginc0eZdZdZedZedZy)r z& Base interface for a margin. cy)aH Return the width that this margin is going to consume. :param cli: :class:`.CommandLineInterface` instance. :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. r)selfcliget_ui_contents 0lib/third_party/prompt_toolkit/layout/margins.py get_widthzMargin.get_widthscgS)a Creates a margin. This should return a list of (Token, text) tuples. :param cli: :class:`.CommandLineInterface` instance. :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`.) rrrwindow_render_infowidthheights r create_marginzMargin.create_margin(s " rN)__name__ __module__ __qualname____doc__rrr rrrr r s/  rr c$eZdZdZddZdZdZy)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. cDt||_t||_yN)rrelativedisplay_tildes)rr(r)s r__init__zNumberredMargin.__init__Es%h/ +N;rcZ|j}tdtd|zdzS)Nz%sr ) line_countmaxlen)rrrr-s rrzNumberredMargin.get_widthIs,#%00 1c$+,q011rc|j|}tj}tjj}|jj j }g} d} t|jD]\} } | | k7r| n| |k(rE|r| j|d| dzzfnd| j|d| dzzj|fn;|rt| |z dz } | j|d| dzzj|f| } | jtdf|j|rE |jkr6| jtjdf| dz } | |jkr6| S)Nz%ir %i  z~ )r(r LineNumberCurrent ui_contentcursor_positiony enumeratedisplayed_linesappendrjustabsr) window_heightTilde) rrrrrr(token token_currentcurrent_linenoresult last_linenor7linenos rr zNumberredMargin.create_marginMso==%  ((00 ,66FFHH "#5#E#EFIAv$>~- }dfqj6I&JK }u 7K6R6RSX6Y&Z[ !$Vn%.is_scroll_buttons)%(#(m)C%D")-?-O-OOOr^r2 v) content_heightrPr=floatminextendr ScrollbarArrowrr:ButtonZeroDivisionError) rrrrr total_heightrPr=rYrBirXs ` @rr zScrollbarMargin.create_margins$)88 ,,S1*88  Q M !,/#lM2RRM P F __**C0__d+ =)#A&MM5??#9#93"?@MM5??C"89 udm, * u44c:;M9! I sD>> E  E N)FrErrrrrs <&rrc(eZdZdZ ddZdZdZy)ra 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. :param get_prompt_tokens: Callable that takes a CommandLineInterface as input and returns a list of (Token, type) tuples to be shown as the prompt at the first line. :param get_continuation_tokens: Callable that takes a CommandLineInterface and a width as input and returns a list of (Token, type) tuples for the next lines of the input. :param show_numbers: (bool or :class:`~prompt_toolkit.filters.CLIFilter`) Display line numbers instead of the continuation prompt. Nc|t|sJ| t|sJt|}||_||_||_yr')callablerget_prompt_tokensget_continuation_tokens show_numbers)rrjrkrls rr*zPromptMargin.__init__sG)***&.(;R2SSS$\2 !2'>$(rcLt|j|}t|S)z" Width to report to the `Window`. )r rjr )rrr5texts rrzPromptMargin.get_widths$"$"8"8"=>$rc|j|dd}|jrt|j||}ng}|j|}d}|jddD]j} |j t df|r<| |k7rH|j t jd| dzzj|fn|j|| }l|S)Nr r2r1) rjrklistrlr9r:r r3r;r`) rrrrrtokenstokens2rllast_yr7s rr zPromptMargin.create_margins'',Q/  ' '477UCDGG((- #33AB7A MM5$- (;MM5#3#3eq1uo5L5LU5S"TU g&F8 r)NFrErrrrrs  CG#) rrN)r$ __future__rabcrrsixr six.movesrprompt_toolkit.filtersrprompt_toolkit.tokenr prompt_toolkit.utilsr utilsr __all__objectr rrrrrrrr~sq('0&+% !^GV ,!H:f:z.2f2j868r