o MEg9@s~ddlmZddlmZmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZmZmZmZmZGd d d eZeeeeeefeeeeffZGd d d eZed krddlmZddlmZddlmZdZe dZ!eZ"e"#de"$de"$de"$eeddde"$e"$de%e"&e!Z'e"$e'e"$e"$de"$e!e"$ddSdS))IntEnum)Dict NamedTupleOptional)cell_len set_cell_size)Style) filterfalse) attrgetter)IterableListSequenceUnionTuplec@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__BELLCARRIAGE_RETURNHOMECLEAR SHOW_CURSOR HIDE_CURSORENABLE_ALT_SCREENDISABLE_ALT_SCREEN CURSOR_UP CURSOR_DOWNCURSOR_FORWARDCURSOR_BACKWARDCURSOR_MOVE_TO_ROWCURSOR_MOVE_TO ERASE_IN_LINEr3r3f/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/venv/lib/python3.10/site-packages/rich/segment.pyr s"rc@seZdZUdZdZeed< dZee ed< dZ ee e ed< defdd Z defd d Zedefd d ZedefddZed7ddZe  d8deddee dee dedfddZe d9deddededfddZededdeedfddZe   d:deddedee dededeedf d d!Ze  d;d"eddedee dededf d#d$Zed"eddefd%d&Zed'eeddeeeffd(d)Ze   ds zSegment.__bool__cCs|jrdSt|jS)zGet cell length of segment.r)r9rr7r<r3r3r4 cell_lengthBszSegment.cell_lengthcCs |jduS)z,Check if the segment contains control codes.N)r9r<r3r3r4 is_controlGs zSegment.is_controlcCs|dS)zMake a new line segment. r3)clsr3r3r4lineLsz 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||VqdSNr3.0r7_styler9)applyrDr3r4 fs  z&Segment.apply_style..c3s6|]\}}}||r dn|r|n|VqdSrHr3rI)rDrGr3r4rMks )__add__)rDrFr8rGr3)rLrDrGr4 apply_styleQs   zSegment.apply_styleFrBcCs |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)filterr r )rDrFrBr3r3r4filter_controlwszSegment.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. rCN)appendr7r9 partition) rDrFrErRsegmentr7r8__textnew_liner3r3r4 split_liness(    zSegment.split_linesTlengthpadinclude_new_linesccsg}|j}|j}|d} |D]B} d| jvrM| jsM| \} }} | rL| d\} }} | r1||| ||rJ|||||d}|rB|| |V|dd=| 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. rC)r8rZN)rRadjust_line_lengthr7r9rS)rDrFrYr8rZr[rErRr\new_line_segmentrTr7rUrVrW cropped_liner3r3r4split_and_crop_liness4    zSegment.split_and_crop_linesrEc 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. cs|]}|jVqdSrHrArJrTr3r3r4rMz-Segment.adjust_line_length.. Nr)sumrRrAr9r) rDrErYr8rZ line_lengthrWrRrTsegment_lengthr7 segment_stylerUr3r3r4r\s.    zSegment.adjust_line_lengthcCstdd|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. csr`rHrarbr3r3r4rM rcz*Segment.get_line_length..)re)rDrEr3r3r4get_line_lengths 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|]}|VqdSrHr3)rJrErir3r4rMsz$Segment.get_shape..r)rimaxlen)rDrj max_widthr3rkr4 get_shapes  zSegment.get_shapewidthheight new_linesc Cs|durt|}g}|rtd||tdgntd||g}|j}|j} t|} t|D]} t| d} | dur=||q-|| | ||dq-|S)a Set 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. Defaults to None. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines that fits width x height. NrdrC)r8)rmr5rRr\iterrangenext) rDrjrprqr8rr shaped_linespad_linerRr\ iter_linesrUrEr3r3r4 set_shapes    zSegment.set_shapeccsvt|}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)rsru StopIterationr5r8r9r7)rDrF iter_segments last_segment_SegmentrTr3r3r4simplifyGs     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)r9r8 update_link)rDrFrTr7r8_controlr3r3r4 strip_linksbs  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)rDrFr7rKr9r3r3r4 strip_stylesss 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)get without_color)rDrFcacher7r8r9colorless_styler3r3r4 remove_colors  zSegment.remove_color)r:r5)NN)F)NTT)NT)NNF)%r r!r"r#r7str__annotations__r8rr r9r ControlCoder>r?r@propertyintrArB classmethodrEr rOrQr rXr_r\rirroryr~rrrr3r3r3r4r5&s    %    . , $    ) r5__main__)Syntax)Text)Consolezfrom 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_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)(enumrtypingrrrcellsrrr8r itertoolsr operatorr r r rrrrrrr5r rich.syntaxr rich.textr rich.consolercode from_markupr7consoleruleprintlistrender fragmentsr3r3r3r4sP    s