File: //lib/python3/dist-packages/uaclient/cli/__pycache__/formatter.cpython-38.pyc
U
��Jh�" � @ s. d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
m Z d dl m
Z
d dlmZ dZdZdZd d
d d �ZG d
d� de�ZG dd� d�Zeje
� d� eeed�dd�Zeed�dd�Zdd� Zeed�dd�Z eee
e d �d d �Z G d d � d e j �Z G d!d"� d"e �Z G d#d$� d$e �Z!dS )%� N)�Enum)�Any�Dict�List�Optional)�UAConfig)�TxtColorz
\033\[.*?mz\033]8;;.+?\033\\+z]8;;\�-�x�*)u —u ✘u ✔c @ s e Zd ZdZdZdS )�ContentAlignment�l�rN)�__name__�
__module__� __qualname__�LEFT�RIGHT� r r �8/usr/lib/python3/dist-packages/uaclient/cli/formatter.pyr s r c @ s8 e Zd ZdZdZeed�dd��Zedd�dd��ZdS ) �ProOutputFormatterConfigT��cfgc C s< t jjd k odt jj�� k| _t j�� o4t�d�d k| _d S )NzUTF-8ZNO_COLOR) �sys�stdout�encoding�upper�use_utf8�isatty�os�getenv� use_color)�clsr r r r �init! s �z ProOutputFormatterConfig.initN��returnc C s
d| _ d S )NF)r! )r" r r r �
disable_color* s z&ProOutputFormatterConfig.disable_color) r r r r r! � classmethodr r# r& r r r r r s r r )�text�urlr% c C s dj || d�S )Nz]8;;{url}\{text}]8;;\)r) r( )�format)r( r) r r r � create_link2 s r+ )r( r% c C s0 t �td| �}t �td|�}|�td�}t|�S )N� )�re�sub�COLOR_FORMATTING_PATTERN�LINK_START_PATTERN�replace�LINK_END�len)r( �resultr r r �real_len6 s r5 c C s t j�� rt�� jS dS )Ni� )r r r r �get_terminal_size�columnsr r r r �_get_default_length@ s
r8 c C st | }t jst�td| �}t jsLt�� D ]\}}|�||�}q&|� dd��
� }t j �
� spt�td|�}|�td�}|S )Nr, �ascii�ignore)r r! r- r. r/ r �UTF8_ALTERNATIVES�itemsr1 �encode�decoder r r r0 r2 )r( �output�charZ alternativer r r �process_formatter_configH s
rA )r( � max_widthr% c C sv t | �|kr| gS | �� }g }d}|D ]8}t |�t |� |krR|�|�� � |}q&|d| 7 }q&|rr|�|�� � |S )Nr, � )r5 �split�append�strip)r( rB ZwordsZ
wrapped_linesZ current_lineZwordr r r � wrap_text[ s rG c @ s0 e Zd Zejdee ed�dd��Zdd� Z dS )�ProOutputFormatterN�� line_lengthr% c C s d S �Nr )�selfrJ r r r � to_stringq s z ProOutputFormatter.to_stringc C s | � � S rK )rM �rL r r r �__str__u s zProOutputFormatter.__str__)N)
r r r �abc�abstractmethodr �int�strrM rO r r r r rH p s rH c @ s� e Zd ZdZdeee eeee eee d�dd�Ze ee
ed�dd��Z e ee
ed�d d
��Z ee
d �d d
�Z
dee
ed�dd�Ze
d �dd�Ze
eee d�dd�Zee ed�dd�ZdS )�Tablez N)�headers�rows� alignmentc C s� |d k r |ng | _ |d k r |ng | _| �� | _|d k r:|ntjgt| j� | _t| j�t| j�krvtd� t| j����| jd | _
d S )Nz&'alignment' list should have length {}���) rU rV �_get_column_sizes� column_sizesr r r3 rW �
ValueErrorr* �last_column_size)rL rU rV rW r r r �__init__| s
����zTable.__init__)�string� total_lengthr% c C s$ t | �}||kr| S | d|| S �NrC �r5 �r^ r_ Z
str_lengthr r r �ljust� s z Table.ljustc C s$ t | �}||kr| S d|| | S r` ra rb r r r �rjust� s z Table.rjustr$ c s� | j s| jstd��| jr6tdd� | jD ��r6td��g }| j rL|�| j � | jr^|�| j� t|d �� t� fdd�|D ��s�td��g }tt|d ��D ] �|�t �fdd�|D ��� q�|S ) Nz:Empty table not supported. Please provide headers or rows.c s s | ]}t |�d kV qdS )r N�r3 ��.0�itemr r r � <genexpr>� s z*Table._get_column_sizes.<locals>.<genexpr>z=Empty row not supported. Please provide content for each row.r c 3 s | ]}t |�� kV qd S rK re rf )�expected_lengthr r ri � s zQMixed lengths in table content. Please provide headers / rows of the same length.c 3 s | ]}t t|� ��V qd S rK )r5 rS rf )�ir r ri � s )
rU rV r[ �anyrE �extendr3 �all�range�max)rL Z all_contentrZ r )rj rk r rY � s0 �� ��zTable._get_column_sizesrI c C s| |d krt � }| j}| �� |kr*| �|�}d}| jrT|tj| �| j� tj d 7 }|D ]}|| �|�7 }|d7 }qXt |�S )Nr, �
)
r8 rV �_get_line_length�wrap_last_columnrU r �BOLD� _fill_row�ENDCrA )rL rJ rV r? �rowr r r rM � s(
����
zTable.to_stringc C s" t | j�t| j�d t| j� S )N� )�sumrZ r3 � SEPARATORrN r r r rr � s �zTable._get_line_length)�
max_lengthr% c C s� |t | jd d� �t| j�d t| j� | _g }| jD ]�}t|d �| jkr\|�|� q:t|d | j�}|�|d d� |d g � |dd � D ](}dgt| j�d |g }|�|� q�q:|S )NrX rx r rC )ry rZ r3 rz r\ rV rE rG )rL r{ Znew_rowsrw Zwrapped_last_columnZ
extra_lineZnew_rowr r r rs � s" ��
�zTable.wrap_last_column)rw r% c C s� d}t t|�d �D ]f}| j| tjkrJ|| �|| | j| �| j 7 }q| j| tjkr|| � || | j| �| j 7 }q| jd tjkr�||d 7 }n&| jd tjkr�|| � |d | j
�7 }|S )Nr, rx rX ) ro r3 rW r r rc rZ rz r rd r\ )rL rw r? rk r r r ru � s ��zTable._fill_row)NNN)N)r r r rz r r rS r r] � staticmethodrR rc rd rY rM rr rs ru r r r r rT y s$ �
� rT c @ sH e Zd ZdZdZd
ee eee d�dd�Z d ee
ed�dd �Z dS ) �Block� rC N��title�contentc C s || _ |d k r|ng | _d S rK r )rL r� r� r r r r] s zBlock.__init__rI c C s� |d krt � }|| j8 }d}| jrB|tjtj | j tj d 7 }| jD ]L}t|t �rd|j
|d�}nd� t t
|�|��d }|t�|| j| j �7 }qHt|�S )Nr, rq )rJ )r8 � INDENT_SIZEr� r rt Z DISABLEGREYrv r� �
isinstancerH rM �joinrG rS �textwrap�indent� INDENT_CHARrA )rL rJ r? rh Zitem_strr r r rM s2
�����
�zBlock.to_string)NN)N) r r r r� r� r rS r r r] rR rM r r r r r} � s �
� r} )"rP r r- r r� �enumr �typingr r r r Zuaclient.configr Zuaclient.messagesr r/ r0 r2 r; r r r# rS r+ rR r5 r8 rA rG �ABCrH rT r} r r r r �<module> s6 �