U
    祡cP(                     @   sf  d dl Z d dlmZ d dlmZmZmZ e jdkr@d dlmZ nd dl	mZ ddl
mZ ddlmZ dd	lmZ dd
lmZ ddlmZ erddlmZmZmZmZ ed Zed ZG dd deZG dd deZedkrbd dlmZmZ d dl m!Z! d dl"m#Z# e! Z$e Ze#ee%e$de&e$de'e$dddddZ(e)ej&e(ddej*d  dS )!    N)chain)TYPE_CHECKINGIterableOptional)      )Literal   )	Constrain)JupyterMixin)Measurement)Segment)	StyleType)ConsoleConsoleOptionsRenderableTypeRenderResultleftcenterrighttopmiddleZbottomc                   @   s0  e Zd ZdZdddddddeee ee eee	 ee	 dddd	Z
ed
ddZedddddddee ee eee	 ee	 d dddZedddddddee ee eee	 ee	 d dddZedddddddee ee eee	 ee	 d dddZddddddZddedddZdS ) Aligna  Align a renderable by adding spaces if necessary.

    Args:
        renderable (RenderableType): A console renderable.
        align (AlignMethod): One of "left", "center", or "right""
        style (StyleType, optional): An optional style to apply to the background.
        vertical (Optional[VerticalAlginMethod], optional): Optional vertical align, one of "top", "middle", or "bottom". Defaults to None.
        pad (bool, optional): Pad the right with spaces. Defaults to True.
        width (int, optional): Restrict contents to given width, or None to use default width. Defaults to None.
        height (int, optional): Set height of align renderable, or None to fit to contents. Defaults to None.

    Raises:
        ValueError: if ``align`` is not one of the expected values.
    r   NT)verticalpadwidthheightr   )
renderablealignstyler   r   r   r   returnc                C   sf   |dkrt d|d|d k	r8|dkr8t d|d|| _|| _|| _|| _|| _|| _|| _d S )Nr   zDinvalid value for align, expected "left", "center", or "right" (not )r   zGinvalid value for vertical, expected "top", "middle", or "bottom" (not )
ValueErrorr   r    r!   r   r   r   r   )selfr   r    r!   r   r   r   r    r&   ./tmp/pip-unpacked-wheel-qkhzhrww/rich/align.py__init__'   s    

zAlign.__init__r"   c                 C   s   d| j d| jdS )NzAlign(z, r#   )r   r    r%   r&   r&   r'   __repr__B   s    zAlign.__repr__)r   r!   r   r   r   r   r"   c             	   C   s   | |d|||||dS )zAlign a renderable to the left.r   r!   r   r   r   r   r&   clsr   r!   r   r   r   r   r&   r&   r'   r   E   s    z
Align.leftc             	   C   s   | |d|||||dS )z!Align a renderable to the center.r   r,   r&   r-   r&   r&   r'   r   [   s    zAlign.centerc             	   C   s   | |d|||||dS )z Align a renderable to the right.r   r,   r&   r-   r&   r&   r'   r   q   s    zAlign.rightr   r   r   consoleoptionsr"   c                 #   s  j  |jj|dj}|tjjd kr0|n
t|j|jd d}t	t
|t
\}}t
||t
 |j| jd k	r|jnd tt
 d fdd}jrt
djp|j  dnt
dttt
 dfd	d
}jp|j}jr|d k	rjdkrH|| }	t| ||	}
nTjdkr|| d }|| | }	t||| ||	}
n|| }t||| }
n| }
jr|jt
|
}
|
E d H  d S )N)r1   r   r)   c                  3   s  dkr$D ]} | E d H  V  qn dkrjj r@td nd }D ]} | E d H  |r`|V  V  qHn dkrԈd }td| }j rtd|  nd }D ](} |r|V  | E d H  |r|V  V  qn8 dkrtd }D ]} |V  | E d H  V  qd S )Nr   r    r      r   )r   r   )liner   r   Z	pad_right)r    excess_spacelinesnew_liner%   r!   r&   r'   generate_segments   s:    







z1Align.__rich_console__.<locals>.generate_segmentsr3   
countr"   c                 3   s    | dkrt | D ]
} V  qd S )Nr   ranger<   _)
blank_liner&   r'   blank_lines   s    z+Align.__rich_console__.<locals>.blank_linesr   r   r4   )r    measurer   maximumrenderr
   r   minupdatelistr   Zsplit_lines	get_shapeZ	set_shaper5   	max_widthr!   	get_styler   r   intr   r   r   Zapply_style)r%   r0   r1   r   renderedr   r9   rB   Zvertical_heightbottom_spaceZiter_segments	top_spacer&   )r    rA   r6   r7   r8   r%   r!   r'   __rich_console__   sR     

 )zAlign.__rich_console__c                 C   s   t ||| j}|S Nr   getr   r%   r0   r1   Zmeasurementr&   r&   r'   __rich_measure__   s    zAlign.__rich_measure__)r   N)N)N)N)__name__
__module____qualname____doc__AlignMethodr   r   VerticalAlignMethodboolrL   r(   strr+   classmethodr   r   r   rP   r   rU   r&   r&   r&   r'   r      s         ] r   c                   @   sZ   e Zd ZdZddee ddddZeddd	Zd
dddddZ	d
de
dddZdS )VerticalCenterzVertically aligns a renderable.

    Warn:
        This class is deprecated and may be removed in a future version. Use Align class with
        `vertical="middle"`.

    Args:
        renderable (RenderableType): A renderable object.
    Nr   )r   r!   r"   c                 C   s   || _ || _d S rQ   )r   r!   )r%   r   r!   r&   r&   r'   r(      s    zVerticalCenter.__init__r)   c                 C   s   d| j dS )NzVerticalCenter(r#   )r   r*   r&   r&   r'   r+      s    zVerticalCenter.__repr__r   r   r   r/   c                 #   s   | j d k	r|| j nd }|j| j|jd ddd}t|\}}t |jpV|j	j}|t
| d }|| t
| }	td|  | ttt d fdd}
|d	kr|
|E d H  |D ]}|E d H  V  q|	d	kr|
|	E d H  d S )
Nr2   F)r   r4   r3   r;   c                 3   s   t | D ]} V  V  qd S rQ   r=   r?   rA   r8   r&   r'   rB     s    z4VerticalCenter.__rich_console__.<locals>.blank_linesr   )r!   rK   Zrender_linesr   rG   r   rI   r5   r   sizelenrL   r   )r%   r0   r1   r!   r7   r   Z_heightr   rO   rN   rB   r5   r&   r`   r'   rP      s(     
 
zVerticalCenter.__rich_console__c                 C   s   t ||| j}|S rQ   rR   rT   r&   r&   r'   rU     s    zVerticalCenter.__rich_measure__)N)rV   rW   rX   rY   r   r   r(   r]   r+   rP   r   rU   r&   r&   r&   r'   r_      s       r_   __main__)r   Group)ReprHighlighter)Panelzalign='left'zalign='center'zalign='right'<   zon dark_blueZAlgin)r   r!   titler   zon red)r   r!   r   )+sys	itertoolsr   typingr   r   r   version_infor   Ztyping_extensionsZ	constrainr
   Zjupyterr   rC   r   segmentr   r!   r   r0   r   r   r   r   rZ   r[   r   r_   rV   Zrich.consolerd   Zrich.highlighterre   Z
rich.panelrf   Zhighlighterr   r   r   Zpanelprintr   r&   r&   r&   r'   <module>   sH   
 T8
