o RhY@s,ddlmZmZmZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZmZddlmZmZddlmZddlmZmZddlmZdd lmZdd lmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+erddl,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2eGdddZ3eGdddZ4Gddde Z5GdddeZ6e7dkrddl8m-Z-ddl9m:Z:ddl;md"d#d$d%d&e=j>d'd(d)e=j>d*d d+d,e=?d-d.d/e=?d0d1d2e=j?d3d4d5d6d%d7e=?d8d9d:d;e@dd?ZAe-Z,e:ZBeAd@e,jCe=dAdBd%e=_DeAdCe,Ce=dDe=_EeAdEe,jCe=dAdBd=e=_EdFe=_DdGdHge=_FeAdIe,jCe=dAdBd=e=_EdFe=_DdGdHge=_Fde=_GeAdJe,jCe=dAdBd=e=_EdFe=_DdGdHge=_Fd%e=_Hde=_GeAdKe,jCe=dAdBWd=d=S1swYd=Sd=S)L) dataclassfieldreplace) TYPE_CHECKINGDictIterableList NamedTupleOptionalSequenceTupleUnion)boxerrors)loop_first_last loop_last) pick_bool)ratio_distribute ratio_reduce)VerticalAlignMethod) JupyterMixin) Measurement)PaddingPaddingDimensions) is_renderable)Segment)Style StyleType)TextTextType)ConsoleConsoleOptions JustifyMethodOverflowMethodRenderableType RenderResultc@s<eZdZUdZdZded< dZded< dZeed< dZ eed< dZ eed< d Z d ed < d Z d ed< dZ ded< dZeeed< dZeeed< dZeeed< dZeeed< dZeed< dZeed< dZeed< eedZeded<d%dd Zededfd!d"Zedefd#d$Z dS)&Columna Defines a column within a ~Table. Args: title (Union[str, Text], optional): The title of the table rendered at the top. Defaults to None. caption (Union[str, Text], optional): The table caption rendered below. Defaults to None. width (int, optional): The width in characters of the table, or ``None`` to automatically fit. Defaults to None. min_width (Optional[int], optional): The minimum width of the table, or ``None`` for no minimum. Defaults to None. box (box.Box, optional): One of the constants in box.py used to draw the edges (see :ref:`appendix_box`), or ``None`` for no box lines. Defaults to box.HEAVY_HEAD. safe_box (Optional[bool], optional): Disable box characters that don't display on windows legacy terminal with *raster* fonts. Defaults to True. padding (PaddingDimensions, optional): Padding for cells (top, right, bottom, left). Defaults to (0, 1). collapse_padding (bool, optional): Enable collapsing of padding around cells. Defaults to False. pad_edge (bool, optional): Enable padding of edge cells. Defaults to True. expand (bool, optional): Expand the table to fit the available space if ``True``, otherwise the table width will be auto-calculated. Defaults to False. show_header (bool, optional): Show a header row. Defaults to True. show_footer (bool, optional): Show a footer row. Defaults to False. show_edge (bool, optional): Draw a box around the outside of the table. Defaults to True. show_lines (bool, optional): Draw lines between every row. Defaults to False. leading (int, optional): Number of blank lines between rows (precludes ``show_lines``). Defaults to 0. style (Union[str, Style], optional): Default style for the table. Defaults to "none". row_styles (List[Union, str], optional): Optional list of row styles, if more than one style is given then the styles will alternate. Defaults to None. header_style (Union[str, Style], optional): Style of the header. Defaults to "table.header". footer_style (Union[str, Style], optional): Style of the footer. Defaults to "table.footer". border_style (Union[str, Style], optional): Style of the border. Defaults to None. title_style (Union[str, Style], optional): Style of the title. Defaults to None. caption_style (Union[str, Style], optional): Style of the caption. Defaults to None. title_justify (str, optional): Justify method for title. Defaults to "center". caption_justify (str, optional): Justify method for caption. Defaults to "center". highlight (bool, optional): Highlight cell contents (if str). Defaults to False. r%headerfooter header_style footer_stylestyleleftr#justifytoprverticalellipsisr$overflowNwidth min_width max_widthratioFno_wrap highlightr_index)default_factory_cellsreturncCs t|gdS)zReturn a copy of this Column.r<)rselfrAL/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/rich/table.pycopyus z Column.copyccs|jEdHdS)z2Get all cells in the column, not including header.Nr>r?rArArBcellsysz Column.cellscCs |jduS)z!Check if this column is flexible.N)r7r?rArArBflexible~ zColumn.flexible)r=r')!__name__ __module__ __qualname____doc__r)__annotations__r*r+rr,r-r/r1r3r4r intr5r6r7r8boolr9r:rlistr<rrCpropertyrrDrErArArArBr'&sL             r'c@s0eZdZUdZdZeeed< dZe ed<dS)RowzInformation regarding a row.Nr-F end_section) rGrHrIrJr-r rrKrQrMrArArArBrPs  rPc@s.eZdZUdZeed< ded< eed<dS)_CellzA single cell in a table.r-r% renderabler1N)rGrHrIrJrrKrrArArArBrRs rRc7@s~eZdZUdZeeed<eeed<ddddej ddddddddddd dd d dddd d dd de ee fde e de e de ede ede ejde ededededededededededede eed e ed!e ed"e ed#e ed$e ed%d&d'd&d(ed)df6d*d+Zeddddd,de ee fdedededed)df d-d.Zed)efd/d0Zejded)dfd1d0Zed)efd2d3Zed)efd4d5Zd6d7d8ed)efd9d:Zd6d7d;dZed)eeeeeffd?d@Z e jded)dfdAd@Z B BdndddddCdDdEddddddF dGdHdIdHd e ed(e ed!e ede edJd&dKdLdMdNde ede edOe edPe edQed)dfdRdSZ!dddTdUe dHde edVed)dfdWdXZ"dodYdZZ# 6 7 ; < ) [dpd\d]Z$d6d7d;dRsz*Table.__rich_measure__..cs|]}|jVqdSN)minimumr measurementrArArB Wz)Table.__rich_measure__..csrrmaximumrrArArBrZr) r6r4rrysum_calculate_column_widthsrrrUclampr5)r@r{r extra_width measurements minimum_width maximum_widthrrArrB__rich_measure__As0     zTable.__rich_measure__cCs|jS)zGet cell padding.)rmr?rArArBr\bsz Table.paddingcCst||_|S)zSet cell padding.)rrlrm)r@r\rArArBr\gs r(r.r0r2) r+r9r,r-r/r1r3r4r5r6r7r8r)r%r*r/r1rr3r$r6r7r8c CsVtt|j|||p d|dur|n|j|pd|pd||| | | | | |d}|j|dS)aAdd a column to the table. Args: header (RenderableType, optional): Text or renderable for the header. Defaults to "". footer (RenderableType, optional): Text or renderable for the footer. Defaults to "". header_style (Union[str, Style], optional): Style for the header, or None for default. Defaults to None. highlight (bool, optional): Whether to highlight the text. The default of None uses the value of the table (self) object. footer_style (Union[str, Style], optional): Style for the footer, or None for default. Defaults to None. style (Union[str, Style], optional): Style for the column cells, or None for default. Defaults to None. justify (JustifyMethod, optional): Alignment for cells. Defaults to "left". vertical (VerticalAlignMethod, optional): Vertical alignment, one of "top", "middle", or "bottom". Defaults to "top". overflow (OverflowMethod): Overflow method: "crop", "fold", "ellipsis". Defaults to "ellipsis". width (int, optional): Desired width of column in characters, or None to fit to contents. Defaults to None. min_width (Optional[int], optional): Minimum width of column, or ``None`` for no minimum. Defaults to None. max_width (Optional[int], optional): Maximum width of column, or ``None`` for no maximum. Defaults to None. ratio (int, optional): Flexible ratio for the column (requires ``Table.expand`` or ``Table.width``). Defaults to None. no_wrap (bool, optional): Set to ``True`` to disable wrapping of this column. r(N)r:r)r*r+r9r,r-r/r1r3r4r5r6r7r8)r'rsrUr9ro)r@r)r*r+r9r,r-r/r1r3r4r5r6r7r8rrArArBrrms$'zTable.add_columnr-rQ renderablesrQc Gsdtddddfdd}t|}|j}t|t|kr)g|dgt|t|}t|D]H\}}|t|krRt||jd} |jD] } || td qA|j| n||} |dur`|| d q-t |rj|| |q-t d t |j d |jt||d dS) aAdd a row of renderables. Args: *renderables (None or renderable): Each cell in a row must be a renderable object (including str), or ``None`` for a blank cell. style (StyleType, optional): An optional style to apply to the entire row. Defaults to None. end_section (bool, optional): End a section and draw a line. Defaults to False. Raises: errors.NotRenderableError: If you add something that can't be rendered. rrSr%r=NcSs|j|dSr)r<ro)rrSrArArBadd_cellszTable.add_row..add_cell)r:r9r(zunable to render z1; a string or other renderable object is requiredr)r'rNrUrs enumerater9rVrrorrNotRenderableErrortyperGrP) r@r-rQrrcell_renderablesrUr|rSr_rArArBadd_rows0    z Table.add_rowcCs|jr d|jd_dSdS)z2Add a new section (draw a line after current row).TN)rVrQr?rArArB add_sectionszTable.add_sectionr&c #s|js tdVdS|j}|jdur|j}|j}||||}t||}|j||j dd ddt dt dddd ffd d }|j rY||j t |jd |jd EdH||EdH|jry||jt |jd|jd EdHdSdS)N )r4r9heightrXtextr-r/r#r=r&cs4t|tr j||ddn|}j|j|ddS)NF)r-r9r/)r)rprq render_strrenderupdate)rr-r/ render_textr{render_optionsrArBrender_annotations z1Table.__rich_console__..render_annotationz table.title)r-r/z table.caption)rX)rUrr6r4ryrrrrr9r rrYr pick_firstrgri_renderrZrhrj)r@r{rr6rwidths table_widthrrArrB__rich_console__sN       zTable.__rich_console__csj}j}fdd|D}dd|D}jj}jredd|D}t|reddt||D} fdd|D} |t| } t| || } t | } t |D]\}}|j rd| |t | ||<qSt|}||kr |dd|D|}t|}||kr||}t|dgt|||}t|}fd dt||D}d d|D}||krjsjd ur|j|kr݈jd ur|ntj||}t|||}d dt||D}|S) zNCalculate the widths of each column, including padding, not including borders.csg|] }|qSrA)rrr{rr@rArBrsz2Table._calculate_column_widths..cSg|]}|jpdqSrrr_rangerArArBrcSsg|] }|jr|jp dqSr)rEr7)rcolrArArBrscSs g|] \}}|jr dn|jqSr)rEr)rrrrArArBrscs&g|]}|jr|jp d|jqSr)rEr4r:r)get_padding_widthrArBrs cSsg|] }|jduo |j qSr)r4r8rrArArBr.srcs$g|]\}}||qSrA)rr)rr4rrrArBr8cSrrrrrArArBr<rNcSsg|]\}}||qSrArA)r_widthpadrArArBrGs)r6rU_get_padding_widthryr_anyziprriterrrEnext_collapse_widthsrrsr5min)r@r{rr6rU width_rangesrrratios fixed_widths flex_minimumflexible_width flex_widthsiter_flex_widthsr|rr excess_width _max_width pad_widthsrA)r{rrr@rBr sb     zTable._calculate_column_widthsrwrapablec st|}||}t|rd|rd|dkrdtddt||Dtfddt||D}|}fddt||D}t|rB|sE |St||gt|} t||| |}t|}||}|rd|dks|S)aMReduce widths so that the total is under max_width. Args: widths (List[int]): List of widths. wrapable (List[bool]): List of booleans that indicate if a column may shrink. max_width (int): Maximum width to reduce to. Returns: List[int]: A new list of widths. rcss|] \}}|r|VqdSrrArr4 allow_wraprArArBr]s  z)Table._collapse_widths..c3s(|]\}}|r|kr|ndVqdS)rNrAr max_columnrArBr`s  cs$g|]\}}|kr|rdndqS)rrrArrrArBrerz*Table._collapse_widths..)rrmaxrrrsr) rvrrr6 total_widthrsecond_max_columncolumn_differencer max_reducerArrBrKs.     zTable._collapse_widths column_indexrc #sr|j|j|jt}|dk|t|jdkidtdtdtttttfffdd }g}|j }|j }|j rS||j pEd||j } || |j f||jpXd} |jD]} || | fq]|jr|||jpnd||j} || |jf|rt} t|D]\}}\}}t|| ||||t|d d p|jVqd S|D]\}}t||t|d d p|jVqd S) z3Get all the cells with padding and optional header.rr first_rowlast_rowr=cs||f}|r |S\}}}}r%std||}|s%td||}s7r+d}r/d}|r3d}|r7d}||||f}|||f<|S)Nr)getr)rrcachedr0rightbottomr.rm_padding_cacher] first_column last_columnr^r\rArB get_paddings*   z%Table._get_cells..get_paddingr(r1N)r]r^r\rrsrUrMr rLror~r`r+r)r-rDrar,r*rrrRgetattrr1)r@r{rr any_paddingr raw_cells_appendr~r+ cell_stylecellr,_Paddingfirstlastr-rSrArrB _get_cellsrsR0  zTable._get_cellscCs2|j\}}}}|jr|dkrtd||}||S)zGet extra width from padding.r)r\r]r)r@rr pad_rightpad_leftrArArBrs zTable._get_padding_widthcCs|j}|dkr tddS||j}|jdur%t|j||j||Sg}g}|j}|j} tj} |||j|D]} | ||| j \} } || | | q:t|rUt |nd|r\t |n||}| |j durkdn|j ||jduryd}|S|j|}|S)z0Get the minimum and maximum width of the column.rrN) r6rrr:r4 with_maximumrorrrSrrr5)r@r{rrr6 padding_width min_widths max_widths append_min append_maxget_render_widthr_min_maxrrArArBrsB     zTable._measure_columnc% #s jpd}| jpd} fddt jD}tt|} jr6 jj|t j j dnd}|rA j sA| n|}t } j} j } j} j} j} j}t |r|j||j||j|f|j||j||j|f|j||j||j|fg}| r|||V| Vng} j}jtt|D]\}\}}}|o| |o| sLjsLj j|| nd}dg}sшrt !n || r|dn|t||| D].\}}}|j"||j#|j$|j%d|j&d}j'|j(||jd}t)t*||+|qt)dd|Dd t,t,t d d d t-d t dt,t,t f fdd fddt|||| D|dd<|r|rk| rk|j.|d| d|V| V||rqdn|rvdnd\}}}|j/0r|n |j/j1|j} t2D])}!| r|Vt3|D]\}"}#|#|!EdH|"s| Vq| r|V| Vqnt2D]}!|D] }#|#|!EdHq| Vq|r|r| r|j.|d| d|V| V|o|j4}$|r5|s|s|$r5|s5| r|t*|dks5| rs5|r&|j.|d| d||Vn |j.|d| d|V| Vq|rJ| rL|5||V| VdSdSdS)Nr(c3s"|] \}}||VqdSr)r)rrr)r{r@rArBrs  z Table._render..)safer)r4r/r8r3rr9r-css|]}t|VqdSr)rs)rrrArArBrEsrr1rr4r-r=csRrd}nr d}|dkr|||S|dkr!|||S|||S)Nrr0middle) align_top align_middle align_bottom)rr1r4r-)_Segment footer_row header_row row_heightrArB align_cellGsz!Table._render..align_cellc s8g|]\}}}}||j||j|qSrA) set_shaper1r-)rr4_cellrr)rrr~ max_heightrrArBrXs  z!Table._render..foot)edgerrxheadmidrow)6r~r-rfrrUrNrr substituterr[r`get_plain_headed_boxrlinerarbrcrd head_left head_right head_verticalmid_left mid_right mid_vertical foot_left foot_right foot_verticalget_toprrrVrr}rr/r8r3r9 render_linesrSrrsrorrLget_rowrstripbackground_stylerangerrQ get_bottom)%r@r{rr table_stylerf _column_cells row_cells_boxnew_linerUr`rarbrcrd box_segmentsrr|rrrow_cellrrDr4rrrlinesr.r_dividerdividerline_no last_cell rendered_cellrQrA) rrr{rr~rr rrr@rBrs*                  "       z Table._render)r(r()r=N)r{r!rr"r=r&),rGrHrIrJrr'rKrPr HEAVY_HEADr rqr r rLBoxrMrrrru classmethodrwrOr_setterryrzrrrr r\rrrrrrrrRrrrrrArArArBrTs           @      !      = / / @ & I (rT__main__)r!)ReprHighlighter)timerz Table renderzStar Wars MovieszRich example tabler)rYrZrjReleased bright_cyancyanT)r+r-r8Titlemagentarz Box Officegreen)r/r-z Dec 20, 2019z Star Wars: The Rise of Skywalkerz $952,110,690z May 25, 2018zSolo: A Star Wars Storyz $393,151,347z Dec 15, 2017z!Star Wars Ep. V111: The Last Jediz$1,332,539,889zon blackrz Dec 16, 2016zRogue One: A Star Wars Storyz$1,332,439,889rr=NcCs"ttt|tdSr)r{printruler9)rrArArBr)s r)z Example TablerXrz expand=True2zwidth=50FdimrWzrow_styles=['dim', 'none']z%leading=1, row_styles=['dim', 'none']z+show_lines=True, row_styles=['dim', 'none'])I dataclassesrrrtypingrrrrr r r r r r(rr_looprr_pickr_ratiorralignrjupyterrmeasurerr\rrprotocolrsegmentrr-rrrrr r{r!r"r#r$r%r&r'rPrRrTrG rich.consolerich.highlighterr5_timerr6tablerrrrqr)r9r=r_r4rerdrcrArArArBs,         ]          $