Your IP : 216.73.216.213
U
ʗRez� � @ sB U d dl Z d dlmZmZ d dlmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lm Z m Z dd l m Z dd
l m!Z! ddl"m#Z# ddl$m%Z% ddl&m'Z' ddl(m)Z) ddl*m+Z+m,Z, e �rddl-m.Z.m/Z/m0Z0m1Z1 dZ2de3d< dZ4de3d< e �5d�Z6ee7df Z8e e7gee, f Z9G d d � d e�Z:G d d� de%�Z;e<d k�r>d d l=m.Z. e;d!�Z>e>�?d"gd#� e>�?d$gd%� e.� Z-e-�@d&� e-jAe>d'd(� e-�A� e-�@d)� e-jAe>d*d+d,� e-�A� e-�@d-� e-jAe>d.d/d,� e-�A� e-�@d0� e-jAe>d1d2d,� e-�A� dS )3� N)�partial�reduce)�gcd)�
itemgetter)
�
TYPE_CHECKING�Any�Callable�Dict�Iterable�List�
NamedTuple�Optional�Tuple�Union� )� loop_last)� pick_bool)� divide_line)� AlignMethod)�cell_len�
set_cell_size)�Lines)�strip_control_codes)� EmojiVariant)� JupyterMixin)� Measurement)�Segment)�Style� StyleType)�Console�ConsoleOptions�
JustifyMethod�OverflowMethod�defaultr! �DEFAULT_JUSTIFY�foldr" �DEFAULT_OVERFLOWz\s+$�Textc @ s� e Zd ZU dZeed< eed< eeef ed< ed�dd�Z e
d�dd �Z ee d e
d f d
�d d �Zed d
�d
d�Zed d
�dd�ZdS )�Spanz A marked up region in some text.�start�end�style��returnc C sT t | jt�r2| jjr2d| j� d| j� d| j�d�S d| j� d| j� dt| j�� d�S )NzSpan(z, �))�
isinstancer+ r �_metar) r* �repr��self� r4 ��/builddir/build/BUILDROOT/alt-python38-pip-22.2.1-2.el8.x86_64/opt/alt/python38/lib/python3.8/site-packages/pip/_vendor/rich/text.py�__repr__7 s
�� �z
Span.__repr__c C s | j | jkS �N)r* r) r2 r4 r4 r5 �__bool__> s z
Span.__bool__��offsetr- c C sV || j k r| dfS || jkr$| dfS | \}}}t|t||�|�}t|j||�}||fS )z)Split a span in to 2 from a given offset.N)r) r* r( �min)r3 r: r) r* r+ �span1�span2r4 r4 r5 �splitA s
z
Span.splitc C s | \}}}t || || |�S )z�Move start and end by a given offset.
Args:
offset (int): Number of characters to add to start and end.
Returns:
TextSpan: A new TextSpan with adjusted position.
)r( �r3 r: r) r* r+ r4 r4 r5 �moveN s
z Span.movec C s( | \}}}||kr| S t |t||�|�S )z�Crop the span at the given offset.
Args:
offset (int): A value between start and end.
Returns:
Span: A new (possibly smaller) span.
)r( r; r? r4 r4 r5 �
right_cropZ s
zSpan.right_cropN)�__name__�
__module__� __qualname__�__doc__�int�__annotations__r �strr r6 �boolr8 r r
r> r@ rA r4 r4 r4 r5 r( - s
r( c
@ s� e Zd ZdZdddddddd d
g Zd�d d d d
dd d�eeeef ed ed ee eee
ee e d d� dd�Z
e
d�dd�Ze d�dd�Zed�dd�Zed�d d �Zed d �d d �Zee d �d!d"�Zee d �d#d$�Zee
ef d d%�d&d'�Zee
d�d(d)��Zeed�d*d+��Ze d d,d d d d
d-�eeeef e ee ed ed ed d.�d/d0��Z e d d d d d
dd1�eeeef ed ed ee eee
d d2�d3d4��Z e d�d d d5�ee ed ed d d6�d7d8��Z!e d d d d d
dd d9�eed e"ee f f eeef ed ed ee ee
ee#eef d d:� d;d<��Z$eed�d=d>��Z%e%j&ed d?�d@d>��Z%ee e d�dAdB��Z'e'j&e e d dC�dDdB��Z'd�ed dE�dFdG�Z(d d�dHdI�Z)d�eeef e
ee
d dK�dLdM�Z*d�e#eef e
ee
d dN�dOdP�Z+d�ee#eef ed dQ�dRdS�Z,ed dT�dUdV�Z-dWe
edX�dYdZ�Z.d�d d[�eeee/e f ee
d\�d]d^�Z0d,d_�e1e eeef e e
d`�dadb�Z2d d�dcdd�Z3e
d de�dfdg�Z4e
d dh�didj�Z5dWdke1e6 dl�dmdn�Z7dWdke8dl�dodp�Z9d�dWee1dq dr�dsdt�Z:e1d d du�dvdw�Z;d�ee
d dx�dydz�Z<d d{d|�e
ed e d d}�d~d�Z=d d�d�d��Z>d�e
ed d��d�d��Z?d�e
ed d��d�d��Z@d�e
ed d��d�d��ZAd�eBe
ed d��d�d��ZCd�ed ef eeed�f d d��d�d��ZDd d d��d�d��ZEe1e"eee f d d��d�d��ZFd d d��d�d��ZGd�d{d{d��ee e eHd��d�d��ZIe1e
eHd��d�d��ZJd�e
d d��d�d��ZKd d dd d��dWe
ed ed e
ee eHd��d�d��ZLe
eHd��d�d��ZMe
d�d�d��ZNd�d�d�d��ee
ee d d��d�d��ZOd S )�r' a� Text with color / style.
Args:
text (str, optional): Default unstyled text. Defaults to "".
style (Union[str, Style], optional): Base style for text. Defaults to "".
justify (str, optional): Justify method: "left", "center", "full", "right". Defaults to None.
overflow (str, optional): Overflow method: "crop", "fold", "ellipsis". Defaults to None.
no_wrap (bool, optional): Disable text wrapping, or None for default. Defaults to None.
end (str, optional): Character to end text with. Defaults to "\\n".
tab_size (int): Number of spaces per tab, or ``None`` to use ``console.tab_size``. Defaults to 8.
spans (List[Span], optional). A list of predefined style spans. Defaults to None.
�_textr+ �justify�overflow�no_wrapr* �tab_size�_spans�_length� N�
� )rK rL rM r* rN �spansr! r" ) �textr+ rK rL rM r* rN rT r- c
C sL t |�} | g| _|| _|| _|| _|| _|| _|| _|p:g | _t | �| _
d S r7 ) r rJ r+ rK rL rM r* rN rO �lenrP )
r3 rU r+ rK rL rM r* rN rT �sanitized_textr4 r4 r5 �__init__� s
z
Text.__init__r, c C s | j S r7 )rP r2 r4 r4 r5 �__len__� s z Text.__len__c C s
t | j�S r7 )rI rP r2 r4 r4 r5 r8 � s z
Text.__bool__c C s | j S r7 ��plainr2 r4 r4 r5 �__str__� s z Text.__str__c C s d| j �d| j�d�S )Nz<text � �>)r[ rO r2 r4 r4 r5 r6 � s z
Text.__repr__)�otherr- c C s( t |ttf�r$| �� }|�|� |S tS r7 )r/ rH r' �copy�append�NotImplemented)r3 r_ �resultr4 r4 r5 �__add__� s
z Text.__add__c C s& t |t�stS | j|jko$| j|jkS r7 )r/ r' rb r[ rO �r3 r_ r4 r4 r5 �__eq__� s
z Text.__eq__c C s. t |t�r|| jkS t |t�r*|j| jkS dS )NF)r/ rH r[ r' re r4 r4 r5 � __contains__� s
zText.__contains__)�slicer- c sf t dd�� fdd� }t|t �r&||�S |�t� j��\}}}|dkrZ� �||g�}|d S td��d S )Nr' r9 c s. t � t�j� � �fdd��jD �dd�}|S )Nc s6 g | ].\}}}|� kr"|krn q� d d|��qS )r r r4 ��.0r) r* r+ ��_Spanr: r4 r5 �
<listcomp>� s
�z9Text.__getitem__.<locals>.get_text_at.<locals>.<listcomp>rQ )rT r* )r( r' r[ rO )r: rU r2 rk r5 � get_text_at� s �� z%Text.__getitem__.<locals>.get_text_atr z%slices with step!=1 are not supported)rF r/ �indicesrV r[ �divide� TypeError)r3 rh rn r) �stop�step�linesr4 r2 r5 � __getitem__� s
zText.__getitem__c C s
t | j�S )z5Get the number of cells required to render this text.)r r[ r2 r4 r4 r5 r � s z
Text.cell_lenc C s� ddl m} g }| j}dd| jffdd� | jD �dd� | jD �t|�d| jff�}|jtdd�d � d}|j}|D ]N\}}} ||kr�|||||� �� |}| rr||r�d
| � d �n
d | � d �� qrd
� |�}
|
S )z~Get console markup to render this Text.
Returns:
str: A string potentially creating markup tags.
r )�escaper Fc s s | ]}|j d |jfV qdS �FN)r) r+ �rj �spanr4 r4 r5 � <genexpr>� s z Text.markup.<locals>.<genexpr>c s s | ]}|j d |jfV qdS �TN)r* r+ rx r4 r4 r5 rz � s T��keyz[/�]�[rQ )
�markuprv r[ r+ rO rV �sortr ra �join) r3 rv �outputr[ � markup_spans�positionra r: �closingr+ r� r4 r4 r5 r� � s(
��"
z Text.markupT)r+ �emoji�
emoji_variantrK rL r* )rU r+ r� r� rK rL r* r- c
C s2 ddl m} |||||d�} || _|| _|| _| S )a, Create Text instance from markup.
Args:
text (str): A string containing console markup.
emoji (bool, optional): Also render emoji code. Defaults to True.
justify (str, optional): Justify method: "left", "center", "full", "right". Defaults to None.
overflow (str, optional): Overflow method: "crop", "fold", "ellipsis". Defaults to None.
end (str, optional): Character to end text with. Defaults to "\\n".
Returns:
Text: A Text instance with markup rendered.
r )�render)r� r� )r� r� rK rL r* )
�clsrU r+ r� r� rK rL r* r� �
rendered_textr4 r4 r5 � from_markup� s zText.from_markup�r+ rK rL rM r* rN )rU r+ rK rL rM r* rN r- c C sF ddl m} td||||||d�} |� }
| �dd� |
�|�D ��} | S )a� Create a Text object from a string containing ANSI escape codes.
Args:
text (str): A string containing escape codes.
style (Union[str, Style], optional): Base style for text. Defaults to "".
justify (str, optional): Justify method: "left", "center", "full", "right". Defaults to None.
overflow (str, optional): Overflow method: "crop", "fold", "ellipsis". Defaults to None.
no_wrap (bool, optional): Disable text wrapping, or None for default. Defaults to None.
end (str, optional): Character to end text with. Defaults to "\\n".
tab_size (int): Number of spaces per tab, or ``None`` to use ``console.tab_size``. Defaults to 8.
r )� AnsiDecoderrR )rK rL rM r* rN r+ c s s | ]
}|V qd S r7 r4 �rj �liner4 r4 r5 rz 9 s z!Text.from_ansi.<locals>.<genexpr>)�ansir� r' r� �decode) r� rU r+ rK rL rM r* rN r� �joiner�decoderrc r4 r4 r5 � from_ansi s � zText.from_ansi�rK rL )rU r+ rK rL r- c C s | |||d�}|� |� |S )aa Construct a Text instance with a pre-applied styled. A style applied in this way won't be used
to pad the text when it is justified.
Args:
text (str): A string containing console markup.
style (Union[str, Style]): Style to apply to the text. Defaults to "".
justify (str, optional): Justify method: "left", "center", "full", "right". Defaults to None.
overflow (str, optional): Overflow method: "crop", "fold", "ellipsis". Defaults to None.
Returns:
Text: A text instance with a style applied to the entire string.
r� )�stylize)r� rU r+ rK rL � styled_textr4 r4 r5 �styled< s
z Text.styled)r+ rK rL rM r* rN �meta) �partsr+ rK rL rM r* rN r� r- c
G sZ | ||||||d�} | j }
t} |D ]$} t| | tf�r>|
| � q"|
| � q"|rV| �|� | S )aR Construct a text instance by combining a sequence of strings with optional styles.
The positional arguments should be either strings, or a tuple of string + style.
Args:
style (Union[str, Style], optional): Base style for text. Defaults to "".
justify (str, optional): Justify method: "left", "center", "full", "right". Defaults to None.
overflow (str, optional): Overflow method: "crop", "fold", "ellipsis". Defaults to None.
end (str, optional): Character to end text with. Defaults to "\\n".
tab_size (int): Number of spaces per tab, or ``None`` to use ``console.tab_size``. Defaults to 8.
meta (Dict[str, Any], optional). Meta data to apply to text, or None for no meta data. Default to None
Returns:
Text: A new text instance.
r� )ra r' r/ rH �
apply_meta)
r� r+ rK rL rM r* rN r� r� rU ra �_Text�partr4 r4 r5 �assembleU s" �
z
Text.assemblec C s0 t | j�dkr&d�| j�g| jdd�< | jd S )z Get the text as a single string.r rQ Nr )rV rJ r� r2 r4 r4 r5 r[ � s z
Text.plain)�new_textr- c C sH || j krDt|�}|g| jdd�< | j}t|�| _|| jkrD| �� dS )z Set the text to a new value.N)r[ r rJ rP rV � _trim_spans)r3 r� rW �
old_lengthr4 r4 r5 r[ � s
c C s | j S )z.Get a reference to the internal list of spans.�rO r2 r4 r4 r5 rT � s z
Text.spans)rT r- c C s |dd� | _ dS )z
Set spans.Nr� )r3 rT r4 r4 r5 rT � s )r[ r- c C s& t || j| j| j| j| j| jd�}|S )zOReturn a new Text instance with copied meta data (but not the string or spans).r� )r' r+ rK rL rM r* rN )r3 r[ � copy_selfr4 r4 r5 �
blank_copy� s � zText.blank_copyc C s8 t | j| j| j| j| j| j| jd�}| j|jdd�< |S )z Return a copy of this instance.r� N) r' r[ r+ rK rL rM r* rN rO )r3 r� r4 r4 r5 r` � s � z Text.copyr )r+ r) r* r- c C sj |rft | �}|dk r || }|dkr(|}|dk r8|| }||ksH||krLdS | j�t|t||�|�� dS )a{ Apply a style to the text, or a portion of the text.
Args:
style (Union[str, Style]): Style instance or style definition to apply.
start (int): Start offset (negative indexing is supported). Defaults to 0.
end (Optional[int], optional): End offset (negative indexing is supported), or None for end of text. Defaults to None.
r N)rV rO ra r( r; )r3 r+ r) r* �lengthr4 r4 r5 r� � s z Text.stylize)r� r) r* r- c C s t �|�}| j|||d� dS )ah Apply meta data to the text, or a portion of the text.
Args:
meta (Dict[str, Any]): A dict of meta information.
start (int): Start offset (negative indexing is supported). Defaults to 0.
end (Optional[int], optional): End offset (negative indexing is supported), or None for end of text. Defaults to None.
)r) r* N)r � from_metar� )r3 r� r) r* r+ r4 r4 r5 r� � s
zText.apply_meta)r� �handlersr- c K s< |dkr i n|}|� dd� |�� D �� | �t�|�� | S )a� Apply event handlers (used by Textual project).
Example:
>>> from rich.text import Text
>>> text = Text("hello world")
>>> text.on(click="view.toggle('world')")
Args:
meta (Dict[str, Any]): Mapping of meta information.
**handlers: Keyword args are prefixed with "@" to defined handlers.
Returns:
Text: Self is returned to method may be chained.
Nc S s i | ]\}}d |� �|�qS )�@r4 )rj r} �valuer4 r4 r5 �
<dictcomp>� s zText.on.<locals>.<dictcomp>)�update�itemsr� r r� )r3 r� r� r4 r4 r5 �on� s zText.on)�suffixr- c C s | j �|�r| �t|�� dS )zaRemove a suffix if it exists.
Args:
suffix (str): Suffix to remove.
N)r[ �endswithrA rV )r3 r� r4 r4 r5 �
remove_suffix� s zText.remove_suffixr )�consoler: r- c C sf |dk rt | �| }|j}|| j��� }| jD ]2\}}}|| krL|kr.n q.|||dd�7 }q.|S )a
Get the style of a character at give offset.
Args:
console (~Console): Console where text will be rendered.
offset (int): Offset in to text (negative indexing supported)
Returns:
Style: A Style instance.
r rQ �r# )rV � get_styler+ r` rO )r3 r� r: r� r+ r) r* �
span_styler4 r4 r5 �get_style_at_offset� s zText.get_style_at_offset)� style_prefix)� re_highlightr+ r� r- c C s� d}| j j}t}| j}t�||�D ]�}|j} |rv| � \}
} t|�rR|||
| � �n|} | dk rv| |
krv|||
| | �� |d7 }|�� � � D ]8}
| |
�\}
} |
dkr�| |
kr�|||
| |� |
� ��� q�q"|S )a Highlight text with a regular expression, where group names are
translated to styles.
Args:
re_highlight (str): A regular expression.
style (Union[GetStyleCallable, StyleType]): Optional style to apply to whole match, or a callable
which accepts the matched text and returns a style. Defaults to None.
style_prefix (str, optional): Optional prefix to add to style group names.
Returns:
int: Number of regex matches
r Nr ���)
rO ra r( r[ �re�finditerry �callable� groupdict�keys)r3 r� r+ r� �count� append_spanrl r[ �match�get_spanr) r* � match_style�namer4 r4 r5 �highlight_regex s"
zText.highlight_regex)�case_sensitive)�wordsr+ r� r- c C sp d� dd� |D ��}| jj}d}t}tj|| j|r6dntjd�D ]*}|�d�\} }
||| |
|�� |d7 }q@|S )aD Highlight words with a style.
Args:
words (Iterable[str]): Worlds to highlight.
style (Union[str, Style]): Style to apply.
case_sensitive (bool, optional): Enable case sensitive matchings. Defaults to True.
Returns:
int: Number of words highlighted.
�|c s s | ]}t �|�V qd S r7 )r� rv �rj �wordr4 r4 r5 rz J s z'Text.highlight_words.<locals>.<genexpr>r ��flagsr ) r� rO ra r( r� r� r[ �
IGNORECASEry ) r3 r� r+ r� �re_words�add_spanr� rl r� r) r* r4 r4 r5 �highlight_words9 s �
zText.highlight_wordsc C s | j �� | _ dS )z"Strip whitespace from end of text.N)r[ �rstripr2 r4 r4 r5 r� V s z Text.rstrip)�sizer- c C sN t | �}||krJ|| }t�| j�}|dk rJt |�d��}| �t||�� dS )z�Remove whitespace beyond a certain width at the end of the text.
Args:
size (int): The desired size of the text.
Nr )rV �_re_whitespace�searchr[ �grouprA r; )r3 r� � text_length�excess�whitespace_match�whitespace_countr4 r4 r5 �
rstrip_endZ s zText.rstrip_end)�
new_lengthr- c C s: t | �}||kr6||k r(| �|| � n| �|| � dS )z<Set new length of the text, clipping or padding is required.N)rV � pad_rightrA )r3 r� r� r4 r4 r5 �
set_lengthh s
zText.set_lengthr )r� �optionsr- c
c s� |j p| j pd}| jp |jp t}| jp.|jp.t}| j||j|||pDdt| j|jd�d�}t d��
|�}|j || j d�E d H d S )NrS F�rK rL rN rM rR �r* )
rN rK r$ rL r& �wrap� max_widthr rM r' r� r� r* )r3 r� r� rN rK rL rt � all_linesr4 r4 r5 �__rich_console__q s �zText.__rich_console__c C sT | j }|�� }|r$tdd� |D ��nd}|�� }|rFtdd� |D ��n|}t||�S )Nc s s | ]}t |�V qd S r7 �r r� r4 r4 r5 rz � s z(Text.__rich_measure__.<locals>.<genexpr>r c s s | ]}t |�V qd S r7 r� r� r4 r4 r5 rz � s )r[ �
splitlines�maxr> r )r3 r� r� rU rt �max_text_widthr� �min_text_widthr4 r4 r5 �__rich_measure__� s �zText.__rich_measure__r )r� r* r- c # sX t }| j}| js,t |�V |r(||�V dS t|jt�� d��tt| jd��}�fdd�|D ���| j ��d< ddd � |D �d
d � |D �t
|�d dff�}|j t dd�d � g ��j
}�j}i ��j�tj� td
�� ����fdd� } t||dd� �D ]J\\}
} } \}
}}| �r || � n|| � |
|
kr�|||
|
� | � �V q�|�rT||�V dS )a Render the text as Segments.
Args:
console (Console): Console instance.
end (Optional[str], optional): Optional end character.
Returns:
Iterable[Segment]: Result of render that may be written to the console.
Nr� r c s i | ]\}}|� |j ��qS r4 �r+ �rj �indexry )r� r4 r5 r� � s z Text.render.<locals>.<dictcomp>r �r Fr c s s | ]\}}|j d |fV qdS rw )r) r� r4 r4 r5 rz � s z Text.render.<locals>.<genexpr>c s s | ]\}}|j d |fV qdS r{ r� r� r4 r4 r5 rz � s Tr| r, c sB t �fdd�t��D ��} �| �}|dk r.|S � | �}|�| < |S )z#Construct current style from stack.c 3 s | ]}� | V qd S r7 r4 )rj � _style_id)� style_mapr4 r5 rz � s z9Text.render.<locals>.get_current_style.<locals>.<genexpr>N)�tuple�sorted)�styles� cached_style�
current_style)�combine�stack� style_cache�style_cache_getr� r4 r5 �get_current_style� s z&Text.render.<locals>.get_current_style)r� )r r[ rO r r� r �null�list� enumerater+ rV r� r ra �remove�getr� �zip)r3 r� r* �_SegmentrU �enumerated_spansrT � stack_append� stack_popr� r: �leaving�style_id� next_offset�_r4 )r� r� r� r� r� r� r5 r� � sB
� �
&
z Text.render)rt r- c s� �� � }td d���fdd� }|jj}|jj}|jj}d�t� |� D ]T}||j� |jrt|� ��t|� |j�� |� �fdd�|jD �� �t|�7 �qF�|_ |S )z�Join text together with this instance as the separator.
Args:
lines (Iterable[Text]): An iterable of Text instances to join.
Returns:
Text: A new text instance containing join text.
r' r, c 3 s8 �j r*t� �D ]\} }|V | s�V qn
� E d H d S r7 )r[ r )�lastr� )rt r3 r4 r5 � iter_text� s
z Text.join.<locals>.iter_textr c 3 s( | ] \}}}� �| �| |�V qd S r7 r4 ri rk r4 r5 rz � s �z Text.join.<locals>.<genexpr>)
r� r
rJ �extendrO ra r( r+ rV rP )r3 rt r� r
� extend_textr� � extend_spansrU r4 )rl rt r: r3 r5 r� � s"
�z Text.join)rN r- c
C s d| j krdS d}|dkr | j}|dk s,t�| �� }|j}| j}| jddd�D ]�}|jddd�}|D ]r}|j �d�r�|j dd� d g|_||� |t |�7 }||d | d } | r�|d| |� || 7 }qd||� qdqN|j g| _t | j �| _
|j | j dd�< dS )
ztConverts tabs to spaces.
Args:
tab_size (int, optional): Size of tabs. Defaults to 8.
� Nr rR T)�include_separatorr� r] r ) r[ rN �AssertionErrorr� ra r+ r> r� rJ rV rP rO )
r3 rN �posrc ra �_styler� r� r� �spacesr4 r4 r5 � expand_tabs� s0
zText.expand_tabsF)rL �pad)r� rL r r- c C s� |p | j p t}|dkr�t| j�}||krV|dkrHt| j|d �d | _nt| j|�| _|r�||k r�|| }| j� d| � �g| _t| j�| _dS )au Truncate text if it is longer that a given width.
Args:
max_width (int): Maximum number of characters in text.
overflow (str, optional): Overflow method: "crop", "fold", or "ellipsis". Defaults to None, to use self.overflow.
pad (bool, optional): Pad with spaces if the length is less than max_width. Defaults to False.
�ignore�ellipsisr u …r] N)rL r& r r[ r rJ rV rP )r3 r� rL r � _overflowr� r r4 r4 r5 �truncate s
z
Text.truncatec s2 t | j��t� � �fdd�| jD �| jdd�< dS )z=Remove or modify any spans that are over the end of the text.c s< g | ]4}|j �k r|j�k r |n� |j t�|j�|j��qS r4 �r) r* r; r+ rx �rl �
max_offsetr4 r5 rm 3 s
��z$Text._trim_spans.<locals>.<listcomp>N)rV r[ r( rO r2 r4 r r5 r� / s
�zText._trim_spansr] )r� � characterr- c s\ t |�dkstd���rX|� }|� | j� |� �| _t� � �fdd�| jD �| jdd�< dS )zxPad left and right with a given number of characters.
Args:
count (int): Width of padding.
r �&Character must be a string of length 1c s& g | ] \}}}� |� |� |��qS r4 r4 ri �rl r� r4 r5 rm H s �z Text.pad.<locals>.<listcomp>N�rV r r[ r( rO )r3 r� r �pad_charactersr4 r r5 r = s �zText.padc sT t |�dkstd���rP|� � | j� �| _t� � �fdd�| jD �| jdd�< dS )z�Pad the left with a given character.
Args:
count (int): Number of characters to pad.
character (str, optional): Character to pad with. Defaults to " ".
r r c s& g | ] \}}}� |� |� |��qS r4 r4 ri r r4 r5 rm X s �z!Text.pad_left.<locals>.<listcomp>Nr �r3 r� r r4 r r5 �pad_leftM s �z
Text.pad_leftc C s0 t |�dkstd��|r,| j� || � �| _dS )z�Pad the right with a given character.
Args:
count (int): Number of characters to pad.
character (str, optional): Character to pad with. Defaults to " ".
r r N)rV r r[ r" r4 r4 r5 r� ] s zText.pad_right)�align�widthr r- c C sp | � |� |t| j� }|rl|dkr2| �||� n:|dkr`|d }| �||� | �|| |� n | �||� dS )z�Align text to a given width.
Args:
align (AlignMethod): One of "left", "center", or "right".
width (int): Desired width.
character (str, optional): Character to pad with. Defaults to " ".
�left�center� N)r r r[ r� r# )r3 r$ r% r � excess_spacer& r4 r4 r5 r$ h s
z
Text.alignr )rU r+ r- c s t |ttf�std��t|��r
t |t�r~t|�}| j�|� t| �}t|��|dk rn| j�t ||� |�� | j
�7 _
n�t |t��r
t � |dk r�t d��| j
�|j dk r�| j�� ��t|� |j �� | j�|j
� | j�� �fdd�|jD �� | j
t|�7 _
| S )z�Add text with an optional style.
Args:
text (Union[Text, str]): A str or Text to append.
style (str, optional): A style name. Defaults to None.
Returns:
Text: Returns self for chaining.
z(Only str or Text can be appended to TextNz2style must not be set when appending Text instancec 3 s( | ] \}}}� |� |� |�V qd S r7 r4 ri �rl r� r4 r5 rz � s �z Text.append.<locals>.<genexpr>)r/ rH r' rq rV r rJ ra rO r( rP �
ValueErrorr+ r[ r )r3 rU r+ rW r: r4 r* r5 ra | s8
�
��z Text.append)rU r- c st t � | j�|jdk r2| j�� ��t|� |j�� | j�|j� | j�� �fdd�|jD �� | jt|�7 _| S )z�Append another Text instance. This method is more performant that Text.append, but
only works for Text.
Returns:
Text: Returns self for chaining.
Nc 3 s( | ] \}}}� |� |� |�V qd S r7 r4 ri r* r4 r5 rz � s �z#Text.append_text.<locals>.<genexpr>) r( rP r+ rO ra rV rJ r[ r �r3 rU r4 r* r5 � append_text� s
�zText.append_text)�tokensr- c C sh | j j}| jj}t}t| �}|D ]<\}}||� |dk rP||||t|� |�� |t|�7 }q || _| S )a( Append iterable of str and style. Style may be a Style instance or a str style definition.
Args:
pairs (Iterable[Tuple[str, Optional[StyleType]]]): An iterable of tuples containing str content and style.
Returns:
Text: Returns self for chaining.
N)rJ ra rO r( rV rP )r3 r. r- r� rl r: �contentr+ r4 r4 r5 �
append_tokens� s zText.append_tokensc C s | j �|j � dS )z�Copy styles from another Text instance.
Args:
text (Text): A Text instance to copy styles from, must be the same length.
N)rO r r, r4 r4 r5 � copy_styles� s zText.copy_styles)r � allow_blank)� separatorr r2 r- c s� � s t d��| j�� �kr(t| �� g�S |rP| �dd� t�t�� ���D ��}n6tt d�� �fdd� }t� fdd�| �|� �D ��}|s���
� �r�|� � |S )a� Split rich text in to lines, preserving styles.
Args:
separator (str, optional): String to split on. Defaults to "\\n".
include_separator (bool, optional): Include the separator in the lines. Defaults to False.
allow_blank (bool, optional): Return a blank line if the text ends with a separator. Defaults to False.
Returns:
List[RichText]: A list of rich text, one per line of the original.
zseparator must not be emptyc s s | ]}|� � V qd S r7 r� �rj r� r4 r4 r5 rz � s z Text.split.<locals>.<genexpr>r, c 3 s4 t �t �� ���D ] } | �� \}}|V |V qd S r7 )r� r� rv ry )r� r) r* �r3 rU r4 r5 �
flatten_spans� s z!Text.split.<locals>.flatten_spansc 3 s | ]}|j � kr|V qd S r7 rZ r� )r3 r4 r5 rz � s
) r r[ r r` rp r� r� rv r
rF r� �pop)r3 r3 r r2 rt r6 r4 r5 r5 r> � s
�
�z
Text.split)�offsetsr- c s� t |�}|st| �� g�S | j�t��}d ||f�}t t||dd� ��}| j�| j�| j�t � t� ����fdd�|D ��}| j
s�|S dd� |j D �}t|�}t } | j
D �]$\}
} �d} |}
| |
d }|| \}}|
|k r�|d }
n|
|kr�|d } n�q | |
d }q�| |k �r |}nT| }} |}
|| \}}| |k �rH|d }
n| |k�rp|d } n�qp| |
d }�q(t
||d �D ]P}|| \}}td|
| �}t| | || �}||k�r~|| | ||��� �q~q�|S )
z�Divide text in to a number of lines at given offsets.
Args:
offsets (Iterable[int]): Offsets used to divide text.
Returns:
Lines: New RichText instances between offsets.
r r Nc 3 s* | ]"\}}� �||� ���d �V qdS ))r+ rK rL Nr4 )rj r) r* �r� rK rL r+ rU r4 r5 rz s �
�z Text.divide.<locals>.<genexpr>c S s g | ] }|j j�qS r4 )rO ra r� r4 r4 r5 rm ) s z Text.divide.<locals>.<listcomp>r( )r )r� r r` r[ rV r r+ rK rL r' rO �_linesr( �ranger� r; )r3 r8 �_offsetsr� �divide_offsets� line_ranges� new_lines�
_line_appends�
line_countrl �
span_start�span_end� lower_bound� upper_bound�
start_line_no�
line_start�line_end� end_line_no�line_no� new_start�new_endr4 r9 r5 rp s` �
z Text.divider )�amountr- c sX t | j�| �t� � �fdd�| jD �| jdd�< | jd| � g| _| j|8 _dS )z7Remove a number of characters from the end of the text.c s< g | ]4}|j �k r|j�k r |n� |j t�|j�|j��qS r4 r rx r r4 r5 rm Z s
��z#Text.right_crop.<locals>.<listcomp>N)rV r[ r( rO rJ rP )r3 rM r4 r r5 rA V s � zText.right_cropr� )r� r% rK rL rN rM r- c
C s� |p | j p t}|p| jpt}t|| jd�p0|dk}t� } | jdd�D ]�}
d|
krZ|
�|� |rjt|
g�} n t t
|
�||dkd�} |
� | �} | D ]}
|
� |� q�|r�| j ||||d� | D ]}
|
j
||d � q�| �| � qD| S )
a� Word wrap the text.
Args:
console (Console): Console instance.
width (int): Number of characters per line.
emoji (bool, optional): Also render emoji code. Defaults to True.
justify (str, optional): Justify method: "default", "left", "center", "full", "right". Defaults to "default".
overflow (str, optional): Overflow method: "crop", "fold", or "ellipsis". Defaults to None.
tab_size (int, optional): Default tab size. Defaults to 8.
no_wrap (bool, optional): Disable wrapping, Defaults to False.
Returns:
Lines: Number of lines.
Fr T�r2 r r% )r% r� )rL )rK r$ rL r& r rM r r> r r rH rp r� r r )
r3 r� r% rK rL rN rM � wrap_justify�
wrap_overflowrt r� r? r8 r4 r4 r5 r� f s0
� z Text.wrap)r% r- c C s0 t � }|j}| �� D ]}|�|� ||� q|S )z�Fit the text in to given width by chopping in to lines.
Args:
width (int): Maximum characters in a line.
Returns:
Lines: List of lines.
)r ra r> r� )r3 r% rt ra r� r4 r4 r5 �fit� s
zText.fitc C sX dd� t jd| jt jd�D �}z ttdd� |D ��p6d}W n tk
rR d}Y nX |S )zrAuto-detect indentation of code.
Returns:
int: Number of spaces used to indent code.
c S s h | ]}t |�d ���qS )r )rV r� r4 r4 r4 r5 � <setcomp>� s �z*Text.detect_indentation.<locals>.<setcomp>�
^( *)(.*)$r� c S s g | ]}|d s|�qS )r( r4 )rj �indentr4 r4 r5 rm � s z+Text.detect_indentation.<locals>.<listcomp>r )r� r� r[ � MULTILINEr r rq )r3 �
_indentations� indentationr4 r4 r5 �detect_indentation� s ��
zText.detect_indentationu │z dim green)r r+ )� indent_sizer r+ r- c C s: |dkr| � � n|}| �� }|�� |� d|d � �}t�d�}g }|j} d}
|jdd�D ]�} |�| j�} | rz| � d�s�|
d7 }
q\| � d�}
t
t |
�|�\}}|| � d| � �}|| jt |�d� | _| � |dt |�� |
r�|�
t||d �g|
� d}
| | � q\|
�r&|�
td
|d �g|
� |�d ��|�}|S ) u� Adds indent guide lines to text.
Args:
indent_size (Optional[int]): Size of indentation, or None to auto detect. Defaults to None.
character (str, optional): Character to use for indentation. Defaults to "│".
style (Union[Style, str], optional): Style of indent guides.
Returns:
Text: New text with indentation guides.
Nr] r rS r TrN r( r� rQ rR )rX r` r r� �compilera r> r� r[ r� �divmodrV r� r r' r� r� )r3 rY r r+ � _indent_sizerU � indent_line� re_indentr? �add_line� blank_linesr� r� rT � full_indents�remaining_space�
new_indentr� r4 r4 r5 �with_indent_guides� s4
zText.with_indent_guides)rQ rQ )rQ )rQ )r N)r N)N)N)rQ )N)r] )r] )r] )r] )N)rR )r )N)PrB rC rD rE � __slots__rH r r r
rI rF r r( rX rY r8 r\ r6 r rd �objectrf rg rh ru �propertyr r� � classmethodr r� r� r r� r r r� r[ �setterrT r� r` r� r� r� r� r� �GetStyleCallabler� r
r� r� r� r� r r� r r� r� r� r r r� r r# r� r r$ ra r- r0 r1 r r> rp rA r� rQ rX rd r4 r4 r4 r5 r' i s� � ��
� �
� �
�% ����
�,
�
� �
� ��� +�
�
� � ;'&�� �
� ,�
��� ,P�� 1 ����__main__)r a�
Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.
ZLorem�boldZipsum�italiczjustify='left'�redr� zjustify='center'�greenr' )r+ rK zjustify='right'�blue�rightzjustify='full'�magenta�full)Br� � functoolsr r �mathr �operatorr �typingr r r r r
r r r
r r �_loopr �_pickr �_wrapr r$ r �cellsr r �
containersr �controlr r� r �jupyterr �measurer �segmentr r+ r r r� r r r! r" r$ rG r& rZ r� rH �TextTyperj r( r' rB �pip._vendor.rich.consolerU r� �rule�printr4 r4 r4 r5 �<module> sn 0
<
�