a )ja@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 Z5GdddeZ6e7dkr$ddl8m-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:e@d;d<d=d>ZAe-Z,e:ZBeAd?e,jCe=d@dAd%e=_DeAdBe,Ce=dCe=_EeAdDe,jCe=d@dAd;e=_EdEe=_DdFdGge=_FeAdHe,jCe=d@dAd;e=_EdEe=_DdFdGge=_Fde=_GeAdIe,jCe=d@dAd;e=_EdEe=_DdFdGge=_Fd%e=_Hde=_GeAdJe,jCe=d@dAWd;n1s0Yd;S)K) 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<dddd Zeeddd!d"Zeedd#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. 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<)rselfrB0E:\soccer\ml_env\Lib\site-packages\rich/table.pycopytsz Column.copyccs|jEdHdS)z2Get all cells in the column, not including header.Nr?r@rBrBrCcellsxsz Column.cellscCs |jduS)z!Check if this column is flexible.N)r7r@rBrBrCflexible}szColumn.flexible)!__name__ __module__ __qualname____doc__r)__annotations__r*r+rr,r-r/r1r3r4r intr5r6r7r8boolr9r:rlistr<rrDpropertyrrErFrBrBrBrCr'&s,            r'c@s.eZdZUdZdZeeed<dZe ed<dS)RowzInformation regarding a row.Nr-F end_section) rGrHrIrJr-r rrKrQrMrBrBrBrCrPs  rPc@s*eZdZUdZeed<ded<eed<dS)_CellzA single cell in a table.r-r% renderabler1N)rGrHrIrJrrKrrBrBrBrCrRs rRc@seZdZUdZeeed<eeed<ddddej ddddddddddd dd d dddd d dd e ee fe e e e e ee ee eje eeeeeeeeeeee eee ee ee ee ee eddeddddZeddddde ee feeeeddddZeedddZejeddddZeedddZeedddZdeed d!d"Zdd#ed$d%d&Zeeeeeefdd'd(Z e jedd)d*d(Z dPddddd,d-d.dddddd/ d0d0e ee ee ee edd1d2e ee ee ee eedd3d4d5Z!ddd6e d0e eedd7d8d9Z"ddd:d;Z#dd#dZ$dd#eed$d?d@Z%eeeeeeeedAdBdCZ&deeee'dDdEdFZ(eedGdHdIZ)dd#eedJdKdLZ*dd#eedcGsg|_g|_||_||_||_||_||_||_t ||_ | |_ | |_ | |_ | |_| |_||_||_||_||_|pxd|_|pd|_||_||_||_||_||_||_t|pg|_|jj}|D]0}t|t r|j!|dqt"|j|_#||qdS)Nr()r))$rUrVrYrZr4r5rr[runpack_paddingr^_expandr`rarbrcrdr]r-r+r,rfrgrhrirjr9rNreappend isinstancestr add_columnlenr:)rArYrZr4r5rr[r\r]r^r_r`rarbrcrdr-rer+r,rfrgrhrirjr9rk append_columnr)rBrBrC__init__sB     zTable.__init__)r\r]r^r_)rkr\r]r^r_r>c Gs||d||ddd||dS)aGet a table with no lines, headers, or footer. Args: *headers (Union[Column, str]): Column headers, either as a string, or :class:`~rich.table.Column` instance. padding (PaddingDimensions, optional): Get padding around cells. Defaults to 0. collapse_padding (bool, optional): Enable collapsing of padding around cells. Defaults to True. pad_edge (bool, optional): Enable padding around edges of table. Defaults to False. expand (bool, optional): Expand the table to fit the available space if ``True``, otherwise the table width will be auto-calculated. Defaults to False. Returns: Table: A table instance. NF)rr\r]r`rarbr^r_rB)clsr\r]r^r_rkrBrBrCgridsz Table.gridr=cCs|jp|jduS)z-Setting a non-None self.width implies expand.N)rnr4r@rBrBrCr_sz Table.expand)r_r>cCs ||_dS)z Set expand.N)rn)rAr_rBrBrCr_"scCs4d}|jr|jr|d7}|jr0|t|jd7}|S)z'Get extra width to add to cell content.rr)rrbrsrU)rAr4rBrBrC _extra_width's  zTable._extra_widthcCs t|jS)zGet the current number of rows.)rsrVr@rBrBrC row_count1szTable.row_countr!)consoleindexr>cCsRt}|jr,|||j|t|j7}|j|j}|durN|||7}|S)zGet the current row style.N)rnullre get_stylersrVr-)rAr{r|r- row_stylerBrBrC get_row_style6s zTable.get_row_styler")r{optionsr>csj|jdur|jdkr(tddS|j}t|||jfdd|jD}tdd|D|}|jdurtdd|D|n|j}t||}| |j }|S)Nrcsg|]}|qSrB) update_width.0column_measure_columnr{r6rrBrC Qsz*Table.__rich_measure__..css|] }|jVqdSN)minimumr measurementrBrBrC Vz)Table.__rich_measure__..css|] }|jVqdSrmaximumrrBrBrCrYr) r6r4rrysum_calculate_column_widthsrrrUclampr5)rAr{r extra_width measurements minimum_width maximum_widthrrBrrC__rich_measure__@s0    zTable.__rich_measure__cCs|jS)zGet cell padding.)rmr@rBrBrCr\asz Table.padding)r\r>cCst||_|S)zSet cell padding.)rrlrm)rAr\rBrBrCr\fs r(r.r0r2) r+r9r,r-r/r1r3r4r5r6r7r8r%rr$)r)r*r+r9r,r-r/r1r3r4r5r6r7r8r>c CsVtt|j|||pd|dur |n|j|p*d|p0d||| | | | | |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)rAr)r*r+r9r,r-r/r1r3r4r5r6r7r8rrBrBrCrrls$'zTable.add_columnr-rQ) renderablesr-rQr>c Gstddddd}t|}|j}t|t|krNg|dgt|t|}t|D]\}}|t|krt||jd} |jD]} || tdq~|j| n||} |dur|| dqVt |r|| |qVt dt |j d qV|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. r%N)rrSr>cSs|j|dSr)r<ro)rrSrBrBrCadd_cellszTable.add_row..add_cell)r:r9r(zunable to render z1; a string or other renderable object is requiredr)r'rNrUrs enumerater9rVrrorrNotRenderableErrortyperGrP) rAr-rQrrcell_renderablesrUr|rSr_rBrBrCadd_rows0    z Table.add_rowcCs|jrd|jd_dS)z2Add a new section (draw a line after current row).TN)rVrQr@rBrBrC add_sectionszTable.add_sectionr&c#s|jstdVdS|j}|jdur*|j}|j}||||}t||}|j||j ddd t t dddfdd }|j r||j t |jd |jd EdH||EdH|jr||jt |jd |jd EdHdS) N )r4r9heightrXr#r&)textr-r/r>cs4t|trj||ddn|}j|j|ddS)NF)r-r9r/)r)rprq render_strrenderupdate)rr-r/ render_textr{render_optionsrBrCrender_annotations z1Table.__rich_console__..render_annotationz table.title)r-r/z table.caption)rX)rUrr6r4ryrrrrr9r rrYr pick_firstrgri_renderrZrhrj)rAr{rr6rwidths table_widthrrBrrC__rich_console__s@       zTable.__rich_console__csj}j}fdd|D}dd|D}jj}jrdd|D}t|rddt||D} fdd|D} |t| } t| || } t | } t |D]"\}}|j r| |t | ||<qt|}||krX |dd|D|}t|}||kr.||}t|dgt|||}t|}fd dt||D}d d|D}||krjjsjd ur|j|krȈjd ur|ntj||}t|||}d dt||D}|S) zNCalculate the widths of each column, including padding, not including borders.csg|]}|qSrB)rrr{rrArBrCrsz2Table._calculate_column_widths..cSsg|]}|jpdqSrrr_rangerBrBrCrrcSsg|]}|jr|jpdqSr)rFr7)rcolrBrBrCrrcSs g|]\}}|jrdn|jqSr)rFr)rrrrBrBrCrscs&g|]}|jr|jpd|jqSr)rFr4r:r)get_padding_widthrBrCrscSsg|]}|jduo|j qSr)r4r8rrBrBrCr/rrcs$g|]\}}||qSrB)rr)rr4rrrBrCr9scSsg|]}|jpdqSrrrrBrBrCr=rNcSsg|]\}}||qSrBrB)r_widthpadrBrBrCrHr)r6rU_get_padding_widthryr_anyziprriterrrFnext_collapse_widthsrrsr5min)rAr{rr6rU width_rangesrrratios fixed_widths flex_minimumflexible_width flex_widthsiter_flex_widthsr|rr excess_width _max_width pad_widthsrB)r{rrrArCr sf       zTable._calculate_column_widths)rwrapabler6r>c st|}||}t|r|r|dkrtddt||Dtfddt||D}|}fddt||D}t|r|sqt||gt|} t||| |}t|}||}q|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|VqdSrrBrr4 allow_wraprBrBrCr^sz)Table._collapse_widths..c3s&|]\}}|r|kr|ndVqdS)rNrBr max_columnrBrCrascs$g|]\}}|kr|rdndqS)rrrBrrrBrCrfsz*Table._collapse_widths..)rrmaxrrrsr) rvrrr6 total_widthrsecond_max_columncolumn_differencer max_reducerBrrCrLs*     zTable._collapse_widths)r{ column_indexrr>c#st|j|j|jt}|dk|t|jdkitttttttfdfdd }g}|j }|j }|j r||j pd||j } || |j f||jpd} |jD]} || | fq|jr||jpd||j} || |jf|rBt} t|D]:\}}\}}t|| ||||t|ddp6|jVqn.|D](\}}t||t|ddpf|jVqFdS) z3Get all the cells with padding and optional header.rr) first_rowlast_rowr>cs||f}|r|S\}}}}rJs8td||}|sJtd||}snrVd}r^d}|rfd}|rnd}||||f}|||f<|S)Nr)getr)rrcachedr0rightbottomr.rm_padding_cacher] first_column last_columnr^r\rBrC get_paddings*   z%Table._get_cells..get_paddingr(r1N)r]r^r\rrsrUrMr rLror~r`r+r)r-rErar,r*rrrRgetattrr1)rAr{rr any_paddingr raw_cells_appendr~r+ cell_stylecellr,_Paddingfirstlastr-rSrBrrC _get_cellsssL,   zTable._get_cells)rr>cCsT|j\}}}}|jr$d}t||}|jsL|dkr6d}|t|jdkrLd}||S)zGet extra width from padding.rr)r\r]absr^rsrU)rArr pad_rightpad_leftrBrBrCrs zTable._get_padding_width)r{rrr>cCs|j}|dkrtddS||j}|jdurJt|j||j||Sg}g}|j}|j} tj} |||j|D]&} | ||| j \} } || | | qtt|rt |nd|rt |n||}| |j durdn|j ||jdurdn|j|}|S)z0Get the minimum and maximum width of the column.rrN) r6rrr:r4 with_maximumrorrrSrrr5)rAr{rrr6 padding_width min_widths max_widths append_min append_maxget_render_widthr_min_maxrrBrBrCrs<    zTable._measure_column)r{rrr>c% #s jp d}| jpd} fddt jD}tt|} jrj jj|t j j dnd}|r j s| n|}t } j} j } j} j} j} j}t |r>|j||j||j|f|j||j||j|f|j||j||j|fg}| rB|||V| Vng} j}jtt|D]\}\}}}|or| |o|| ss j|| nd}dg}srt !n|| r|dn|t||| D]^\}}}|j"||j#|j$|j%d|j&d}j'|j(||jd}t)t*||+|qt)dd|Dt,t,t d t-t t,t,t d fd d fd dt|||| D|dd<|r|r| r|j.|d| d|V| V||rdn |rdnd\}}}|j/0r|n|j/j1|j} t2D]R}!| r>|Vt3|D]$\}"}#|#|!EdH|"sF| VqF| rx|V| Vq.n0t2D]&}!|D]}#|#|!EdHq| Vq|r|r| r|j.|d| d|V| V|o|j4}$|rZ|s|s|$rZ|sZ| r&|t*|dksZ| r2sZ|rV|j.|d| d||Vn|j.|d| d|V| VqZ|r| r|5||V| VdS)Nr(c3s |]\}}||VqdSr)r)rrr)r{rArBrCrsz Table._render..)safer)r4r/r8r3rr9r-css|]}t|VqdSr)rs)rrrBrBrCrNrr)rr1r4r-r>csRr d}nrd}|dkr*|||S|dkrB|||S|||S)Nrr0middle) align_top align_middle align_bottom)rr1r4r-)_Segment footer_row header_row row_heightrBrC align_cellPsz!Table._render..align_cellc s8g|]0\}}}}||j||j|qSrB) set_shaper1r-)rr4_cellrr)rrr~ max_heightrrBrCras  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)%rAr{rr table_stylerf _column_cells row_cells_boxnew_linerUr`rarbrcrd box_segmentsrr|rrrow_cellr rEr4rrrlinesr.r_dividerdividerline_no last_cell rendered_cellrQrB) rrr{rr~rrrrrArCrs                    "        z Table._render)r(r(),rGrHrIrJrr'rKrPr HEAVY_HEADr rqr r rLBoxrMrrrru classmethodrwrOr_setterryrzrrrr r\rrrrrrrrRrrrrrBrBrBrCrTs*      @     ! = / 1 A' I ( rT__main__)r!)ReprHighlighter)timerz Table renderzStar Wars MovieszRich example tabler)rYrZrjZReleased bright_cyancyanT)r+r-r8ZTitlemagentarz 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,889N)rr>cCs"ttt|tdSr)r{printruler9)rrBrBrCr)sr)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'rPrRrTrGZ rich.consoleZrich.highlighterr2Z_timerr3tablerrrrqr)r9r8r_r4rerdrcrBrBrBrCs,        \