U cc^@sddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZmZmZmZmZdd lmZmZdd lmZe rdd l m!Z!m"Z"m#Z#ed Z$Gd ddeZ%eee%ee%ee&e'ffee%e&e&ffZ(eGdddeZ)GdddZ*GdddZ+e,dkrddl-m!Z!ddl.m/Z/ddl0m1Z1dZ2e13dZ4e!Z e 5de 6de 6de 6e/e2ddd e 6e 6d!e7e 8e4Z9e 6e9e 6e 6d"e 6e4e 6d#d$S)%)IntEnum) lru_cache) filterfalse) getLogger) attrgetter) TYPE_CHECKINGDictIterableList NamedTupleOptionalSequenceTupleTypeUnion)_is_single_cell_widthscached_cell_lencell_lenget_character_cell_size set_cell_size)Result rich_repr)Style)ConsoleConsoleOptions RenderResultrichc@sPeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdS) ControlTypezDNon-printable control codes which typically translate to ANSI codes.r N)__name__ __module__ __qualname____doc__ZBELLZCARRIAGE_RETURNHOMEZCLEARZ SHOW_CURSORZ HIDE_CURSORZENABLE_ALT_SCREENZDISABLE_ALT_SCREENZ CURSOR_UPZ CURSOR_DOWNZCURSOR_FORWARDZCURSOR_BACKWARDZCURSOR_MOVE_TO_COLUMNZCURSOR_MOVE_TOZ ERASE_IN_LINEZSET_WINDOW_TITLEr3r3R/var/www/html/project/venv/lib/python3.8/site-packages/pip/_vendor/rich/segment.pyr#s"rc @s*eZdZUdZeed<dZeeed<dZ ee e ed<e e dddZedd d Zedd d Ze edd dZeedde eddddZe eddddZeddddZedEedeeeeeddddZedFedeeddddZeedeedd d!d"ZedGede eeeeeedd$d%d&ZedHede eeeedd'd(d)Z eede d*d+d,Z!eeedee e fd-d.d/Z"edIeede ee eeeeedd0d1d2Z#edJe$deede e eeeedd3d4d5Z%edKe$deede e eeeedd3d6d7Z&edLe$deede e eeeedd3d8d9Z'eededd d:d;Z(eededd dd?Z*eededd d@dAZ+eedee eeddBdCdDZ,dS)MSegmentaA piece of text with associated style. Segments are produced by the Console render process and are ultimately converted in to strings to be written to the terminal. Args: text (str): A piece of text. style (:class:`~rich.style.Style`, optional): An optional style to apply to the text. control (Tuple[ControlCode], optional): Optional sequence of control codes. Attributes: cell_length (int): The cell length of this Segment. textNstylecontrol)returncCs|\}}}|rdSt|S)zzThe number of terminal cells required to display self.text. Returns: int: A number of cells. r)r)selfr6_styler8r3r3r4 cell_lengthQs zSegment.cell_lengthccs:|jV|jdkr&|jdk r6|jVn|jV|jVdSN)r6r8r7r:r3r3r4 __rich_repr__[s    zSegment.__rich_repr__cCs t|jS)z#Check if the segment contains text.)boolr6r>r3r3r4__bool__dszSegment.__bool__cCs |jdk S)z,Check if the segment contains control codes.N)r8r>r3r3r4 is_controlhszSegment.is_controli@)r5r5)segmentcutr9c Cs(|\}}}t}|j}||kr,||d||fSt}t||t|} |d| } t| } | |kr|| ||||| d||fS| t|kr|| } | d7} | || 7} |d| } | |kr|| ||||| d||fS| |kr|| d| dd|||d|| d||fSqtddS)Nr zWill never reach here)r5r<rintlenrAssertionError) clsrCrDr6r7r8_Segmentr< cell_sizeposbeforeZcell_poscharr3r3r4 _split_cellsms6      zSegment._split_cells)rDr9cCsb|\}}}t|rV|t|kr.|td||fSt|d|||t||d||fS|||S)a0Split segment in to two segments at the specified column. If the cut point falls in the middle of a 2-cell wide character then it is replaced by two spaces, to preserve the display width of the parent segment. Returns: Tuple[Segment, Segment]: Two segments. rEN)rrHr5rP)r:rDr6r7r8r3r3r4 split_cellss  zSegment.split_cellscCs|dS)zMake a new line segment. r3)rJr3r3r4linesz Segment.line)segmentsr7 post_styler9cs>|}|r"|jfdd|D}r:fdd|D}|S)aApply style(s) to an iterable of segments. Returns an iterable of segments where the style is replaced by ``style + segment.style + post_style``. Args: segments (Iterable[Segment]): Segments to process. style (Style, optional): Base style. Defaults to None. post_style (Style, optional): Style to apply on top of segment style. Defaults to None. Returns: Iterable[Segments]: A new iterable of segments (possibly the same iterable). c3s,|]$\}}}||rdn||VqdSr=r3.0r6r;r8)applyrJr3r4 sz&Segment.apply_style..c3s4|],\}}}||rdn|r$|n|VqdSr=r3rV)rJrUr3r4rYs )__add__)rJrTr7rUZresult_segmentsr3)rXrJrUr4 apply_styles   zSegment.apply_styleF)rTrBr9cCs$|rttd|Sttd|SdS)a2Filter segments by ``is_control`` attribute. Args: segments (Iterable[Segment]): An iterable of Segment instances. is_control (bool, optional): is_control flag to match in search. Returns: Iterable[Segment]: And iterable of Segment instances. r8N)filterrr)rJrTrBr3r3r4filter_controlszSegment.filter_control)rTr9c csg}|j}|D]d}d|jkrj|jsj|\}}}|rr|d\}} }|rR||||| r,|Vg}|j}q,q||q|r~|VdS)aSplit a sequence of segments in to a list of lines. Args: segments (Iterable[Segment]): Segments potentially containing line feeds. Yields: Iterable[List[Segment]]: Iterable of segment lists, one per line. rRN)appendr6r8 partition) rJrTrSr^rCr6r7__textnew_liner3r3r4 split_liness    zSegment.split_linesT)rTlengthr7padinclude_new_linesr9ccsg}|j}|j}|d} |D]} d| jkr| js| \} } } | r| d\}}} |r`|||| |r:|||||d}|r|| |V|dd=q:q|| q|r|||||dVdS)aSplit segments in to lines, and crop lines greater than a given length. Args: segments (Iterable[Segment]): An iterable of segments, probably generated from console.render. length (int): Desired line length. style (Style, optional): Style to use for any padding. pad (bool): Enable padding of lines that are less than `length`. Returns: Iterable[List[Segment]]: An iterable of lines of segments. rR)r7reN)r^adjust_line_lengthr6r8r_)rJrTrdr7rerfrSr^rgZnew_line_segmentrCr6 segment_styler`rarbZ cropped_liner3r3r4split_and_crop_lines s2   zSegment.split_and_crop_lines)rSrdr7rer9c Cstdd|D}||krF|r8||d|||g}q|dd}n||krg}|j}d}|D]X}|j} || |ks||jr|||| 7}q`|\} } } t| ||} ||| | qq`n |dd}|S)aAdjust a line to a given width (cropping or padding as required). Args: segments (Iterable[Segment]): A list of segments in a single line. length (int): The desired width of the line. style (Style, optional): The style of padding if used (space on the end). Defaults to None. pad (bool, optional): Pad lines with spaces if they are shorter than `length`. Defaults to True. Returns: List[Segment]: A line of segments with the desired length. css|] }|jVqdSr=)r<rWrCr3r3r4rYKsz-Segment.adjust_line_length..rFNr)sumr^r<r8r) rJrSrdr7reZ line_lengthrbr^rCZsegment_lengthr6rhr`r3r3r4rg8s(   zSegment.adjust_line_length)rSr9csttfdd|DS)zGet the length of list of segments. Args: line (List[Segment]): A line encoded as a list of Segments (assumes no '\\n' characters), Returns: int: The length of the line. c3s|]}|jVqdSr=)r6rj _cell_lenr3r4rYpsz*Segment.get_line_length..)rrk)rJrSr3rlr4get_line_lengthes zSegment.get_line_length)linesr9cs0|j|r tfdd|Dnd}|t|fS)zGet the shape (enclosing rectangle) of a list of lines. Args: lines (List[List[Segment]]): A list of lines (no '\\n' characters). Returns: Tuple[int, int]: Width and height in characters. c3s|]}|VqdSr=r3rWrSrnr3r4rY}sz$Segment.get_shape..r)rnmaxrH)rJro max_widthr3rqr4 get_shapers zSegment.get_shape)rowidthheightr7 new_linesr9c s|p t|}|r$|ddgn|dg}|j|d|}fdd|D|dd<t||kr||g|t||S)aSet the shape of a list of lines (enclosing rectangle). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style, optional): Style of any padding added. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. rFrRNcsg|]}|dqS))r7r3rprgr7rur3r4 sz%Segment.set_shape..)rHrgextend) rJrorurvr7rw_heightblankZ shaped_linesr3rxr4 set_shapes &  zSegment.set_shape)rJrorurvr7rwr9cCs`|t|}|s|ddS|d|}|r>|d|d|n |d||}||gg|}|S)aAligns lines to top (adds extra lines to bottom as required). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style): Style of any padding added. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. NrFrRrHrJrorurvr7rw extra_linesr|r3r3r4 align_tops   $zSegment.align_topcCs`|t|}|s|ddS|d|}|r>|d|d|n |d||}|gg||}|S)aAligns render to bottom (adds extra lines above as required). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style): Style of any padding added. Defaults to None. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. NrFrRr~rr3r3r4 align_bottoms   $zSegment.align_bottomc Cs||t|}|s|ddS|d|}|r>|d|d|n |d||}|d}||} |gg|||gg| }|S)aAligns lines to middle (adds extra lines to above and below as required). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style): Style of any padding added. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. NrFrRrr~) rJrorurvr7rwrr|Z top_linesZ bottom_linesr3r3r4 align_middles   $zSegment.align_middleccsvt|}z t|}Wntk r*YdSXt}|D]6}|j|jkr`|js`||j|j|j}q4|V|}q4|VdS)a)Simplify an iterable of segments by combining contiguous segments with the same style. Args: segments (Iterable[Segment]): An iterable of segments. Returns: Iterable[Segment]: A possibly smaller iterable of segments that will render the same way. N)iternext StopIterationr5r7r8r6)rJrT iter_segmentsZ last_segmentrKrCr3r3r4simplifys   zSegment.simplifyccsJ|D]@}|js|jdkr |Vq|\}}}|||r<|dndVqdS)zRemove all links from an iterable of styles. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with link removed. N)r8r7 update_link)rJrTrCr6r7_controlr3r3r4 strip_linkss  zSegment.strip_linksccs"|D]\}}}||d|VqdS)zRemove all styles from an iterable of segments. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with styles replace with None Nr3)rJrTr6r;r8r3r3r4 strip_styles)s zSegment.strip_stylesccsZi}|D]L\}}}|rF||}|dkr6|j}|||<||||Vq||d|VqdS)zRemove all color from an iterable of segments. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with colorless style. N)get without_color)rJrTcacher6r7r8Zcolorless_styler3r3r4 remove_color6s  zSegment.remove_color)rTcutsr9ccsDg}|j}t|}t|d}|dkr(gS|dkr2q:gVqd}|j}|j} t} |D]} | \} } }| rR|rl|n || | }||kr|| |}qR||kr|| | V||}t|d}|dkr|r| VdSqRn6| ||\}} | \} } }||| V||}t|d}|dkr`|r.| VdSq`qR| VdS)zDivides an iterable of segments in to portions. Args: cuts (Iterable[int]): Cell positions where to divide. Yields: [Iterable[List[Segment]]]: An iterable of Segments in List. rN)r^rrclearcopyrrQ)rJrTrZsplit_segmentsZ add_segmentZ iter_cutsrDrMZsegments_clearZ segments_copyrmrCr6r;r8Zend_posrNr3r3r4divideLsX       zSegment.divide)NN)F)NTT)NT)NNF)F)F)F)-r.r/r0r1str__annotations__r7r rr8r ControlCodepropertyrGr<rr?r@rArB classmethodrrrPrQrSr r[r]r rcrirgrnrtr}rrrrrrrrrr3r3r3r4r5?s    &( ., "   #        r5c@s:eZdZdZd eeeddddZddd d d d ZdS)Segmentsa=A simple renderable to render an iterable of segments. This class may be useful if you want to print segments outside of a __rich_console__ method. Args: segments (Iterable[Segment]): An iterable of segments. new_lines (bool, optional): Add new lines between segments. Defaults to False. FN)rTrwr9cCst||_||_dSr=)listrTrw)r:rTrwr3r3r4__init__s zSegments.__init__rrrconsoleoptionsr9ccs8|jr(t}|jD]}|V|Vqn |jEdHdSr=)rwr5rSrT)r:rrrSrCr3r3r4__rich_console__s   zSegments.__rich_console__)F) r.r/r0r1r r5r@rrr3r3r3r4rs rc@s:eZdZd eeeeddddZdddd d d ZdS) SegmentLinesFN)rorwr9cCst||_||_dS)a=A simple renderable containing a number of lines of segments. May be used as an intermediate in rendering process. Args: lines (Iterable[List[Segment]]): Lists of segments forming lines. new_lines (bool, optional): Insert new lines after each line. Defaults to False. N)rrorw)r:rorwr3r3r4rs zSegmentLines.__init__rrrrccsF|jr,t}|jD]}|EdH|Vqn|jD]}|EdHq2dSr=)rwr5rSro)r:rrrbrSr3r3r4rs    zSegmentLines.__rich_console__)F) r.r/r0r r r5r@rrr3r3r3r4rs  r__main__)r)Syntax)Textzfrom rich.console import Console console = Console() text = Text.from_markup("Hello, [bold magenta]World[/]!") console.print(text)zHello, [bold magenta]World[/]!z rich.Segmentz]A Segment is the last step in the Rich render process before generating text with ANSI codes.z Consider the following code: pythonT)Z line_numberszVWhen you call [b]print()[/b], Rich [i]renders[/i] the object in to the the following: zAThe Segments are then processed to produce the following output: zS You will only need to know this if you are implementing your own Rich renderables.N):enumr functoolsr itertoolsrloggingroperatorrtypingrrr r r r r rrrcellsrrrrrreprrrr7rrrrrlogrrGrrr5rrr.pip._vendor.rich.consoleZpip._vendor.rich.syntaxrpip._vendor.rich.textrcode from_markupr6ruleprintrrender fragmentsr3r3r3r4sf     0   V