File: //usr/local/lib/python3.7/site-packages/pip/_vendor/rich/__pycache__/layout.cpython-37.pyc
B
L��g�6 � @ st d dl mZmZ d dlmZ d dlmZ d dlmZ d dl m
Z
m Z m Z m
Z
mZmZmZmZmZ ddlmZ ddlmZ dd lmZmZmZmZ dd
l m Z dd l m Z dd l m!Z! dd
l"m#Z# ddl$m%Z%m&Z& ddl'm(Z( ddl)m*Z* e
r�d dl+m,Z, G dd� de�Z-e de#f Z.e de-f Z/G dd� de0�Z1G dd� de1�Z2G dd� d�Z3G dd � d e�Z4G d d � d e4�Z5G d d � d e4�Z6e&G d!d� d��Z7e8d"k�rpd d#l9mZ e� Ze7� Z:e:�;e7d$d%d&�e7dd'd(�e7d)d*d+�� e:d' �<e7d,d-�e7d.d/d0�� e:d. �<e7d1d/d0�e7d2d-�� e:d2 �;e7d3d-�e7d4d-�e7d5d-�� e:d, �;e7e:j=d6d-�e7d7d-�� e:d1 �>d8� e�?e:� d9S ):� )�ABC�abstractmethod)�islice)�
itemgetter)�RLock) �
TYPE_CHECKING�Dict�Iterable�List�
NamedTuple�Optional�Sequence�Tuple�Union� )�
ratio_resolve)�Align)�Console�ConsoleOptions�RenderableType� RenderResult)�ReprHighlighter)�Panel)�Pretty)�Region)�Result� rich_repr)�Segment)� StyleType)�Treec @ s* e Zd ZU dZeed< eee ed< dS )� LayoutRenderz An individual layout render.�region�renderN)�__name__�
__module__� __qualname__�__doc__r �__annotations__r
r � r( r( �;/tmp/pip-unpacked-wheel-hv55ucu3/pip/_vendor/rich/layout.pyr s
r �Layoutc @ s e Zd ZdZdS )� LayoutErrorzLayout related error.N)r# r$ r% r& r( r( r( r) r+ + s r+ c @ s e Zd ZdZdS )�
NoSplitterz"Requested splitter does not exist.N)r# r$ r% r& r( r( r( r) r, / s r, c @ s< e Zd ZdZe� Zd dedd�dd�Zee e
d�d d
�Z dS ) � _Placeholderz4An internal renderable used as a Layout placeholder.� r* N)�layout�style�returnc C s || _ || _d S )N)r/ r0 )�selfr/ r0 r( r( r) �__init__8 s z_Placeholder.__init__)�console�optionsr1 c c sx |j }|jp|jj}| j}|jr8|j�d|� d|� d�nd|� d|� d�}ttjt|�dd�| j | �
|�d|d�V d S ) Nz (z x �)�(�middle)ZverticalZblue)r0 �titleZ border_style�height) � max_widthr: �sizer/ �namer r �centerr r0 � highlighter)r2 r4 r5 �widthr: r/ r9 r( r( r) �__rich_console__< s z _Placeholder.__rich_console__)r. ) r# r$ r% r& r r? r r3 r r r rA r( r( r( r) r- 3 s
r- c @ sV e Zd ZU dZdZeed< eed�dd��Zee d e
e e de
f d�d d
��Z
d S ) �SplitterzBase class for a splitter.r. r= )r1 c C s dS )z(Get the icon (emoji) used in layout.treeNr( )r2 r( r( r) �
get_tree_iconU s zSplitter.get_tree_iconr* )�childrenr! r1 c C s dS )z�Divide a region amongst several child layouts.
Args:
children (Sequence(Layout)): A number of child layouts.
region (Region): A rectangular region to divide.
Nr( )r2 rD r! r( r( r) �divideY s zSplitter.divideN)r# r$ r% r& r= �strr' r rC r
r r r rE r( r( r( r) rB P s
rB c @ sD e Zd ZdZdZed�dd�Zed ee e
def d�dd �Z d
S ) � RowSplitterz!Split a layout region in to rows.�row)r1 c C s dS )Nu [layout.tree.row]⬌r( )r2 r( r( r) rC j s zRowSplitter.get_tree_iconr* )rD r! r1 c c sZ |\}}}}t ||�}d}t} x6t||�D ](\}
} |
| || || |�fV || 7 }q*W d S )Nr )r r �zip) r2 rD r! �x�yr@ r: Z
render_widths�offset�_Region�childZ child_widthr( r( r) rE m s
zRowSplitter.divideN) r# r$ r% r& r= rF rC r
r r r rE r( r( r( r) rG e s
rG c @ sD e Zd ZdZdZed�dd�Zed ee e
def d�dd �Z d
S ) �ColumnSplitterz$Split a layout region in to columns.�column)r1 c C s dS )Nu [layout.tree.column]⬍r( )r2 r( r( r) rC ~ s z ColumnSplitter.get_tree_iconr* )rD r! r1 c c sZ |\}}}}t ||�}d}t} x6t||�D ](\}
} |
| ||| || �fV || 7 }q*W d S )Nr )r r rI ) r2 rD r! rJ rK r@ r: Zrender_heightsrL rM rN Z child_heightr( r( r) rE � s
zColumnSplitter.divideN) r# r$ r% r& r= rF rC r
r r r rE r( r( r( r) rO y s
rO c
@ s� e Zd ZdZeed�Zd8dddddd�ee ee ee
e
e
e dd�dd �Z e
d
�d d �Zeed
�d
d��Zeed d
�dd��Zeed
�dd��Ze ed d�dd�Ze d d�dd�Zedd
�dd��Zdd �ed ef eee f dd �d d �Zed ef dd �d!d"�Zed ef dd �d#d$�Z ed ef dd �d%d&�Z dd
�d'd(�Z edd)�d*d+�Z d,e dd-�d.d/�Z e
e
e!d0�d1d2�Z"e#e$ed3�d4d5�Z%e#e$e&d3�d6d7�Z'dS )9r* aB A renderable to divide a fixed height in to rows or columns.
Args:
renderable (RenderableType, optional): Renderable content, or None for placeholder. Defaults to None.
name (str, optional): Optional identifier for Layout. Defaults to None.
size (int, optional): Optional fixed size of layout. Defaults to None.
minimum_size (int, optional): Minimum size of layout. Defaults to 1.
ratio (int, optional): Optional ratio for flexible layout. Defaults to 1.
visible (bool, optional): Visibility of layout. Defaults to True.
)rH rP Nr T)r= r< � minimum_size�ratio�visible)�
renderabler= r< rQ rR rS r1 c C sR |p
t | �| _|| _|| _|| _|| _|| _| jd � | _g | _ i | _
t � | _ d S )NrP )
r- � _renderabler< rQ rR r= rS � splitters�splitter� _children� _render_mapr �_lock)r2 rT r= r< rQ rR rS r( r( r) r3 � s
zLayout.__init__)r1 c c s< d| j d fV d| jd fV d| jdfV d| jdfV d S )Nr= r< rQ r rR )r= r< rQ rR )r2 r( r( r) �
__rich_repr__� s zLayout.__rich_repr__c C s | j r
| S | jS )zLayout renderable.)rX rU )r2 r( r( r) rT � s zLayout.renderablec C s dd� | j D �S )z Gets (visible) layout children.c S s g | ]}|j r|�qS r( )rS )�.0rN r( r( r) �
<listcomp>� s z#Layout.children.<locals>.<listcomp>)rX )r2 r( r( r) rD � s zLayout.childrenc C s | j S )z Get a map of the last render.)rY )r2 r( r( r) �map� s z
Layout.map)r= r1 c C s8 | j |kr| S x$| jD ]}|�|�}|dk r|S qW dS )z�Get a named layout, or None if it doesn't exist.
Args:
name (str): Name of layout.
Returns:
Optional[Layout]: Layout instance or None if no layout was found.
N)r= rX �get)r2 r= rN Z named_layoutr( r( r) r_ � s
z
Layout.getc C s$ | � |�}|d kr td|����|S )NzNo layout with name )r_ �KeyError)r2 r= r/ r( r( r) � __getitem__� s
zLayout.__getitem__r c s� ddl m� ddlm� ddlm} d�d�� �fdd� �| }|�|�d |jj� �d
d �}d dd
d���fdd� ��|| � |S )z/Get a tree renderable to show layout structure.r )�Styled)�Table)r r* )r/ r1 c sF | j �� }�jdd�}| jr$t| �n � t| �d�}|�||� |}|S )N)r r r r )�paddingZdim)rW rC ZgridrS r Zadd_row)r/ Zicon�table�textZ_summary)rb rc r( r) �summary� s
z Layout.tree.<locals>.summaryz layout.tree.T)� guide_styleZ highlightr N)�treer/ r1 c s6 x0|j D ]&}� | j�|�d|jj� �d�|� qW d S )Nz layout.tree.)rh )rX �addrW r= )ri r/ rN )�recurserg r( r) rk � s z Layout.tree.<locals>.recurse)Zpip._vendor.rich.styledrb Zpip._vendor.rich.tablerc �pip._vendor.rich.treer rW r= )r2 r r/ ri r( )rb rc rk rg r) ri � s
z Layout.treerP )rW )�layoutsrW r1 c G sd dd� |D �}y t |t�r |n
| j| � | _W n" tk
rP td|����Y nX || jdd�< dS )z�Split the layout in to multiple sub-layouts.
Args:
*layouts (Layout): Positional arguments should be (sub) Layout instances.
splitter (Union[Splitter, str]): Splitter instance or name of splitter.
c S s" g | ]}t |t�r|nt|��qS r( )�
isinstancer* )r\ r/ r( r( r) r] s z Layout.split.<locals>.<listcomp>zNo splitter called N)rn rB rV rW r` r, rX )r2 rW rm �_layoutsr( r( r) �split s z Layout.split)rm r1 c G s dd� |D �}| j �|� dS )z�Add a new layout(s) to existing split.
Args:
*layouts (Union[Layout, RenderableType]): Positional arguments should be renderables or (sub) Layout instances.
c s s$ | ] }t |t�r|nt|�V qd S )N)rn r* )r\ r/ r( r( r) � <genexpr>' s z#Layout.add_split.<locals>.<genexpr>N)rX �extend)r2 rm ro r( r( r) � add_split s zLayout.add_splitc G s | j |ddi� dS )z�Split the layout in to a row (layouts side by side).
Args:
*layouts (Layout): Positional arguments should be (sub) Layout instances.
rW rH N)rp )r2 rm r( r( r) � split_row, s zLayout.split_rowc G s | j |ddi� dS )z�Split the layout in to a column (layouts stacked on top of each other).
Args:
*layouts (Layout): Positional arguments should be (sub) Layout instances.
rW rP N)rp )r2 rm r( r( r) � split_column4 s zLayout.split_columnc C s | j dd�= dS )z Reset splits to initial state.N)rX )r2 r( r( r) �unsplit< s zLayout.unsplit)rT r1 c C s | j � || _W dQ R X dS )zjUpdate renderable.
Args:
renderable (RenderableType): New renderable object.
N)rZ rU )r2 rT r( r( r) �update@ s z
Layout.updater )r4 � layout_namer1 c C sl | j �\ | | }| j| \}}|\}}}} |�||j�|| ��}
t||
�| j|< |�|
||� W dQ R X dS )z�Refresh a sub-layout.
Args:
console (Console): Console instance where Layout is to be rendered.
layout_name (str): Name of layout.
N)rZ rY � render_linesr5 �update_dimensionsr Zupdate_screen_lines) r2 r4 rx r/ r! �_linesrJ rK r@ r: �linesr( r( r) �refresh_screenI s zLayout.refresh_screen)r@ r: r1 c
C s� | t dd||�fg}|j}|j}g }|j}xH|rr||� � |d \}} |j}
|
r,x |j�|
| �D ] } || � q`W q,W dd� t|td�d�D �} | S )z,Create a dict that maps layout on to Region.r ���c S s i | ]\}}||�qS r( r( )r\ r/ r! r( r( r) �
<dictcomp>i s z+Layout._make_region_map.<locals>.<dictcomp>r )�key)r �append�poprD rW rE �sortedr )
r2 r@ r: �stack�pushr� �layout_regionsZappend_layout_regionr/ r! rD Zchild_and_region�
region_mapr( r( r) �_make_region_mapZ s
zLayout._make_region_map)r4 r5 r1 c
C sz |j }|jp|j}| �||�}dd� |�� D �}i }|j}|j} x4|D ],\}
} ||
j| | j| j��} t| | �||
< qFW |S )z�Render the sub_layouts.
Args:
console (Console): Console instance.
options (ConsoleOptions): Console options.
Returns:
RenderMap: A dict that maps Layout on to a tuple of Region, lines
c S s g | ]\}}|j s||f�qS r( )rD )r\ r/ r! r( r( r) r] } s z!Layout.render.<locals>.<listcomp>) r; r: r� �itemsry rz rT r@ r )
r2 r4 r5 Z render_widthZ
render_heightr� r� �
render_mapry rz r/ r! r| r( r( r) r" o s
z
Layout.renderc c s� | j �� |jp|j}|jp |j}| �||�||��}|| _dd� t|�D �}t}xN|� � D ]B\}} |\}
} } }
x,t
||| | |
�| �D ]\}}|� |� q�W qZW t �
� }x |D ]}|E d H |V q�W W d Q R X d S )Nc S s g | ]}g �qS r( r( )r\ �_r( r( r) r] � s z+Layout.__rich_console__.<locals>.<listcomp>)rZ r; r@ r: r" rz rY �ranger �valuesrI rr r �line)r2 r4 r5 r@ r: r� Z layout_lines�_islicer! r| �_xrK Z
_layout_widthZ
layout_heightrH r� �new_lineZ
layout_rowr( r( r) rA � s
zLayout.__rich_console__)N)(r# r$ r% r& rG rO rV r r rF �int�boolr3 r r[ �propertyrT r
rD � RenderMapr^ r_ ra ri r rB rp rs rt ru rv rw r} � RegionMapr� r r r" r rA r( r( r( r) r* � s@
*
�__main__)r �header� )r= r< �main)rR r= �
�footer)r< r= Zside)r= �body� )r= rR �content�s2�topr8 ZbottomZleft1Zleft2ZfooN)@�abcr r � itertoolsr �operatorr � threadingr �typingr r r r
r r r
r r Z_ratior Zalignr r4 r r r r r? r Zpanelr �prettyr r! r �reprr r �segmentr r0 r rl r r r� r� � Exceptionr+ r, r- rB rG rO r* r# Zpip._vendor.rich.consoler/ ru rt ri rw �printr( r( r( r) �<module> sV ,