o =Ѯa @sZdZddlZddlZddlmZedZddZddd Zdd d Z d d Z ddZ dS)z6Functions that handle alignment, padding, widths, etc.N) terminal_sizez(\033\[[\d;]+m)c Csld|vr td|}z|d}Wn ttfyYnwd}|D]}t|dvr/|d7}q!|d7}q!|S)a(Get the visible width of a unicode string. Some CJK unicode characters are more than one byte unlike ASCII and latin unicode characters. From: https://github.com/Robpol86/terminaltables/pull/9 :param str string: String to measure. :return: String's width. :rtype: int u8r)FW) RE_COLOR_ANSIsubdecodeAttributeErrorUnicodeEncodeError unicodedataZeast_asian_width)stringwidthcharrFusr/lib/python3.10/site-packages/terminaltables/width_and_alignment.py visible_width s   r c Cst|ds t|}|pdg}|dr|dd|vr7dg|dt||d|dg|d}nAd|vr`|dt|}dg|d|d|d|dg|d|d}ndg|d|dg|dt||d}t|D]R\}}|d t|t|} d |vr||d | |||d||<q|d |vr||d | | |||d||<q|||d | | |d|||<q||S) aTAlign a string horizontally and vertically. Also add additional padding in both dimensions. :param str string: Input string to operate on. :param tuple align: Tuple that contains one of left/center/right and/or top/middle/bottom. :param tuple inner_dimensions: Width and height ints to expand string to without padding. :param iter padding: Number of space chars for left, right, top, and bottom (4 ints). :param str space: Character to use as white space for resizing/padding (use single visible chars only). :return: Padded cell split into lines. :rtype: list splitlinesr Zbottomr rmiddlerrightcenter) hasattrstrrendswithappendlen enumeraterrjustrljust) rZalignZinner_dimensionspaddingspacelinesdeltailineZ new_widthrrralign_and_pad_cell*s&   2:0&*&r+c sdg|r tdd|Dnd}dgt|}t|D]?\}}t|D]6\} } t| dr1t| ds5t| } | s8q#t||| dd||<t|| gdd | DR|| <q#qfd d |D} fd d |D} ||| | fS) aDGet maximum widths of each column and maximum height of each row. :param iter table_data: List of list of strings (unmodified table data). :param int padding_left: Number of space chars on left side of cell. :param int padding_right: Number of space chars on right side of cell. :param int padding_top: Number of empty lines on top side of cell. :param int padding_bottom: Number of empty lines on bottom side of cell. :return: 4-item tuple of n-item lists. Inner column widths and row heights, outer column widths and row heights. :rtype: tuple rcss|]}t|VqdS)N)r!).0rrrr `sz!max_dimensions..countrrr cSsg|]}t|qSr)r)r,lrrr ksz"max_dimensions..csg|]}|qSrrr,r)) padding_left padding_rightrrr1ncsg|]}|qSrrr2)padding_bottom padding_toprrr1or5)maxr!r"rrr/r) Z table_datar3r4r7r6 inner_widthsZ inner_heightsjrowr)cell outer_widthsZ outer_heightsr)r6r3r4r7rmax_dimensionsTs (  r>c CsNt|}td}|}|||d7}|||7}t|||}|||S)a Determine the maximum width of a column based on the current terminal width. :param iter inner_widths: List of widths (no padding) for each column. :param int column_number: The column number to query. :param int outer_border: Sum of left and right outer border visible widths. :param int inner_border: Visible width of the inner border character. :param int padding: Total padding per cell (left + right padding). :return: The maximum width the column can be without causing line wrapping. rr )r!rsum) r9Z column_number outer_border inner_borderr% column_countZterminal_widthnon_data_space data_spacerrrcolumn_max_widthts   rEcCs0t|}|}|r|||d7}t|}||S)aiDetermine the width of the entire table including borders and padding. :param iter outer_widths: List of widths (with padding) for each column. :param int outer_border: Sum of left and right outer border visible widths. :param int inner_border: Visible width of the inner border character. :return: The width of the table. :rtype: int r )r!r?)r=r@rArBrCrDrrr table_widths  rF)r)rrrr) __doc__rerZterminaltables.terminal_iorcompiler rr+r>rErFrrrrs    *