File: //usr/lib/python3/dist-packages/pip/_vendor/rich/__pycache__/columns.cpython-310.pyc
o
@%Ne� � @ sF d dl mZ d dlmZ d dlmZ d dlmZmZm Z m
Z
m Z ddl m
Z
mZ ddlmZmZmZmZ ddlmZ dd lmZ dd
lmZmZ dd lm Z dd l m Z dd
l m Z G dd� de �Z!e"dkr�d dl#Z#e� Zdd� e$e%e#�&� ��D �Z'e!e'dddd�Z(e�)e(� e�*� de(_+e�)e(� de(_,e�*� e�)e(� dS dS )� )� defaultdict)�chain)�
itemgetter)�Dict�Iterable�List�Optional�Tuple� )�Align� AlignMethod)�Console�ConsoleOptions�RenderableType� RenderResult�� Constrain)� Measurement)�Padding�PaddingDimensions)�Table)�TextType)� JupyterMixinc @ s� e Zd ZdZ ddddddddd�deee dedee d e d
e d e d e d
ee
dee ddfdd�Z deddfdd�Z
dededefdd�ZdS )�Columnsa� Display renderables in neat columns.
Args:
renderables (Iterable[RenderableType]): Any number of Rich renderables (including str).
width (int, optional): The desired width of the columns, or None to auto detect. Defaults to None.
padding (PaddingDimensions, optional): Optional padding around cells. Defaults to (0, 1).
expand (bool, optional): Expand columns to full width. Defaults to False.
equal (bool, optional): Arrange in to equal sized columns. Defaults to False.
column_first (bool, optional): Align items from top to bottom (rather than left to right). Defaults to False.
right_to_left (bool, optional): Start column from right hand side. Defaults to False.
align (str, optional): Align value ("left", "right", or "center") or None for default. Defaults to None.
title (TextType, optional): Optional title for Columns.
N�r r
F)�width�expand�equal� column_first�
right_to_left�align�title� renderables�paddingr r r r r r r! �returnc
C sB t |pg �| _|| _|| _|| _|| _|| _|| _|| _| | _ d S �N)
�listr" r r# r r r r r r! )
�selfr" r# r r r r r r r! � r( �:/usr/lib/python3/dist-packages/pip/_vendor/rich/columns.py�__init__ s
zColumns.__init__�
renderablec C s | j �|� dS )zxAdd a renderable to the columns.
Args:
renderable (RenderableType): Any renderable object.
N)r" �append)r' r+ r( r( r) �add_renderable6 s zColumns.add_renderable�console�optionsc # sD � �j ��fdd�� jD ���sd S t�� j�\}}}}t||�}�j}tt�} t ��}
t
j ����fdd��D ��� j rFt��gt �� �dtdt
tttt f f��� fdd� } tj� jdd d
�} � j| _� j| _� jd ur�|� j| }
t|
�D ] }
| j� jd � q|nA|
d kr�| �� d
}| |
�D ]-\}}
t| | |�| |< t| �� �|t | �d }||kr�t | �d }
n|d |
}q�n|
d ks�td ���fdd�| |
�D �}� j r�fdd�|D �}� j r�� j �t � � �fdd�|D �}� j }| j }td
t |�|
�D ]}||||
� }|�r|d d d� }||� �q| V d S )Nc s" g | ]
}t |t�r
� |�n|�qS r( )�
isinstance�str��.0r+ )�
render_strr( r) �
<listcomp>B � ��z,Columns.__rich_console__.<locals>.<listcomp>c s g | ] }�� �|�j �qS r( )�maximumr2 )r. �get_measurementr/ r( r) r5 O s ��� column_countr$ c 3 s( � t ��}�jrvtt����}|� g� }t|� �D ]
}|| d7 < q |� d � }� fdd�t|�D �}d }}t|�D ] } | || |< || d8 < || r[|d7 }qB|d7 }d}qBt�|�D ]
} | dkro n|| V qgnt���E d H |� r�t� |� �D ]}
dV q�d S d S )Nr
c s g | ]}d g� �qS )���r( )r3 �_�r9 r( r) r5 b s zFColumns.__rich_console__.<locals>.iter_renderables.<locals>.<listcomp>r r: )r N)�lenr r& �zip�ranger �
from_iterable) r9 �
item_count�width_renderables�column_lengths�col_no� row_count�cells�row�col�indexr; )�renderable_widthsr" r' r< r) �iter_renderablesV s8 �
��z2Columns.__rich_console__.<locals>.iter_renderablesTF)r# �collapse_padding�pad_edge)r r
r c s g | ]}� |��qS r( r( )r3 � _renderable)�get_renderabler( r) r5 � s ��c s& g | ]}|d u r
d nt |� d ��qS )Nr r r2 )rJ r( r) r5 � s �� �c s" g | ]
}|d u r
d n� |���qS r% r( r2 )�_Alignr r( r) r5 � r6 r: ) r4 r" r �unpackr# �max� max_widthr �intr= r �getr r r r r r �gridr r! r r? �
add_column�clear�sum�valuesr r r r �add_row)r' r. r/ �_top�right�_bottom�left�
width_paddingrS �widthsr9 rK �tabler; � column_no�renderable_width� total_width� _renderablesr r[ �startrG r( )
rP r r. r8 rO r/ r4 rJ r" r' r) �__rich_console__> s� �
�
���!
�
� �
�
� �
zColumns.__rich_console__)Nr )�__name__�
__module__� __qualname__�__doc__r r r r rT �boolr r r* r- r
r r rh r( r( r( r) r sR ��
������ �
� �
����r �__main__Nc C s g | ] \}}|� d |� ��qS )� r( )r3 �i�sr( r( r) r5 � s r5 r F)r# r r T)-� collectionsr � itertoolsr �operatorr �typingr r r r r r r r r. r
r r r � constrainr �measurer r# r r rb r �textr �jupyterr r ri �os� enumerate�sorted�listdir�files�columns�print�ruler r r( r( r( r) �<module> s6
�