o h`@sddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZmZmZmZmZdd lmZmZdd lmZe r^dd 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$Sd$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 RenderResultZrichc@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_RETURNHOMEZCLEAR SHOW_CURSOR HIDE_CURSORZENABLE_ALT_SCREENZDISABLE_ALT_SCREENZ CURSOR_UPZ CURSOR_DOWNZCURSOR_FORWARDZCURSOR_BACKWARDZCURSOR_MOVE_TO_COLUMNZCURSOR_MOVE_TOZ ERASE_IN_LINEZSET_WINDOW_TITLEr4r4^/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/pip/_vendor/rich/segment.pyr#s$rc@seZdZUdZeed<dZeeed<dZ ee e ed<e de fddZdefd d Zdefd d Ze defd dZeedddde dedfddZde dedfddZedHddZe  dIdeddeedeededfddZe dJdeddededfdd Zededdeedfd!d"Ze  # #dKdedd$e deed%ed&edeedf d'd(Ze  #dLd)edd$e deed%ededf d*d+Z ed)edde fd,d-Z!ed.eeddee e ffd/d0Z"e   dMd.eedd1e d2ee deed3edeedf d4d5Z#e dJd6e$dd.eedd1e d2e ded3edeedfd7d8Z%e dJd6e$dd.eedd1e d2e ded3edeedfd9d:Z&e dJd6e$dd.eedd1e d2e ded3edeedfd;d<Z'ededdedfd=d>Z(ededdedfd?d@Z)ededdedfdAdBZ*ededdedfdCdDZ+ededdEee deedfdFdGZ,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. Attributes: cell_length (int): The cell length of this Segment. textNstylecontrolreturncCs|\}}}|r dSt|S)zzThe number of terminal cells required to display self.text. Returns: int: A number of cells. r)r)selfr7_styler9r4r4r5 cell_lengthQs zSegment.cell_lengthccsB|jV|jdur|jdur|jVdSdS|jV|jVdSN)r7r9r8r;r4r4r5 __rich_repr__[s    zSegment.__rich_repr__cCs t|jS)z#Check if the segment contains text.)boolr7r?r4r4r5__bool__ds zSegment.__bool__cCs |jduS)z,Check if the segment contains control codes.N)r9r?r4r4r5 is_controlhs zSegment.is_controli@segmentcut)r6r6c Cs>|\}}}t}|j}||kr||d||fSt}t||t|} |d| } t| } | |} | sC|| ||||| d||fS| dkri||| dkri||d| d|||d|| dd||fS| dkr||| ddkr||d| dd|||d|| d||fS| |kr| d7} n| d8} q#)a\Split a segment in to two at a given cell position. Note that splitting a double-width character, may result in that character turning into two spaces. Args: segment (Segment): A segment to split. cut (int): A cell position to cut on. Returns: A tuple of two segments. TNr r)r6r=rintlenr) clsrDrEr7r8r9_Segmentr=Z cell_sizeposbeforeZcell_posZout_byr4r4r5 _split_cellsms6    zSegment._split_cellscCsn|\}}}|dks Jt|r1|t|kr|td||fSt|d|||t||d||fS|||S)axSplit 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. Args: cut (int): Offset within the segment to cut. Returns: Tuple[Segment, Segment]: Two segments. rrFN)rrJr6rO)r;rEr7r8r9r4r4r5 split_cellss   zSegment.split_cellscCs|dS)zMake a new line segment. r4)rKr4r4r5linesz Segment.linesegments post_stylecs>|}|r|jfdd|D}rfdd|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.|]\}}}||r dn||VqdSr>r4.0r7r<r9)applyrKr4r5 s  z&Segment.apply_style..c3s6|]\}}}||r dn|r|n|VqdSr>r4rU)rKrTr4r5rXs  )__add__)rKrSr8rTZresult_segmentsr4)rWrKrTr5 apply_styles    zSegment.apply_styleFrCcCs |r ttd|Sttd|S)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. r9)filterrr)rKrSrCr4r4r5filter_controlszSegment.filter_controlc csg}|j}|D]3}d|jvr7|js7|\}}}|r6|d\}} }|r*||||| r4|Vg}|j}|sq||q|rC|VdSdS)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. rQN)appendr7r9 partition) rKrSrRr]rDr7r8__textnew_liner4r4r5 split_liness(    zSegment.split_linesTlengthpadinclude_new_linesccsg}|j}|j}|d} |D]A} d| jvrL| jsL| \} } } | rK| d\}}} |r1|||| |rI|||||d}|rB|| |V|| s q|| q|r^|||||dVdSdS)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. rQ)r8rdN)r]adjust_line_lengthr7r9r^clear)rKrSrcr8rdrerRr]rfZnew_line_segmentrDr7 segment_styler_r`raZ cropped_liner4r4r5split_and_crop_liness4   zSegment.split_and_crop_linesrRc Cstdd|D}||kr%|r||d|||g}|S|dd}|S||krag}|j}d}|D],}|j} || |ks@|jrI|||| 7}q2|\} } } t| ||} ||| | |S|S|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=)rVrDr4r4r5rXXsz-Segment.adjust_line_length..rHNr)sumr]r=r9r) rKrRrcr8rdZ line_lengthrar]rDZsegment_lengthr7rhr_r4r4r5rfEs.    zSegment.adjust_line_lengthcsttfdd|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"|] \}}}|s|VqdSr>r4)rVr7r8r9 _cell_lenr4r5rX}s z*Segment.get_line_length..)rrj)rKrRr4rkr5get_line_lengthrs zSegment.get_line_lengthlinescs0|j|rtfdd|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>r4rVrRrmr4r5rXsz$Segment.get_shape..r)rmmaxrJ)rKrn max_widthr4rpr5 get_shapes  zSegment.get_shapewidthheight new_linesc s|pt|}|r|ddgn|dg}|j|d|}fdd|D|dd<t||krD||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. rHrQNcsg|] }|dqS))r8r4rorfr8rtr4r5 sz%Segment.set_shape..)rJrfextend) rKrnrtrur8rvZ_heightblankZ shaped_linesr4rwr5 set_shapes &  zSegment.set_shaperKcCs`|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. NrHrQrJrKrnrtrur8rv extra_linesrzr4r4r5 align_top   $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. NrHrQr|r}r4r4r5 align_bottomrzSegment.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. NrHrQrr|) rKrnrtrur8rvr~rzZ top_linesZ bottom_linesr4r4r5 align_middles   $zSegment.align_middleccsvt|}zt|}Wn tyYdSwt}|D]}|j|jkr0|js0||j|j|j}q|V|}q|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 StopIterationr6r8r9r7)rKrSZ iter_segmentsZ last_segmentrLrDr4r4r5simplify s     zSegment.simplifyccsL|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)r9r8Z update_link)rKrSrDr7r8Z_controlr4r4r5 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 Nr4)rKrSr7r<r9r4r4r5 strip_styles6s zSegment.strip_stylesccs\i}|D]&\}}}|r$||}|dur|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)getZ without_color)rKrScacher7r8r9Zcolorless_styler4r4r5 remove_colorCs  zSegment.remove_colorcutsccsFg}|j}t|} t|d}|dkrdS|dkrqgVq d}|j}|j} t} |D]q} | \} } }| r|r8|n|| | }||krI|| |}q||krm|| | V||}t|d}|dkrl|ri| VdSq| ||\}} | \} } }||| V||}t|d}|dkr|r| VdS| s4q+| 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. TrGNr)r]rrrgcopyrrP)rKrSrZsplit_segmentsZ add_segmentZ iter_cutsrErMZsegments_clearZ segments_copyrlrDr7r<r9end_posrNr4r4r5divideYsb       #zSegment.divide)r:r6)NNF)NTT)NT)NNF)-r-r.r/r0str__annotations__r8r rr9r ControlCodepropertyrIr=rr@rArBrC classmethodrrrOrPrRr rZr\r rbrirfrmrsr{rrrrrrrrrr4r4r4r5r6?s>    /  (    . , $    #           r6c@sBeZdZdZddeededdfddZ  dddZdS)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. FrSrvr:NcCst||_||_dSr>)listrSrv)r;rSrvr4r4r5__init__s  zSegments.__init__consoleroptionsrrccs<|jrt}|jD]}|V|Vq dS|jEdHdSr>)rvr6rRrS)r;rrrRrDr4r4r5__rich_console__s zSegments.__rich_console__rrrrrr:r) r-r.r/r0r r6rArrr4r4r4r5rsrc@sBeZdZddeeededdfddZ   dd dZdS) SegmentLinesFrnrvr:NcCst||_||_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)rrnrv)r;rnrvr4r4r5rs  zSegmentLines.__init__rrrrrccsJ|jrt}|jD] }|EdH|Vq dS|jD]}|EdHqdSr>)rvr6rRrn)r;rrrarRr4r4r5rs    zSegmentLines.__rich_console__rr) r-r.r/r r r6rArrr4r4r4r5rs  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_numberszRWhen you call [b]print()[/b], Rich [i]renders[/i] the object in to 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 rrrcellsrrrrrreprrrr8rrrrrlogrrIrrr6rrr-Zpip._vendor.rich.consoleZpip._vendor.rich.syntaxrZpip._vendor.rich.textrcodeZ from_markupr7Zruleprintrrender fragmentsr4r4r4r5sj    0   c