CasperSecurity
o
��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� dededefdd�Zdedefdd�Zdd� Zdedefdd �Z ded ede
e fd 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 @ s4 e Zd ZdZdZedefdd��Zed dd��ZdS )
�ProOutputFormatterConfigT�cfgc C s< t jjd uo
dt jj�� v | _t j�� ot�d�d u | _d S )NzUTF-8�NO_COLOR) �sys�stdout�encoding�upper�use_utf8�isatty�os�getenv� use_color)�clsr r r r �init! s � z ProOutputFormatterConfig.init�returnNc C s
d| _ d S )NF)r! )r" r r r �
disable_color* s
z&ProOutputFormatterConfig.disable_color)r$ N) 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* 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 r4 c C s t j�� r
t�� jS dS )Ni� )r r r r �get_terminal_size�columnsr r r r �_get_default_length@ s
r7 c C st | }t js t�td| �}t js&t�� D ]
\}}|�||�}q|� dd��
� }t j �
� s8t�td|�}|�td�}|S )Nr+ �ascii�ignore)r r! r, r- r. r �UTF8_ALTERNATIVES�itemsr0 �encode�decoder r r r/ r1 )r' �output�char� alternativer r r �process_formatter_configH s
rA � max_widthc C sv t | �|kr | gS | �� }g }d}|D ] }t |�t |� |kr)|�|�� � |}q|d| 7 }q|r9|�|�� � |S )Nr+ � )r4 �split�append�strip)r' rB �words�
wrapped_lines� current_line�wordr r r � wrap_text[ s rK c @ s2 e Zd Zejddee defdd��Zdd� Z dS ) �ProOutputFormatterN� line_lengthr$ c C s d S �Nr )�selfrM r r r � to_stringq s z ProOutputFormatter.to_stringc C s | � � S rN )rP �rO r r r �__str__u s zProOutputFormatter.__str__rN )
r r r �abc�abstractmethodr �int�strrP rR r r r r rL p s rL c @ s� e Zd ZdZ d deee deeee deee fdd�Ze ded e
d
efd d ��Z e ded e
d
efd
d��Z d
ee
fdd�Z
d dee
d
efdd�Zd
e
fdd�Zde
d
eee fdd�Zdee d
efdd�ZdS ) �Tablez N�headers�rows� alignmentc C s� |d ur|ng | _ |d ur|ng | _| �� | _|d ur |ntjgt| j� | _t| j�t| j�kr;td� t| j����| jd | _
d S )Nz&'alignment' list should have length {}���) rX rY �_get_column_sizes� column_sizesr r r2 rZ �
ValueErrorr) �last_column_size)rO rX rY rZ r r r �__init__| s
����zTable.__init__�string� total_lengthr$ c C s$ t | �}||kr
| S | d|| S �NrC �r4 �ra rb �
str_lengthr r r �ljust� � z Table.ljustc C s$ t | �}||kr
| S d|| | S rc rd re r r r �rjust� rh z Table.rjustc s� | j s
| js
td��| jrtdd� | jD ��rtd��g }| j r&|�| j � | jr/|�| j� t|d �� t� fdd�|D ��sDtd��g }tt|d ��D ]�|�t �fdd�|D ��� qN|S ) Nz:Empty table not supported. Please provide headers or rows.c s s � | ] }t |�d kV qdS )r N�r2 ��.0�itemr r r � <genexpr>� � � 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 rN rj rk )�expected_lengthr r rn � ro zQMixed lengths in table content. Please provide headers / rows of the same length.c 3 s � | ] }t t|� ��V qd S rN )r4 rV rk )�ir r rn � s � )
rX rY r^ �anyrE �extendr2 �all�range�max)rO � all_contentr] r )rp rq r r\ � s0 �� ��zTable._get_column_sizesrM c C s| |d u rt � }| j}| �� |kr| �|�}d}| jr*|tj| �| j� tj d 7 }|D ]
}|| �|�7 }|d7 }q,t |�S )Nr+ �
)
r7 rY �_get_line_length�wrap_last_columnrX r �BOLD� _fill_row�ENDCrA )rO rM rY r> �rowr r r rP � s(
����
zTable.to_stringc C s" t | j�t| j�d t| j� S )N� )�sumr] r2 � SEPARATORrQ r r r ry � s �zTable._get_line_length�
max_lengthc C s� |t | jd d� �t| j�d t| j� | _g }| jD ]B}t|d �| jkr.|�|� q t|d | j�}|�|d d� |d g � |dd � D ]}dgt| j�d |g }|�|� qJq |S )Nr[ r r rC )r� r] r2 r� r_ rY rE rK )rO r� �new_rowsr~ �wrapped_last_column�
extra_line�new_rowr r r rz � s$ ��
� �zTable.wrap_last_columnr~ c C s� d}t t|�d �D ]3}| j| tjkr%|| �|| | j| �| j 7 }q
| j| tjkr=|| � || | j| �| j 7 }q
| jd tjkrN||d 7 }|S | jd tjkra|| � |d | j
�7 }|S )Nr+ r r[ ) ru r2 rZ r r rg r] r� r ri r_ )rO r~ r> rq r r r r| � s" ��� �zTable._fill_row)NNNrN )r r r r� r r rV r r` � staticmethodrU rg ri r\ rP ry rz r| r r r r rW y s* �
��
� rW c @ sP e Zd ZdZdZ d dee deee fdd�Z d
dee
d efd
d �Z dS )�Block� rC N�title�contentc C s" || _ |d ur || _d S g | _d S rN )r� r� )rO r� r� r r r r` s zBlock.__init__rM r$ c C s� |d u rt � }|| j8 }d}| jr!|tjtj | j tj d 7 }| jD ]&}t|t �r2|j
|d�}n d� t t
|�|��d }|t�|| j| j �7 }q$t|�S )Nr+ rx )rM )r7 � INDENT_SIZEr� r r{ � DISABLEGREYr} r� �
isinstancerL rP �joinrK rV �textwrap�indent� INDENT_CHARrA )rO rM r> rm �item_strr r r rP s0
�����
�zBlock.to_string)NNrN ) r r r r� r� r rV r r r` rU rP r r r r r� � s ��
� r� )"rS r r, r r� �enumr �typingr r r r �uaclient.configr �uaclient.messagesr r. r/ r1 r: r r r# rV r* rU r4 r7 rA rK �ABCrL rW r� r r r r �<module> s8 �