AHjgFdZddlmZddlmZddlmZmZmZm Z ddl Z ddl m Z mZddlmZddlmZmZmZdd lmZdd lmZdd lmZmZdd lmZmZdd lm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'gdZ(ddZ)GddZ*e jVjXGddZ-y)z This module contains the `Strip` class and related objects. A `Strip` contains the result of rendering a widget. See [Line API](/guide/widgets#line-api) for how to use Strips. ) annotations) lru_cache)AnyIterableIteratorSequenceN)cell_len set_cell_size) ColorSystem)ConsoleConsoleOptions RenderResult) Measurement)Segment)Style StyleType)index_to_cell_positionline_pad) FIFOCache)Color)AlignHorizontal AlignVertical) LineFilter) 12345678921515253c pt}t|Dcgc]\}}}|r ||c}}}Scc}}}w)zGet the line length (total length of all segments). Args: segments: Iterable of segments. Returns: Length of line in cells. )r sum)segments _cell_lentext_controls E/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/strip.pyget_line_lengthr/s2I PP$4D!W $P QQPs 1 1 cBeZdZdZdddZ ddZ d dZy) StripRenderablez7A renderable which renders a list of strips into lines.Nc ||_||_yN)_strips_width)selfstripswidths r.__init__zStripRenderable.__init__.s  c#vKtj}|jD]}|Ed{|y7 wr3)rliner4)r6consoleoptionsnew_linestrips r.__rich_console__z StripRenderable.__rich_console__2s7<<>\\ E  N  s )97 9c|jtd|jD}n |j}t||S)Nc34K|]}|jywr3) cell_length).0r@s r. z3StripRenderable.__rich_measure__..>sDe))D)r5maxr4r)r6r=r>r8s r.__rich_measure__z StripRenderable.__rich_measure__:s8 ;; Dt||DDEKKE5%((r:r3)r7 list[Strip]r8 int | NonereturnNone)r=r r>r rLr)r=z 'Console'r>z'ConsoleOptions'rLr)__name__ __module__ __qualname____doc__r9rArIr:r.r1r1+sDA)7 ) )+;) )r:r1ceZdZdZgdZ d* d+dZd,dZed-dZed.dZ e d*d/dZ e d* d0d Z e d1d Z d2d Zed3d Ze d4d Zd5dZd6dZd7dZd7dZd3dZd8dZd9dZed3dZd*d:dZd*d:dZd;dZd;dZddZd*d?dZ d@dZ!dAdZ"dBd Z#dCd!Z$e e%d"#dDd$Z&e dEd%Z'dFd&Z(dGd'Z)dHd(Z*dId)Z+y)JStripa(Represents a 'strip' (horizontal line) of a Textual Widget. A Strip is like an immutable list of Segments. The immutability allows for effective caching. Args: segments: An iterable of segments. cell_length: The cell length if known, or None to calculate on demand. ) _segments _cell_length _divide_cache _crop_cache _style_cache _filter_cache _render_cache_line_length_cache_crop_extend_cache_offsets_cache _link_ids _cell_countNc<t||_||_td|_td|_td|_td|_td|_td|_ td|_ d|_ d|_ d|_ y)N)listrUrVrrWrXrYrZr\r]r^r[r_r`)r6r)rDs r.r9zStrip.__init__^sh'FOPQl>Gm5>r]IRST aL  aL BK1)-*.'+r:c#dK |j|jy#t$rYywxYwwr3)rUrDAttributeErrorr6s r. __rich_repr__zStrip.__rich_repr__ts2 .. "" "   s0!0 -0-0cFdjd|jDS)z Segment text.c34K|]}|jywr3)r+rEsegments r.rFzStrip.text..~sBw||BrG)joinrUrgs r.r+z Strip.text{swwB4>>BBBr:c|j2|jDchc]\}}}| |jc}}|_|jScc}}w)z$A set of the link ids in this Strip.)r_rU_link_id)r6r,styles r.link_idszStrip.link_idssI >> !26..#.1eQEDUDN~~s A A c~t|trtj|n|}|t d|z|g|S)zCreate a blank strip. Args: cell_length: Desired cell length. style: Style of blank. Returns: New strip.  ) isinstancestrrparser)clsrDrq segment_styles r.blankz Strip.blanks9/9.D E*% GC+-}=> LLr:c:|Dcgc] }||| c}Scc}w)aConvert lines (lists of segments) to a list of Strips. Args: lines: List of lines, where a line is a list of segments. cell_length: Cell length of lines (must be same) or None if not known. Returns: List of strips. rR)rxlinesrDr)s r. from_lineszStrip.from_liness .blank_liness;ue45u=E5\   s-0rtopmiddlebottomleftcenterright)rintrLIterable[Strip])rDrHlenrTrrU)rxr7rqr8height horizontalverticalr@ line_lengths shape_width shape_heightrtop_blank_linesbottom_blank_linesvertical_excess_space left_spaces` `` r.alignz Strip.aligns0 7=>e))> >,' <(  010,  $'6L+@$A !5 %:"X%"71"<%:_%L"X%"7&777   $$-K !UU=N=N5NPUV  8 #Q 349J $$-K !OO&!E$5$55 B!   7 " $$-K %%:K:K2KQPUV   "#56 6 6 G?<8L 7s/F+F!A!F+F&DF+F) F+)F+c.t|j|S)a?Given a character index, return the cell position of that character. This is the sum of the cell lengths of all the characters *before* the character at `index`. Args: index: The index to convert. Returns: The cell position of the character at `index`. )rrUr6indexs r.rzStrip.index_to_cell_positions&dnne<>""r:cXt|txr|j|jk(Sr3)rurTrU)r6r@s r.__eq__z Strip.__eq__=s!%'OT^^u-NOr:ct|trt||dz}|j|j|j |j S|j S)N)rurslicecropstartstop cell_countrs r. __getitem__zStrip.__getitem__@sT eS !%+Eyy KKEJJ,>  DIJJ  r:ct|j!td|jD|_|jS)zNumber of cells in the stripc3FK|]}t|jywr3)rr+rls r.rFz#Strip.cell_count..Ks"S3w||#4"Ss!)r`r(rUrgs r.rzStrip.cell_countGs3    #""SDNN"SSD r:c|j|kr8||jz }|jtd|z|gz}t||S|S)zExtend the cell length if it is less than the given value. Args: cell_length: Required minimum cell length. style: Style for padding if the cell length is extended. Returns: A new Strip. rt)rDrUrrT)r6rDrq missing_spacer)s r.extend_cell_lengthzStrip.extend_cell_lengthNsQ   k )'$*:*::M~~}1De)L(MMH;/ /Kr:c|j|k(r|S||f}|jj|}||S|j}|j}t}||kr ||d||z z|gz}t ||} nr||kDrkg}|j } d} |D]I} | j} | | z|kr| | | | z } %| \}}}t||| z }| |||nt ||} n|} | |j|<| S)aAdjust the cell length, possibly truncating or extending. Args: cell_length: New desired cell length. style: Style when extending, or `None`. Returns: A new strip with the supplied cell length. rtr)rDr\getrUrrTappendr )r6rDrq cache_key cached_stripr<current_cell_length_Segmentr?r@r line_lengthrmsegment_lengthr+ryr,s r.adjust_cell_lengthzStrip.adjust_cell_length_sE   { *K %( ..229=  # ~~"..  , .A ABEJH(K0E ; .H__FK !(!4!4/+=7O>1K-4*D-({[/HID8D-89 (K0EE-2 * r:clttj|j|j}|S)zhSimplify the segments (join segments with same style). Returns: New strip. )rTrsimplifyrUrV)r6r<s r.rzStrip.simplifys0    T^^ ,     r:c|dd}t|jDcgc] }|| c}|jScc}w)zPRemove all meta from segments. Returns: New strip. c^|\}}}||S|j}d|_t|||S)zBuild a Segment with no meta. Args: segment: Segment. Returns: Segment, sans meta. N)copy_metar)rmr+rqr-s r.remove_meta_from_segmentz4Strip.discard_meta..remove_meta_from_segments;$+ D%}JJLEEK40 0r:)rmrrLr)rTrUrV)r6rrms r. discard_metazStrip.discard_metas: 1 >Bnn M7 %g . M     Ms9c|jj||f}|At|j|j||j }||j||f<|S)zApply a filter to all segments in the strip. Args: filter: A line filter object. Returns: A new Strip. )rZrrTapplyrUrV)r6filter backgroundrs r. apply_filterzStrip.apply_filtersg))--vz.BC    T^^Z8$:K:KL8DD   3 4r:c t}||jvr|S|jDcgc]6\}}}|||r$|js|j|k(r |||zndn||8}}}}t ||j Scc}}}w)zApply a style to Segments with the given link_id. Args: link_id: A link id. link_style: Style to apply. Returns: New strip (or same Strip if no changes). N)rrrrU_nullrprTrV)r6link_id link_stylerr+rqr-r)s r. style_linkszStrip.style_linkss $-- 'K)-   %eW ekkenn6O,1+  $ ''U388D-2 * r:cttd|}| |jnt|j|}|dk(r||jk(r|S||kr tgdS||f}|jj |}||St }d}g}|j}t|j} d} ||jk\r tgd} n| D]7} ||| jz} | |kDr| j||z d} n| }9||jk\r5| r|| |j| t||j|z } ne|}| R||| jz} | |kr || n|| j||z dn| }t| d} | Rt|||z } | |j|<| S)zCrop a strip between two cell positions. Args: start: The start cell position (inclusive). end: The end cell position (exclusive). Returns: A new Strip. rNr)rHrDminrTrXrr rrrUr+ split_cellsextendnext) r6rrrcachedr*posoutput_segments add_segment iter_segmentsrmr@end_poss r.rz Strip.cropsAu "%+d3t7G7G3M A:#!1!11K %<Q< CL !!%%i0  M )+%,, T^^, "& D$$ $"aLE(  ',, 77U?%11%#+>qAG  d&&&(&&}5ot/?/?%/GH)!Igll$;;G}#G,#G$7$7c $B1$EF!C"=$7G)osU{;&+# r:cd}|j}|Dcgc] }||ks | }}t|}|jj|}||S||gk(r|g}nWg}|j}t t j|j||D]\} }|t| ||z |}||j|<|Scc}w)zDivide the strip into multiple smaller strips by cutting at given (cell) indices. Args: cuts: An iterable of cell positions as ints. Returns: A new list of strips. r) rDtuplerWrrziprdividerUrT) r6cutsrrDcutrrr7 add_stripr)s r.rz Strip.divide8s&& #:sk'9::$K ##'' 2  M K= VFF I!$W^^DNND%I4!P  #%#)45 )/9% %;s CCc|jj|}||Sttj|j ||j }||j|<|S)zApply a style to the Strip. Args: style: A Rich style. Returns: A new strip. )rYrrTr apply_stylerUrD)r6rqr styled_strips r.rzStrip.apply_styleXsc""&&u-  M    68H8H $0% r:cNtj|}|j|S)zApply meta to all segments. Args: meta: A dict of meta information. Returns: A new strip. )r from_metar)r6meta meta_styles r. apply_metazStrip.apply_metajs#__T*  ++r:c |j}t}|Dcgc];\}}}|r+|||j|nd|jvr||zn||n||=}}}}t ||j Scc}}}w)Nz@click)rUrrrrTrV)r6rr)rr+rqr-s r._apply_link_stylezStrip._apply_link_stylews>>)1  %eW!;;.4< 4Jej0PUd^ $  Xt0011! sAA3i@)maxsizec|j|jzx}r8t}t|j Dcgc]}|d|zzr||}}ng}|j x}.|j |j|j|jx}/|j |j|jddj|x} |_ | Scc}w)zRender ANSI codes for a give style. Args: style: A Rich style. color_system: Color system enumeration. Returns: A string of ANSI escape sequences to render the style. rF;) _attributes_set_attributes SGR_STYLESr bit_length_colorr downgradeget_ansi_codes_bgcolorrn_ansi) rxrq color_system attributes _style_map bit_offsetsgrcolorbgcoloransis r. render_ansizStrip.render_ansis**U-B-BB B: B#J#( (=(=(?"@j1:&C C\\ !E . JJu|4CCE F~~ %G 2 JJw((6EEeL M XXc]*u{ sC)c|jx}|j||}|r d|d|dn|}|jr d|jd|jd|d}|S)zRender a Rich style and text. Args: style: Style to render. text: Content string. color_system: Color system enumeration. Returns: Text with ANSI escape sequences. z[mzz]8;id=rz\z]8;;\)rr _linkrp)rxrqr+rr outputs r. render_stylezStrip.render_stylesoKK D (??5,7D155avW-4 ;;U^^,Aekk]&W  r:c |jm|jxstj}|j}dj |j Dcgc]\}}}||n ||||c}}}|_|jScc}}}w)zRender the strip into terminal sequences. Args: console: Console instance. Returns: Rendered sequences. rj)r)r[ _color_systemr TRUECOLORrrnrU)r6r=rrenderr+rqr,s r.rz Strip.renders    %"00IK4I4IL&&F!#+/..  'eQ!=#E4lKL "D !!!sB c0||jk7r|j||}n|}|s|s|S|jj}|r|j dt d|z||r|j t d|z|t|||z|zS)aYCrop the strip to `cell_length`, and add optional padding. Args: cell_length: Cell length of strip prior to padding. left: Additional padding on the left. right: Additional padding on the right. style: Style of any padding. Returns: Cropped and padded strip. rrt)rDrrUrinsertrrrT)r6rDrrrqr@r)s r.crop_padzStrip.crop_pads $** *++K?EEL??'')  OOAwsTz59 :  OOGC%K7 8X{T1E9::r:c L|dk(rR|j|k(r|Stt|jd||jz t j |S|dk(rqt d||jz dz}|j|k(r|Stt|j|||jz |z t j |S|dk(rR|j|k(r|Stt|j||jz dt j |Sy)Nrrrrr)rDrTrrUrnullrH)r6r8rrs r. text_alignzStrip.text_aligns F?5( T^^Q8H8H0H%**,Wh Q(8(8 89Q>J5( " 0 00:=  g 5( T^^UT5E5E-Eq%**,W r:cr||f}|jj|x}|S|j}g}|D][}|\}} } tjd||fi} |j t || r| | zn| |t|jz }]t||j} | |j|<| S)zApply offsets used in text selection. Args: x: Offset on X axis (column). y: Offset on Y axis (row). Returns: New strip. offset) r^rrUrrrrrr+rTrV) r6xyrrr)strip_segmentsrmr+rqr, offset_styler@s r. apply_offsetszStrip.apply_offsetssF  //33I> >L K >>(* #G$ND% ??Hq!f+=>L  ! !eel2N  W\\" "A  #nd&7&78).I& r:r3)r)Iterable[Segment]rDrKrLrM)rLzrich.repr.Result)rLrv)rLzset[str])rDrrqzStyleType | NonerLrT)r|zlist[list[Segment]]rDrKrLrJ)r7rJrqrr8rrrKrrrrrLr)rrrLr)rLr)r7zIterable[Strip | None]rLrT)rrTrLrT)rLbool)rLzIterator[Segment])r@objectrLr#)rz int | slicerLrT)rDrrq Style | NonerLrT)rLrT)rrrrrLrT)rrvrrrLrT)rrrrrqr%rLrT)rrrrKrLrT)rz Iterable[int]rLzSequence[Strip])rqrrLrT)rzdict[str, Any]rLrT)rrrLrT)rqrrr rLrv)rqrr+rvrr rLrv)r=r rLrv) rDrrrrrrqrrLrT)r8rrrrLrT)rrrrrLrT),rNrOrPrQ __slots__r9rhpropertyr+rr classmethodrzr}rrrDrnrrrrrrrrrrrrrrrrrrrrrr rrrrr!rRr:r.rTrTDs I FJ,),8B, ,,CC M MCG B' B6@ B  B B]7]7]7 ]7  ]7 $ ]7 ]7 ]7]7~ =!!**$)&$(#P   "5n  8"28&8t@$ ,2*u6("0;2 Dr:rT)r)r"rLr).rQ __future__r functoolsrtypingrrrr rich.reprrich rich.cellsr r rich.colorr rich.consoler r r rich.measurer rich.segmentr rich.stylerrtextual._segment_toolsrr textual.cacher textual.colorrtextual.css.typesrrtextual.filterrrr/r1reprautorTrRr:r.r;sx#44.">>$ 'C#<% R  R))2d d d r: