OwlCyberSecurity - MANAGER
Edit File: segment.cpython-39.pyc
a �F�f�^ � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z d dl m Z m Z m Z mZmZmZmZmZmZmZ ddlmZmZmZmZmZ dd lm Z m Z dd l m Z e r�dd l m!Z!m"Z"m#Z# ed �Z$G d d� de�Z%eee% ee%ee&e'f f ee%e&e&f f Z(e � G dd� de��Z)G dd� d�Z*G dd� d�Z+e,dk�r�d dl-m!Z! d dl.m/Z/ d dl0m1Z1 dZ2e1�3d�Z4e!� Z e �5d� e �6d � e �6d � e �6e/e2d d d �� e �6� e �6d!� e7e �8e4��Z9e �6e9� e �6� e �6d"� e �6e4� e �6d#� d$S )%� )�IntEnum)� lru_cache)� filterfalse)� getLogger)� attrgetter) � TYPE_CHECKING�Dict�Iterable�List� NamedTuple�Optional�Sequence�Tuple�Type�Union� )�_is_single_cell_widths�cached_cell_len�cell_len�get_character_cell_size� set_cell_size)�Result� rich_repr)�Style)�Console�ConsoleOptions� RenderResult�richc @ sP e Zd ZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdS )� ControlTypezDNon-printable control codes which typically translate to ANSI codes.r � � � � � � � � � � � � � � � N)�__name__� __module__� __qualname__�__doc__ZBELLZCARRIAGE_RETURN�HOMEZCLEAR� SHOW_CURSOR� HIDE_CURSORZENABLE_ALT_SCREENZDISABLE_ALT_SCREENZ CURSOR_UPZ CURSOR_DOWNZCURSOR_FORWARDZCURSOR_BACKWARDZCURSOR_MOVE_TO_COLUMNZCURSOR_MOVE_TOZ ERASE_IN_LINEZSET_WINDOW_TITLE� r5 r5 �Z/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/pip/_vendor/rich/segment.pyr # s" r c @ s* e Zd ZU dZeed< dZee ed< dZ ee e ed< e e d�dd��Zed�d d �Zed�d d �Ze ed�d d��Zeed�d e ed d�dd���Ze ed d�dd�Zed d�dd��ZedEed ee ee ed d�dd��ZedFed eed d �d d ��Z eed ee d d �d!d"��Z edGed e ee eeee d d$�d%d&��Z edHe d e ee ee d d'�d(d)��Z ee d e d*�d+d,��Z!ee e d ee e f d-�d.d/��Z"edIe e d e ee ee ee e d d0�d1d2��Z#edJe$d e e d e e eee e d d3�d4d5��Z%edKe$d e e d e e eee e d d3�d6d7��Z&edLe$d e e d e e eee e d d3�d8d9��Z'eed ed d �d:d;��Z(eed ed d �d<d=��Z)eed ed d �d>d?��Z*eed ed d �d@dA��Z+eed ee ee d dB�dCdD��Z,dS )M�Segmenta� A piece of text with associated style. Segments are produced by the Console render process and are ultimately converted in to strings to be written to the terminal. Args: text (str): A piece of text. style (:class:`~rich.style.Style`, optional): An optional style to apply to the text. control (Tuple[ControlCode], optional): Optional sequence of control codes. Attributes: cell_length (int): The cell length of this Segment. �textN�style�control)�returnc C s | \}}}|rdS t |�S )zzThe number of terminal cells required to display self.text. Returns: int: A number of cells. r )r )�selfr8 �_styler: r5 r5 r6 � cell_lengthQ s zSegment.cell_lengthc c s: | j V | jd u r&| jd ur6| jV n| jV | jV d S �N)r8 r: r9 �r< r5 r5 r6 � __rich_repr__[ s zSegment.__rich_repr__c C s t | j�S )z#Check if the segment contains text.)�boolr8 r@ r5 r5 r6 �__bool__d s zSegment.__bool__c C s | j duS )z,Check if the segment contains control codes.N)r: r@ r5 r5 r6 � is_controlh s zSegment.is_controli @ )r7 r7 )�segment�cutr; c C s, |\}}}t }|j}||kr,||d||�fS t}t|| t|�d �} |d | � } t| �} | |kr�|| ||�||| d � ||�fS | t|�k �r || } | d7 } | || �7 } |d | � } | |kr�|| ||�||| d � ||�fS | |kr�|| d | d � d ||�|d|| d � ||�fS q�td��d S )N� r � zWill never reach here)r7 r> r �int�lenr �AssertionError) �clsrE rF r8 r9 r: �_Segmentr> � cell_size�pos�beforeZcell_pos�charr5 r5 r6 � _split_cellsm s6 � ��zSegment._split_cells)rF r; c C sb | \}}}t |�rV|t|�kr.| td||�fS t|d|� ||�t||d� ||�fS | �| |�S )a0 Split segment in to two segments at the specified column. If the cut point falls in the middle of a 2-cell wide character then it is replaced by two spaces, to preserve the display width of the parent segment. Returns: Tuple[Segment, Segment]: Two segments. rG N)r rJ r7 rR )r< rF r8 r9 r: r5 r5 r6 � split_cells� s �zSegment.split_cellsc C s | d�S )zMake a new line segment.� r5 )rL r5 r5 r6 �line� s z Segment.line)�segmentsr9 � post_styler; c s> |}|r"|j � � �fdd�|D �}�r:��fdd�|D �}|S )a Apply style(s) to an iterable of segments. Returns an iterable of segments where the style is replaced by ``style + segment.style + post_style``. Args: segments (Iterable[Segment]): Segments to process. style (Style, optional): Base style. Defaults to None. post_style (Style, optional): Style to apply on top of segment style. Defaults to None. Returns: Iterable[Segments]: A new iterable of segments (possibly the same iterable). c 3 s, | ]$\}}}�||rd n� |�|�V qd S r? r5 ��.0r8 r= r: )�applyrL r5 r6 � <genexpr>� s �z&Segment.apply_style.<locals>.<genexpr>c 3 s4 | ],\}}}� ||rd n|r$|� n�|�V qd S r? r5 rX )rL rW r5 r6 r[ � s ���)�__add__)rL rV r9 rW Zresult_segmentsr5 )rZ rL rW r6 � apply_style� s � � zSegment.apply_styleF)rV rD r; c C s$ |rt td�|�S ttd�|�S dS )a2 Filter segments by ``is_control`` attribute. Args: segments (Iterable[Segment]): An iterable of Segment instances. is_control (bool, optional): is_control flag to match in search. Returns: Iterable[Segment]: And iterable of Segment instances. r: N)�filterr r )rL rV rD r5 r5 r6 �filter_control� s zSegment.filter_control)rV r; c c s� g }|j }|D ]d}d|jv rj|jsj|\}}}|rr|�d�\}} }|rR|| ||�� | r,|V g }|j }q,q||� q|r~|V dS )a Split a sequence of segments in to a list of lines. Args: segments (Iterable[Segment]): Segments potentially containing line feeds. Yields: Iterable[List[Segment]]: Iterable of segment lists, one per line. rT N)�appendr8 r: � partition) rL rV rU r` rE r8 r9 �_�_text�new_liner5 r5 r6 � split_lines� s zSegment.split_linesT)rV �lengthr9 �pad�include_new_linesr; c c s� g }|j }| j}| d�} |D ]�} d| jv r�| js�| \} } } | r�| �d�\}}} |r`|| || �� |r:|||||d�}|r�|� | � |V |�� q:q || � q |r�|||||d�V dS )a� Split segments in to lines, and crop lines greater than a given length. Args: segments (Iterable[Segment]): An iterable of segments, probably generated from console.render. length (int): Desired line length. style (Style, optional): Style to use for any padding. pad (bool): Enable padding of lines that are less than `length`. Returns: Iterable[List[Segment]]: An iterable of lines of segments. rT )r9 rg N)r` �adjust_line_lengthr8 r: ra �clear)rL rV rf r9 rg rh rU r` ri Znew_line_segmentrE r8 � segment_stylerb rc rd Z cropped_liner5 r5 r6 �split_and_crop_lines s, � z Segment.split_and_crop_lines)rU rf r9 rg r; c C s� t dd� |D ��}||k rF|r8|| d|| |�g }q�|dd� }n�||kr�g }|j}d}|D ]X}|j} || |k s||jr�||� || 7 }q`|\} } } t| || �} || | | �� q�q`n |dd� }|S )a Adjust a line to a given width (cropping or padding as required). Args: segments (Iterable[Segment]): A list of segments in a single line. length (int): The desired width of the line. style (Style, optional): The style of padding if used (space on the end). Defaults to None. pad (bool, optional): Pad lines with spaces if they are shorter than `length`. Defaults to True. Returns: List[Segment]: A line of segments with the desired length. c s s | ] }|j V qd S r? )r> )rY rE r5 r5 r6 r[ J � z-Segment.adjust_line_length.<locals>.<genexpr>rH Nr )�sumr` r> r: r ) rL rU rf r9 rg Z line_lengthrd r` rE Zsegment_lengthr8 rk rb r5 r5 r6 ri 7 s( zSegment.adjust_line_length)rU r; c s t � t� fdd�|D ��S )z�Get the length of list of segments. Args: line (List[Segment]): A line encoded as a list of Segments (assumes no '\\n' characters), Returns: int: The length of the line. c 3 s | ]\}}}|s� |�V qd S r? r5 )rY r8 r9 r: �� _cell_lenr5 r6 r[ o rm z*Segment.get_line_length.<locals>.<genexpr>)r rn )rL rU r5 ro r6 �get_line_lengthd s zSegment.get_line_length)�linesr; c s0 | j � |r t� fdd�|D ��nd}|t|�fS )z�Get the shape (enclosing rectangle) of a list of lines. Args: lines (List[List[Segment]]): A list of lines (no '\\n' characters). Returns: Tuple[int, int]: Width and height in characters. c 3 s | ]}� |�V qd S r? r5 �rY rU �rq r5 r6 r[ | rm z$Segment.get_shape.<locals>.<genexpr>r )rq �maxrJ )rL rr � max_widthr5 rt r6 � get_shapeq s zSegment.get_shape)rr �width�heightr9 � new_linesr; c s� |p t |�}|r$| d� d ��gn| d� ��g}| j� |d|� }� ��fdd�|D �|dd�< t |�|k r�|�|g|t |� � |S )a� Set the shape of a list of lines (enclosing rectangle). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style, optional): Style of any padding added. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. rH rT Nc s g | ]}� |��d ��qS ))r9 r5 rs �ri r9 rx r5 r6 � <listcomp>� s z%Segment.set_shape.<locals>.<listcomp>)rJ ri �extend) rL rr rx ry r9 rz �_height�blankZ shaped_linesr5 r{ r6 � set_shape s &� � zSegment.set_shape)rL rr rx ry r9 rz r; c C s` |t |� }|s |dd� S |d|� }|r>| d| d |�n | d| |�}||gg| }|S )a� Aligns lines to top (adds extra lines to bottom as required). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style): Style of any padding added. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. NrH rT �rJ �rL rr rx ry r9 rz � extra_linesr r5 r5 r6 � align_top� s $zSegment.align_topc C s` |t |� }|s |dd� S |d|� }|r>| d| d |�n | d| |�}|gg| | }|S )a� Aligns render to bottom (adds extra lines above as required). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style): Style of any padding added. Defaults to None. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. NrH rT r� r� r5 r5 r6 � align_bottom� s $zSegment.align_bottomc C s| |t |� }|s |dd� S |d|� }|r>| d| d |�n | d| |�}|d }|| } |gg| | |gg| }|S )a� Aligns lines to middle (adds extra lines to above and below as required). Args: lines (List[List[Segment]]): A list of lines. width (int): Desired width. height (int, optional): Desired height or None for no change. style (Style): Style of any padding added. new_lines (bool, optional): Padded lines should include " ". Defaults to False. Returns: List[List[Segment]]: New list of lines. NrH rT r r� ) rL rr rx ry r9 rz r� r Z top_linesZ bottom_linesr5 r5 r6 � align_middle� s $ zSegment.align_middlec c st t |�}z t|�}W n ty( Y dS 0 t}|D ]6}|j|jkr^|js^||j|j |j�}q2|V |}q2|V dS )a) Simplify an iterable of segments by combining contiguous segments with the same style. Args: segments (Iterable[Segment]): An iterable of segments. Returns: Iterable[Segment]: A possibly smaller iterable of segments that will render the same way. N)�iter�next� StopIterationr7 r9 r: r8 )rL rV � iter_segmentsZ last_segmentrM rE r5 r5 r6 �simplify� s �zSegment.simplifyc c sJ |D ]@}|j s|jdu r |V q|\}}}| ||r<|�d�nd�V qdS )z�Remove all links from an iterable of styles. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with link removed. N)r: r9 � update_link)rL rV rE r8 r9 �_controlr5 r5 r6 � strip_links s zSegment.strip_linksc c s" |D ]\}}}| |d|�V qdS )z�Remove all styles from an iterable of segments. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with styles replace with None Nr5 )rL rV r8 r= r: r5 r5 r6 � strip_styles( s zSegment.strip_stylesc c sZ i }|D ]L\}}}|rF|� |�}|du r6|j}|||< | |||�V q| |d|�V qdS )z�Remove all color from an iterable of segments. Args: segments (Iterable[Segment]): An iterable segments. Yields: Segment: Segments with colorless style. N)�get� without_color)rL rV �cacher8 r9 r: Zcolorless_styler5 r5 r6 � remove_color5 s zSegment.remove_color)rV �cutsr; c c sD g }|j }t|�}t|d�}|dkr(g S |dkr2q:g V qd}|j}|j} t} |D ]�} | \} } }| rR|rl|n || | � }||k r�|| � |}qR||kr�|| � | � V |� |}t|d�}|dkr�|r�| � V dS qRn6| �|| �\}} | \} } }||� | � V |� |}t|d�}|dkr`|�r.| � V dS q`qR| � V dS )z�Divides an iterable of segments in to portions. Args: cuts (Iterable[int]): Cell positions where to divide. Yields: [Iterable[List[Segment]]]: An iterable of Segments in List. ���r N)r` r� r� rj �copyr rS )rL rV r� Zsplit_segmentsZ add_segmentZ iter_cutsrF rO Zsegments_clearZ segments_copyrp rE r8 r= r: Zend_posrP r5 r5 r6 �divideK sX zSegment.divide)NN)F)NTT)NT)NNF)F)F)F)-r. r/ r0 r1 �str�__annotations__r9 r r r: r � ControlCode�propertyrI r> r rA rB rC rD � classmethodr r rR rS rU r r] r_ r re rl ri rq rw r� r r� r� r� r� r� r� r� r� r5 r5 r5 r6 r7 ? s� % ��( �� � �. ��, " � �# � � � � � � �r7 c @ s: e Zd ZdZd ee edd�dd�Zddd d �d d �ZdS )�Segmentsa= A simple renderable to render an iterable of segments. This class may be useful if you want to print segments outside of a __rich_console__ method. Args: segments (Iterable[Segment]): An iterable of segments. new_lines (bool, optional): Add new lines between segments. Defaults to False. FN)rV rz r; c C s t |�| _|| _d S r? )�listrV rz )r< rV rz r5 r5 r6 �__init__� s zSegments.__init__r r r ��console�optionsr; c c s8 | j r(t�� }| jD ]}|V |V qn | jE d H d S r? )rz r7 rU rV )r< r� r� rU rE r5 r5 r6 �__rich_console__� s zSegments.__rich_console__)F) r. r/ r0 r1 r r7 rB r� r� r5 r5 r5 r6 r� � s �r� c @ s: e Zd Zd eee edd�dd�Zdddd �d d �ZdS ) � SegmentLinesFN)rr rz r; c C s t |�| _|| _dS )a= A simple renderable containing a number of lines of segments. May be used as an intermediate in rendering process. Args: lines (Iterable[List[Segment]]): Lists of segments forming lines. new_lines (bool, optional): Insert new lines after each line. Defaults to False. N)r� rr rz )r< rr rz r5 r5 r6 r� � s zSegmentLines.__init__r r r r� c c sF | j r,t�� }| jD ]}|E d H |V qn| jD ]}|E d H q2d S r? )rz r7 rU rr )r< r� r� rd rU r5 r5 r6 r� � s z SegmentLines.__rich_console__)F) r. r/ r0 r r r7 rB r� r� r5 r5 r5 r6 r� � s �r� �__main__)r )�Syntax)�Textz�from rich.console import Console console = Console() text = Text.from_markup("Hello, [bold magenta]World[/]!") console.print(text)z Hello, [bold magenta]World[/]!z rich.Segmentz]A Segment is the last step in the Rich render process before generating text with ANSI codes.z Consider the following code: �pythonT)Z line_numberszRWhen you call [b]print()[/b], Rich [i]renders[/i] the object in to the following: zAThe Segments are then processed to produce the following output: zS You will only need to know this if you are implementing your own Rich renderables.N):�enumr � functoolsr � itertoolsr �loggingr �operatorr �typingr r r r r r r r r r �cellsr r r r r �reprr r r9 r r� r r r �logr rI r� r� r7 r� r� r. �pip._vendor.rich.consoleZpip._vendor.rich.syntaxr� �pip._vendor.rich.textr� �code� from_markupr8 �rule�printr� �render� fragmentsr5 r5 r5 r6 �<module> sf 0 �� U � � �