File: //opt/saltstack/salt/lib/python3.10/site-packages/rich/__pycache__/cells.cpython-310.pyc
o
;j/0 � @ s& U d dl mZ d dlmZ d dlmZ d dlmZmZm Z m
Z
d dl m Z
e
eeef Zed�Zg d�Zded < ed
d � eD ��ZejZd ed
< G dd� de�Zedd�d2d3dd��Zed�d2d4dd ��Zd2d4d d �Zd4d d �Z d2d5d"d#�Z d2d6d&d'�Z d2d6d(d)�Z d2d7d+d,�Z d2d8d/d0�Z d1S )9� )� annotations)� lru_cache)�
itemgetter)�Callable�
NamedTuple�Sequence�Tuple)�load� ))� �~ )� � )� i� )ip i� )i % i�% )i ( i�( zlist[tuple[int, int]]�_SINGLE_CELL_UNICODE_RANGESc C s. g | ]\}}t tt||d ��D ]}|�qqS )� )�map�chr�range)�.0Z_start�_end� character� r �>/opt/saltstack/salt/lib/python3.10/site-packages/rich/cells.py�
<listcomp> s ���r zCallable[[str], bool]�_is_single_cell_widthsc @ s* e Zd ZU dZded< ded< ded< dS ) � CellTablezDContains unicode data required to measure the cell widths of glyphs.�str�unicode_versionz Sequence[tuple[int, int, int]]�widthszfrozenset[str]�narrow_to_wideN)�__name__�
__module__� __qualname__�__doc__�__annotations__r r r r r & s
r i )�maxsize�autor r r �return�intc C s� t | �}|r
|dk sd| krdk rdS ndS t|�j}|d }||d kr*dS d}t|�d }||kr[|| d? }|| \}} }
||k rL|d }n || krU|d }n|
S ||ks6dS )a Get the cell size of a character.
Args:
character (str): A single character.
unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version.
Returns:
int: Number of cells (0, 1 or 2) occupied by that character.
r � r
r ���r )�ord�load_cell_tabler �len) r r Z codepoint�tableZ
last_entryZ lower_boundZ upper_bound�index�start�end�widthr r r �get_character_cell_size. s* �
� r4 �textc C s
t | |�S )a� Get the number of cells required to display text.
This method always caches, which may use up a lot of memory. It is recommended to use
`cell_len` over this method.
Args:
text (str): Text to display.
unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version.
Returns:
int: Get the number of cells required to display text.
)� _cell_len�r5 r r r r �cached_cell_lenQ s
r8 c C s t | �dk r t| |�S t| |�S )� Get the cell length of a string (length as it appears in the terminal).
Args:
text: String to measure.
unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version.
Returns:
Length of string in terminal cells.
i )r. r8 r6 r7 r r r �cell_lenb s
r: c
s� t | �rt| �S d| vrd| vrt� fdd�| D ��S t� �}d}d}ddh}d}t| �}||k rc| | }||v rN|dkrB|d7 }n|rM|||jv 7 }d}n
t|� � } r[|}|| 7 }|d7 }||k s1|S )r9 � � ️c 3 s � | ]}t |� �V qd S )N)r4 )r r �r r r � <genexpr>� s �
�z _cell_len.<locals>.<genexpr>r Nr )r r. �sumr- r r4 )
r5 r �
cell_table� total_width�last_measured_character�SPECIALr0 Zcharacter_countr �character_widthr r= r r6 q s6 �
��r6 � 'tuple[list[CellSpan], int]'c C s� t |�}t| �}d}d}d}g }ddh}||k r�| | } |v r�|s/|�||d }df� q| dkrN|||d k r<dnd7 }|d \}
} } |
|| f|d< n2|d7 }|rr|d \}
} } ||jv rjd}| d7 } |d7 }|
|| f|d< n|d \}
} } |
|| f|d< qt| |� }
r�| }|�||d }|
f� ||
7 }n!|r�|d \}
} } |
|d }| f|d< n |�||d }df� ||k s||fS )aT Divide text into spans that define a single grapheme, and additionally return the cell length of the whole string.
The returned spans will cover every index in the string, with no gaps. It is possible for some graphemes to have a cell length of zero.
This can occur for nonsense strings like two zero width joiners, or for control codes that don't contribute to the grapheme size.
Args:
text: String to split.
unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version.
Returns:
A tuple of a list of *spans* and the cell length of the entire string. A span is a list of tuples
of three values consisting of (<START>, <END>, <CELL LENGTH>), where START and END are string indices,
and CELL LENGTH is the cell length of the single grapheme.
r Nr; r< r r
r+ )r- r. �appendr r4 )r5 r r@ Zcodepoint_countr0 rB rA �spansrC r r1 r � cell_lengthrD r r r �split_graphemes� sJ
�-rI �
cell_position�tuple[str, str]c C s. |dkrd| fS t | |�\}}t|| t|� �}ttt|d|� ��} ||krE|t|�kr3| dfS || d }| d|� | |d� fS ||k ro|| \}} }
||
|krf| d|� d d| | d� fS |d7 }||
7 }n'||d \}} }
||
|k r�| d|� d d| | d� fS |d8 }||
8 }q%)�p Split text by cell position.
If the cell position falls within a double width character, it is converted to two spaces.
Args:
text: Text to split.
cell_position Offset in cells.
unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version.
Returns:
Tuple to two split strings.
r � NT� r )rI r) r. r? r �_span_get_cell_len) r5 rJ r rG rH �offsetZ left_sizeZ split_indexr1 r2 � cell_sizer r r � _split_text� s.
�rR c C s, t | �r| d|� | |d� fS t| ||�S )rL N)r rR )r5 rJ r r r r �
split_text s rS �totalc C s� t | �rt| �}||k r| d|| S | d|� S |dkr dS t| �}||kr*| S ||k r6| d|| S t| ||�\} }| S )a$ Adjust a string by cropping or padding with spaces such that it fits within the given number of cells.
Args:
text: String to adjust.
total: Desired size in cells.
unicode_version: Unicode version.
Returns:
A string with cell size equal to total.
rN Nr rM )r r. r: rR )r5 rT r �sizerQ �_r r r �
set_cell_size+ s rW r3 � list[str]c s� t � �r� �fdd�tdt� ���D �S t� |�\}}d}g }d}|D ] \}} }
||
�kr;|�� ||� � |}d}||
7 }q#|rK|�� |d� � |S )ah Split text into lines such that each line fits within the available (cell) width.
Args:
text: The text to fold such that it fits in the given width.
width: The width available (number of cells).
Returns:
A list of strings such that each string in the list has cell width
less than or equal to the available width.
c s g | ]
}� ||� � �qS r r )r r0 �r5 r3 r r r R s z chop_cells.<locals>.<listcomp>r N)r r r. rI rF ) r5 r3 r rG rV Z line_size�linesZ line_offsetr1 r2 rQ r rY r �
chop_cellsF s
r[ N)r' )r r r r r( r) )r5 r r r r( r) )r5 r r r r( rE )r5 r rJ r) r r r( rK )r5 r rT r) r r r( r )r5 r r3 r) r r r( rX )!�
__future__r � functoolsr �operatorr �typingr r r r Zrich._unicode_datar r- r) ZCellSpanrO r r% � frozensetZ
_SINGLE_CELLS�
issupersetr r r4 r8 r: r6 rI rR rS rW r[ r r r r �<module> s:
��
"
1 �K �- �