a Ebl]@sddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZmZmZmZdd lmZmZdd lmZe rdd lm Z m!Z!m"Z"ed Z#Gd ddeZ$eee$ee$e%fee$e%e%ffZ&eGdddeZ'GdddZ(GdddZ)e*dkre*dkrddl+m Z ddl,m-Z-ddl.m/Z/dZ0e/1dZ2e Ze3de4de4de4e-e0ddd e4e4d!e5e6e2Z7e4e7e4e4d"e4e2e4d#d$S)%)IntEnum) lru_cache) filterfalse) getLogger) attrgetter) TYPE_CHECKINGDictIterableList NamedTupleOptionalSequenceTupleTypeUnion)_is_single_cell_widthscell_lenget_character_cell_size set_cell_size)Result rich_repr)Style)ConsoleConsoleOptions RenderResultrichc@sLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS) 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_LINEr1r1X/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pip/_vendor/rich/segment.pyr"s rc @s.eZdZUdZdZeed<dZee ed<dZ ee e ed<e ddd Zedd d Zeedd d ZeedddZeeddeeddddZeeddddZeddddZedFedee ee eddddZedGedeedddd Zeedeedd!d"d#ZedHedeee eeeedd%d&d'Z edIedeee eedd(d)d*Z!eeded+d,d-Z"eeedeeefd.d/d0Z#edJeedeeeee eeedd1d2d3Z$edKe%deedeee eeedd4d5d6Z&edLe%deedeee eeedd4d7d8Z'edMe%deedeee eeedd4d9d:Z(eededd!d;d<Z)eededd!d=d>Z*eededd!d?d@Z+eededd!dAdBZ,eedeeeeddCdDdEZ-dS)NSegmentaA 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. textNstylecontrol)returnccs:|jV|jdur&|jdur6|jVn|jV|jVdSN)r5r7r6selfr1r1r2 __rich_repr__Ms    zSegment.__rich_repr__cCs t|jS)z#Check if the segment contains text.)boolr5r:r1r1r2__bool__VszSegment.__bool__cCs|jr dSt|jS)zGet cell length of segment.r)r7rr5r:r1r1r2 cell_lengthZszSegment.cell_lengthcCs |jduS)z,Check if the segment contains control codes.N)r7r:r1r1r2 is_control_szSegment.is_controli@)r3r3)segmentcutr8c 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||fSqdS)Nr4r )r3r?rintlenr) clsrArBr5r6r7_Segmentr?Z cell_sizeposbeforeZcell_poscharr1r1r2 _split_cellsds4      zSegment._split_cells)rBr8cCsb|\}}}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. r4N)rrEr3rK)r;rBr5r6r7r1r1r2 split_cellss  zSegment.split_cellscCs|dS)zMake a new line segment. r1)rFr1r1r2linesz Segment.line)segmentsr6 post_styler8cs>|}|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||VqdSr9r1.0r5_styler7)applyrFr1r2 sz&Segment.apply_style..c3s4|],\}}}||rdn|r$|n|VqdSr9r1rQ)rFrPr1r2rUs )__add__)rFrOr6rPZresult_segmentsr1)rTrFrPr2 apply_styles   zSegment.apply_styleF)rOr@r8cCs$|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. r7N)filterrr)rFrOr@r1r1r2filter_controlszSegment.filter_control)rOr8c csg}|j}|D]d}d|jvrj|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. rMN)appendr5r7 partition) rFrOrNrZrAr5r6__textnew_liner1r1r2 split_liness    zSegment.split_linesT)rOlengthr6padinclude_new_linesr8ccsg}|j}|j}|d} |D]} d| jvr| 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. rM)r6raN)rZadjust_line_lengthr5r7r[)rFrOr`r6rarbrNrZrcZnew_line_segmentrAr5r\r]r^Z cropped_liner1r1r2split_and_crop_liness,   zSegment.split_and_crop_lines)rNr`r6rar8c 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|] }|jVqdSr9)r?rRrAr1r1r2rU@z-Segment.adjust_line_length..rCNr)sumrZr?r7r) rFrNr`r6raZ line_lengthr^rZrAZsegment_lengthr5Z segment_styler\r1r1r2rc-s(   zSegment.adjust_line_length)rNr8csttfdd|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|]}|jVqdSr9)r5reZ _cell_lenr1r2rUerfz*Segment.get_line_length..)rrg)rFrNr1rhr2get_line_lengthZs zSegment.get_line_length)linesr8cs0|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|]}|VqdSr9r1rRrNrir1r2rUrrfz$Segment.get_shape..r)rimaxrE)rFrj max_widthr1rlr2 get_shapegs zSegment.get_shape)rjwidthheightr6 new_linesr8c 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. rCrMNcsg|]}|dqS))r6r1rkrcr6rpr1r2 sz%Segment.set_shape..)rErcextend) rFrjrprqr6rr_heightblankZ shaped_linesr1rsr2 set_shapeus &  zSegment.set_shape)rFrjrprqr6rrr8cCs`|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. NrCrMrErFrjrprqr6rr extra_linesrwr1r1r2 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. NrCrMryrzr1r1r2 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. NrCrMrry) rFrjrprqr6rrr{rwZ top_linesZ bottom_linesr1r1r2 align_middles   $zSegment.align_middleccstt|}z t|}Wnty(YdS0t}|D]6}|j|jkr^|js^||j|j|j}q2|V|}q2|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 StopIterationr3r6r7r5)rFrOZ iter_segmentsZ last_segmentrGrAr1r1r2simplifys   zSegment.simplifyccsJ|D]@}|js|jdur |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)r7r6 update_link)rFrOrAr5r6_controlr1r1r2 strip_links s  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 Nr1)rFrOr5rSr7r1r1r2 strip_styless zSegment.strip_stylesccsZi}|D]L\}}}|rF||}|dur6|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)rFrOcacher5r6r7Zcolorless_styler1r1r2 remove_color+s  zSegment.remove_color)rOcutsr8c csg}|j}t|}z t|}Wnty4gYS0|dkr@qHgVqd}|D]N}|jrP||j} | |kr|||| }qPz| |kr|||ddV|dd=| }Wz t|}Wn(ty|r|ddVYdS0qPn6|||\} }|| |ddV|dd=|}Wz t|}Wn,tyV|rN|ddVYdS0qVz t|}Wn4ty|r|ddVYYdS00qVqP|ddVdS)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)rZrrrr5r?rL) rFrOrZsplit_segmentsZ add_segmentZ iter_cutsrBrHrAZend_posrIr1r1r2divideAsd              zSegment.divide)NN)F)NTT)NT)NNF)F)F)F).r,r-r.r/r5str__annotations__r6r rr7r ControlCoderr<r=r>propertyrDr?r@ classmethodrrrKrLrNr rWrYr r_rdrcrirorxrr|r}r~rrrrrr1r1r1r2r3;s   $( ., "   #         r3c@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)rOrrr8cCst||_||_dSr9)listrOrr)r;rOrrr1r1r2__init__s zSegments.__init__rrrconsoleoptionsr8ccs8|jr(t}|jD]}|V|Vqn |jEdHdSr9)rrr3rNrO)r;rrrNrAr1r1r2__rich_console__s   zSegments.__rich_console__)F) r,r-r.r/r r3r=rrr1r1r1r2r{s rc@s:eZdZd eeeeddddZdddd d d ZdS) SegmentLinesFN)rjrrr8cCst||_||_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)rrjrr)r;rjrrr1r1r2rs zSegmentLines.__init__rrrrccsF|jr,t}|jD]}|EdH|Vqn|jD]}|EdHq2dSr9)rrr3rNrj)r;rrr^rNr1r1r2rs    zSegmentLines.__rich_console__)F) r,r-r.r r r3r=rrr1r1r1r2rs 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)8enumr functoolsr itertoolsrloggingroperatorrtypingrrr r r r r rrrcellsrrrrreprrrr6rrrrrlogrrDrr3rrr,pip._vendor.rich.consoleZpip._vendor.rich.syntaxrpip._vendor.rich.textrcode from_markupr5ruleprintrrender fragmentsr1r1r1r2sf     0  C