File: //lib/python3/dist-packages/awscli/__pycache__/formatter.cpython-38.pyc
U
>�^�+ � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z m
Z
d dl m Z d dl m
Z
d dlmZ e �e�Zd d
� ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�Zdd� ZdS )� N)�json)�set_value_from_jmespath)� PageIterator)�
MultiTable�Styler�ColorizedStyler)�text)�compat)� json_encoderc C s
t | t�S �N)�
isinstancer )�response� r �2/usr/lib/python3/dist-packages/awscli/formatter.py�is_response_paginated s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� Formatterc C s
|| _ d S r )�_args)�self�argsr r r �__init__" s zFormatter.__init__c C s> d|kr:d|kr:d|d kr4|d d }t �d|� |d= d S )NZErrorsZResponseMetadataZ RequestIdz
RequestId: %s)�LOG�debug)r �
response_dataZ
request_idr r r �_remove_request_id% s z Formatter._remove_request_idc C s t �� S r )r Zget_stdout_text_writer�r r r r �_get_default_stream1 s z Formatter._get_default_streamc C s& z |� � W n tk
r Y nX d S r )�flush�IOError)r �streamr r r �
_flush_stream4 s zFormatter._flush_streamN)�__name__�
__module__� __qualname__r r r r r r r r r ! s r c @ s e Zd Zddd�ZdS )�FullyBufferedFormatterNc
C s� |d kr| � � }t|�r"|�� }n|}| �|� | jjd k rJ| jj�|�}z:z| �|||� W n" t k
r� } zW 5 d }~X Y nX W 5 | �|� X d S r )
r r Zbuild_full_resultr r �query�searchr �_format_responser )r � command_namer
r r �er r r �__call__< s
z FullyBufferedFormatter.__call__)N)r r! r" r) r r r r r# ; s r# c @ s e Zd Zdd� ZdS )�
JSONFormatterc C s* |i kr&t j||dtdd� |�d� d S )N� F)�indent�defaultZ ensure_ascii�
)r �dumpr
�write�r r' r
r r r r r& Y s
�z JSONFormatter._format_responseN)r r! r" r& r r r r r* W s r* c s\ e Zd ZdZd� fdd� Zdd� Zddd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z � Z S )�TableFormatterz�Pretty print a table from a given response.
The table formatter is able to take any generic response
and generate a pretty printed table. It does this without
using the output definition from the model.
Nc s� t t| ��|� |jdkr*tddd�| _nR|jdkrLt� }tdd|d�| _n0|jdkrnt� }tdd|d�| _ntd|j ��d S ) N�autoF�|)�initial_section�column_separatorZoff)r5 r6 �stylerZonzUnknown color option: %s) �superr2 r Zcolorr �tabler r �
ValueError)r r r9 r7 �� __class__r r r l s$
�
�
�
zTableFormatter.__init__c C s6 | � ||�r2z| j�|� W n tk
r0 Y nX d S r )� _build_tabler9 Zrenderr r1 r r r r& | s
z TableFormatter._format_responser c s� |sdS |d k r � j j||d� t|t�r�t|d t�rH� �|||� nV|D ]P}� �|�rj� j �|g� qLt� fdd�|D ��r�� j �|� qL� j d |d� qLt|t�r�� �
||� dS )NF�� indent_levelr c 3 s | ]}� � |�V qd S r )� _scalar_type)�.0Zelr r r � <genexpr>� s z.TableFormatter._build_table.<locals>.<genexpr>)�title�currentT) r9 � new_sectionr �list�dict�_build_sub_table_from_listr@ �add_row�allr= �_build_sub_table_from_dict)r rC rD r? �itemr r r r= � s
zTableFormatter._build_tablec s� | � � �\}}t|�dkr8| j�|d � |d g� n*|rb| j�|� | j�� fdd�|D �� |D ] }| j|� | |d d� qfd S )N� r c s g | ] }� | �qS r r )rA �k�rD r r �
<listcomp>� s z=TableFormatter._build_sub_table_from_dict.<locals>.<listcomp>r> )�_group_scalar_keys�lenr9 rI �add_row_headerr= )r rD r? �headers�more� remainingr rO r rK � s �z)TableFormatter._build_sub_table_from_dictc s� | � |�\}}| j�|� d}|D ]p� |sJ|rJ| jj||d� | j�|� d}| j�� fdd�|D �� |D ]$}|� krl| j|� | |d d� qlq"d S )NTr> Fc s g | ]}� � |d ��qS )� )�get)rA �header��elementr r rP � s z=TableFormatter._build_sub_table_from_list.<locals>.<listcomp>rM )� _group_scalar_keys_from_listr9 rS rE rI r= )r rD r? rC rT rU �firstrV r rZ r rH � s � �z)TableFormatter._build_sub_table_from_listc C s t |ttf� S r )r rF rG )r r[ r r r r@ � s zTableFormatter._scalar_typec C sX t � }t � }|D ]&}| �|�\}}|�|� |�|� qtt|��}tt|��}||fS r )�setrQ �updaterF �sorted)r Z
list_of_dictsrT rU rL Zcurrent_headersZ current_morer r r r\ � s
z+TableFormatter._group_scalar_keys_from_listc C sN g }g }|D ](}| � || �r*|�|� q |�|� q |�� |�� ||fS r )r@ �append�sort)r rD rU rT r[ r r r rQ � s z!TableFormatter._group_scalar_keys)N)r )
r r! r" �__doc__r r& r= rK rH r@ r\ rQ �
__classcell__r r r; r r2 d s
r2 c @ s e Zd Zddd�Zdd� ZdS )�
TextFormatterNc
C s� |d kr| � � }z�t|�r�|j}t|�D ]J\}}|dkr>i }n|j}|D ] }|�|�} t||j| � qH| � ||� q(|j
r�| � dd|j
ii|� n| � |� | � ||� W 5 | �|� X d S )Nr Z NextToken) r r r � result_keys� enumerateZnon_aggregate_partr% r �
expressionr& Z resume_tokenr )
r r' r
r rf �iZpagerD Z
result_key�datar r r r) � s2
� �
zTextFormatter.__call__c C s. | j jd k r | j j}|�|�}t�||� d S r )r r$ r% r Z format_text)r r
r rh r r r r& s
z TextFormatter._format_response)N)r r! r" r) r& r r r r re � s
"re c C s@ | dkrt |�S | dkr t|�S | dkr0t|�S td| ��d S )Nr r r9 zUnknown output type: %s)r* re r2 r: )Z format_typer r r r �
get_formatter
s rk )ZloggingZbotocore.compatr Zbotocore.utilsr Zbotocore.paginater Z awscli.tabler r r Zawsclir r Z awscli.utilsr
Z getLoggerr r r �objectr r# r* r2 re rk r r r r �<module>
s
~+