o iu@sddlZddlmZddlmZmZmZmZmZm Z m Z ddlm Z ddl m Z ddlmZddlmZdd lmZd d d d Z            d(de eefdeedeedede eee dfdededee ee de dfdededeedeed eed!efd"d#Z      d)dede e eee dfdedee e e defdedee d ee d!efd$d%Zd&d'ZdS)*N) zip_longest) CollectionDictIterableListOptionalSequenceUnion)cast)Literal)COLORS)color) supports_ansi<>^)lrcFautoWASABIdataheaderfooterdividerwidthsmax_colspacingaligns)rrr multiline env_prefix color_values fg_colors bg_colorsreturnc s| dus| dur/tt| dur| | dur"fdd| D} | dur/fdd| D} t|tr:t|}|rpg} t|D]+\}}dd|D}| tt|ddi|t |dkrm| t d d|DqB| }|d kr{t ||||}|||| | | d }t d d|Dfi|}g}|r| t |fi||r| |t|D]\}}| t |fi|q|r|r| || t |fi|d d|S)aFormat tabular data. data (Union[Collection, Dict]): The data to render. Either a list of lists (one per row) or a dict for two-column tables. header (Optional[Iterable]): Optional header columns. footer (Optional[Iterable]): Optional footer columns. divider (bool): Show a divider line between header/footer and body. widths (Union[Iterable[int], Literal['auto']]): Column widths in order. If "auto", widths will be calculated automatically based on the largest value. max_col (int): Maximum column width. spacing (int): Spacing between columns, in spaces. aligns (Optional[Union[Iterable[str], str]]): Optional column alignments in order. 'l' (left, default), 'r' (right) or 'c' (center). If a string, value is used for all columns. multiline (bool): If a cell value is a list of a tuple, render it on multiple lines, with one value per line. env_prefix (str): Prefix for environment variables, e.g. WASABI_LOG_FRIENDLY. color_values (Optional[Dict]): Optional color values to add or overwrite, name mapped to value. fg_colors (Optional[Iterable]): Optional foreground colors, one per column. None can be specified for individual columns to retain the default foreground color. bg_colors (Optional[Iterable]): Optional background colors, one per column. None can be specified for individual columns to retain the default background color. RETURNS (str): The formatted table. Ncg|]}||qSget).0fg_colorcolorsr)A/home/ubuntu/.local/lib/python3.10/site-packages/wasabi/tables.py =ztable..cr(r)r*)r,bg_colorr.r)r0r1?r2cSs$g|]}t|ttfr |n|gqSr)) isinstancelisttuple)r,vr)r)r0r1Es$ fillvaluer cSsg|]}dqS)r9r))r,ir)r)r0r1Hr)rr r!r#r%r&cSsg|]}d|qS)-r))r,widthr)r)r0r1Tsz {}  )dictr updater4r5items enumerateextendrlenappendr6_get_max_widthsrowformatjoin)rrrrrrr r!r"r#r$r%r& zipped_datar:itemvalssettings divider_rowrowsr)r.r0tablesR*     rPcsTtd|d}to| o|dup|du}g} ttr'fdd|Dnttt} tdsIdkrIttt fddt t |Dt |D]U\} } t | r_| t | kr_| | nd } dkrjt | nttt | }d | |f}|t| } |r|dur|| nd}|dur|| nd}t| ||d } | | qMd || S) a;Format data as a table row. data (Collection): The individual columns to format. widths (Union[Sequence[int], int, Literal['auto']]): Column widths, either one integer for all columns or an iterable of values. If "auto", widths will be calculated automatically based on the largest value. spacing (int): Spacing between columns, in spaces. aligns (Optional[Union[Sequence[Literal['r', 'c', 'l']], str]]): Optional column alignments in order. 'l' (left, default), 'r' (right) or 'c' (center). If a string, value is used for all columns. env_prefix (str): Prefix for environment variables, e.g. WASABI_LOG_FRIENDLY. fg_colors (Optional[Sequence]): Optional foreground colors for the columns, in order. None can be specified for individual columns to retain the default foreground color. bg_colors (Optional[Sequence]): Optional background colors for the columns, in order. None can be specified for individual columns to retain the default background color. RETURNS (str): The formatted row. z{}_LOG_FRIENDLYFNcg|]}qSr)r)r,_)r!r)r0r1r;zrow..__iter__rcrQr)r)rR)rr)r0r1r;rz{:%s%d})fgbg )osgetenvrHrr4strr rhasattrintrangerDrB ALIGN_MAPr+_colorrErI)rrr r!r#r%r&env_log_friendly show_colorscols_alignsr:colalign col_widthtplrUrVr))r!rr0rGcs,($""  rGcsLt|}|r |||r||dd|D}fddtt|DS)NcSsg|] }dd|DqS)cSsg|]}tt|qSr))rDrZ)r,rdr)r)r0r1r2z._get_max_widths...r))r,rKr)r)r0r1z#_get_max_widths..csg|] }tt|qSr))minmax)r,wrr)r0r1rh)r5rEzip)rrrrall_datarr)rlr0rFs  rF) NNFrrrNFrNNN)rrNrNN)rX itertoolsrtypingrrrrrrr r compatr utilr rr_rr^boolr\rZrPrGrFr)r)r)r0s $             W 4