KidlddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZmZmZmZddlmZmZdd lmZe r dd lmZm Z m!Z!Gd d eZ"eee"ee"ee#e$ffee"e#e#ffZ%eGd de Z&GddZ'GddZ(e)dk(rddl*mZddl+m,Z,ddl-m.Z.dZ/e.j`dZ1eZejedejgdejgdejge,e/ddejgejgde4ejke1Z6ejge6ejgejgd ejge1ejgd!y"y")#)IntEnum) lru_cache) filterfalse) attrgetter) TYPE_CHECKINGDictIterableList NamedTupleOptionalSequenceTupleTypeUnion)_is_single_cell_widthscached_cell_lencell_lenget_character_cell_size set_cell_size)Result rich_repr)Style)ConsoleConsoleOptions RenderResultcPeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZy) ControlTypezDNon-printable control codes which typically translate to ANSI codes.r N)__name__ __module__ __qualname____doc__BELLCARRIAGE_RETURNHOMECLEAR SHOW_CURSOR HIDE_CURSORENABLE_ALT_SCREENDISABLE_ALT_SCREEN CURSOR_UP CURSOR_DOWNCURSOR_FORWARDCURSOR_BACKWARDCURSOR_MOVE_TO_COLUMNCURSOR_MOVE_TO ERASE_IN_LINESET_WINDOW_TITLEX/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/rich/segment.pyrr sYN DO D EKKIKNONMrCrceZdZUdZeed<dZeeed<dZ ee e ed<e de fdZdefdZdefd Ze defd Zeed d dd e dedfdZd e dedfdZed0dZe d1deddeedeededfdZe d2deddededfdZededdeedfdZededdeeedeffdZe d3dedde deedededeedf dZ e d4dedde deedededf dZ!ededde fdZ"ed eeddee e ffd!Z#e d5d eedd"e d#ee deed$edeedf d%Z$e d2d&e%dd eedd"e d#e ded$edeedfd'Z&e d2d&e%dd eedd"e d#e ded$edeedfd(Z'e d2d&e%dd eedd"e d#e ded$edeedfd)Z(ededdedfd*Z)ededdedfd+Z*ededdedfd,Z+ededdedfd-Z,ededd.ee deedfd/Z-y)6SegmentaA 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. textNstylecontrolreturnc,|\}}}|rdSt|S)zzThe number of terminal cells required to display self.text. Returns: int: A number of cells. r)r)selfrG_stylerIs rD cell_lengthzSegment.cell_lengthNs!!%fgq/$/rCc#K|j|j|j|jyy|j|jywN)rGrIrHrLs rD __rich_repr__zSegment.__rich_repr__XsHii << zz%jj &** ,, sAAc,t|jS)z#Check if the segment contains text.)boolrGrQs rD__bool__zSegment.__bool__asDIIrCc|jduS)z,Check if the segment contains control codes.N)rIrQs rD is_controlzSegment.is_controles||4''rCi@segmentcut)rFrFc|\}}}t}|j}||k\r ||d||fSt}t||z t |z} |d| } t | } | |z } | s|| ||||| d||fS| dk(r1||| dk(r#||d| dz|||d|| dzdz||fS| dk(r4||| dz dk(r#||d| dz dz|||d|| dz||fS| |kr| dz } n| dz} )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. Nr r)rFrNrintlenr) clsrXrYrGrHrI_SegmentrN cell_sizeposbeforecell_posout_bys rD _split_cellszSegment._split_cellsjss 'eW)) + HR88 8+ 3$D 12$3ZF'H^FVUG4T#$Z8| $s) 4 9T$3Z#-ug>S4a ?2E7C| $sQw- 8A =T)C!G_s2E7CS4:-ug>#~qq-rCc|\}}}|dk\sJt|r=|t|k\r|td||fSt|d|||t||d||fS|j||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. rr[N)rr_rFrg)rLrYrGrHrIs rD split_cellszSegment.split_cellss $eWaxx !$ 'c$iWR888Tc E73ST E73    s++rCc|dS)zMake a new line segment. rB)r`s rDlinez Segment.lines4yrCsegments post_stylec`|}|r|jfd|D}r fd|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). c3NK|]\}}}||rdn||ywrPrB).0rGrMrIapplyr`s rD z&Segment.apply_style..s0)D&'D'$uV}gFs"%c3PK|]\}}}||rdn|r|zn|ywrPrB)rqrGrMrIr`rns rDrsz&Segment.apply_style..sB *D&'#5;fz1 s#&)__add__)r`rmrHrnresult_segmentsrrs` ` @rD apply_stylezSegment.apply_stylesC&# MME-<O  .= OrCrWcZ|rttd|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. rI)filterrr)r`rmrWs rDfilter_controlzSegment.filter_controls+ *Y/: :z)4h? ?rCc#Kg}|j}|D]k}d|jvrS|jsG|\}}}|s&|jd\}} }|r||||| r|g}|j}|r=d||m|r|yyw)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. rkNappendrGrI partition r`rmrlr}rXrGrH__textnew_lines rD split_lineszSegment.split_liness! Gw||#GOO!(eQ,0NN4,@)E8Ts5%01" !!%w  J s6B B 8B lengthpadinclude_new_linesc#Kg}|j}|j}|d} |D]} d| jvrt| jsh| \} } } | s&| j d\}}} |r|||| |r3|||||}|r|j| ||j | r^|| |r|||||yyw)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. rk)rHrN)r}adjust_line_lengthrGrIr~clear)r`rmrrHrrrlr}rnew_line_segmentrXrG segment_stylerrr cropped_lines rDsplit_and_crop_lineszSegment.split_and_crop_lines2s*! 33t9 Gw||#GOO)0&mQ,0NN4,@)E8Ts5-89'9 &3( -(//0@A** w! " $T6CH H sA C AC+Crlc^td|D}||kr|r||d||z z|gz}|S|dd}|S||kDrmg}|j}d}|D]V}|j} || z|ks |jr|||| z }1|\} } } t | ||z } ||| | |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. c34K|]}|jywrP)rN)rqrXs rDrsz-Segment.adjust_line_length..tsB''--Bsr]Nr)sumr}rNrIr) r`rlrrHr line_lengthrr}rXsegment_lengthrGrrs rDrzSegment.adjust_line_lengthas&BTBB   3sf{.B'CU#K"LL&# 7"!6 !H__FK !(!4!4/&8GOO7O>1K-4*D-(v /CDD3t]34 AwHrCc8ttfd|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. c3<K|]\}}}|r |ywrPrB)rqrGrHrI _cell_lens rDrsz*Segment.get_line_length..sS';tUG79T?Ss  )rr)r`rlrs @rDget_line_lengthzSegment.get_line_lengths StSSSrClinescf|j|rtfd|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. c3.K|] }|ywrPrB)rqrlrs rDrsz$Segment.get_shape..s@$-@sr)rmaxr_)r`r max_widthrs @rD get_shapezSegment.get_shapes3--DIC@%@@q 3u:&&rCwidthheight new_linesc$|xs t|}|r|d|zdz|gn |d|z|g}|j}|d|} |D cgc] } || ||c} | ddt| |kr!| j|g|t| z z| Scc} w)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. r]rkN)rH)r_rextend) r`rrrrHr_heightblankr shaped_linesrls rD set_shapezSegment.set_shapes*&CJ1:Sut#U + ,C%KQV@W?X !33Xg EJ =A tU% 8 Q | w &   7S5F+F G H  sB r`c|t|z }|s|ddS|d|}|r|d|zdz|n |d|z|}||gg|zz}|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. Nr]rkr_r`rrrrHr extra_linesrs rD align_topzSegment.align_topsf*s5z) 8Ogv2;C%K$&.S5[RWAX% K// rCc|t|z }|s|ddS|d|}|r|d|zdz|n |d|z|}|gg|z|z}|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. Nr]rkrrs rD align_bottomzSegment.align_bottomsf*s5z) 8Ogv2;C%K$&.S5[RWAX K'%/ rCc|t|z }|s|ddS|d|}|r|d|zdz|n |d|z|}|dz}||z } |gg|z|z|gg| zz}|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. Nr]rkrr) r`rrrrHrrr top_lines bottom_liness rD align_middlezSegment.align_middles*s5z) 8Ogv2;C%K$&.S5[RWAX1$ "Y.  I%-% L0HH rCc#&Kt|} t|}t}|D]X}|j|jk(r7|j s+||j |j z|j}S||}Z|y#t$rYywxYww)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 StopIterationrFrHrIrG)r`rm iter_segments last_segmentrarXs rDsimplifyzSegment.simplify&sX   .L$ 'G!!W]]27??' %% 4l6H6H  #"&  '   s( B BA(B B B BBc#K|D]C}|js |j| |\}}}|||r|jdndEyw)zRemove all links from an iterable of styles. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with link removed. N)rIrH update_link)r`rmrXrGrH_controls rD strip_linkszSegment.strip_linksAsV  LG'--"7 (/%eX$5 1 1$ 7dKK  LsA A c#:K|D]\}}}||d|yw)zRemove all styles from an iterable of segments. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with styles replace with None NrB)r`rmrGrMrIs rD strip_styleszSegment.strip_stylesRs-&. + !D&'dD'* * +sc#Ki}|D]E\}}}|r1|j|}||j}|||<||||:||d|Gyw)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)r`rmcacherGrHrIcolorless_styles rD remove_colorzSegment.remove_color_sl%'$, / D%"'))E"2"*&+&9&9O#2E%L$99$g.. /sAAcutsc#VKg}|j}t|} t|d}|dk(ry|dk7rngd}|j}|j} t } |D]} | \} } }| s |r|n || | z}||kr || |}+||k(r9|| | ||}t|d}|dk(r |r | yi| j ||z \}} | \} } }||| ||}t|d}|dk(r |r | y| r| yw)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. r\Nr)r}rrrcopyrri)r`rmrsplit_segments add_segment iter_cutsrYrcsegments_clear segments_copyrrXrGrMrIend_posrds rDdividezSegment.divideusp+-$++ J y"%CbyaxH '--&++ # # G$+ !D&'!(#cIdO.CS=(!Cc>('/)"$!Cy"-Cby)"//1'.&9&9#)&DOFG,3)D&'''/)"$C9b)"9%+o-C# JosA%D)(B4D) D))rJrF)NNF)NTT)NT)NNF).r.r/r0r1str__annotations__rHr rrIr ControlCodepropertyr^rNrrRrTrUrW classmethodrrrgrirlr rwrzr rrrrrrrrrrrrrrrrrBrCrDrFrF<s  I!E8E?!/3GXh{+ ,3 0S00v$(D((y-9-3-5AU;V--^,s,u-A'B,4"&&* &9%&&UO & )  &&P?D@ *@8<@ ) @@$8I#68DO;T8  * %Y-. /  < "&"& ,I9%,I,I ,I  ,I  ,I $y/ ",I,I\ "& *9o** *  * i **X T4 ? Ts T T 'd4 ?3 'c3h ' ' !%!% !DO$!! !  !  ! d9o !!F  )_DO$      d9o 8  )_DO$      d9o 8  )_DO$      d9o < 38K4L8I#6L8I;NLL  +HY$7 +HY><<>D==    }} $ $sA A A ArrrrrrJr) r.r/r0r1r rFrTrrrBrCrDrrsD#'!2#t#PT# %  %+; %  %rCrcBeZdZddeeededdfdZ ddZy) SegmentLinesrrrJNc2t||_||_y)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)rrr)rLrrs rDrzSegmentLines.__init__s%[ "rCc#K|jr4tj}|jD]}|Ed{|y|jD] }|Ed{y7'7 wrP)rrFrlr)rLrrrrls rDrzSegmentLines.__rich_console__s` >>||~H        s!5A#AA#A!A#!A#rr) r.r/r0r r rFrTrrrBrCrDrrsC #htG}5 #$ #SW #     +;    rCr__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) 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)7enumr functoolsr itertoolsroperatorrtypingrrr r r r r rrrcellsrrrrrreprrrrHrrrrrrr^rrrFrrr. rich.console rich.syntaxr rich.textrcode from_markuprGruleprintrrender fragmentsrBrCrDrs!   $>>'* + +uS#X &' +sC  "  { j{  { |%%2  2 z$" D 4  < =DiG LL  MMg MM45 MM&xd;< MMO MM]W^^D)*I MM) MMO MMVW MM$ MM^;rC