File: //lib/python3.6/site-packages/awscli/__pycache__/formatter.cpython-36.opt-1.pyc
3
�T_�+ � @ 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 je�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
� /usr/lib/python3.6/formatter.py�is_response_paginated s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� Formatterc C s
|| _ d S )N)�_args)�self�argsr
r
r �__init__" s zFormatter.__init__c C s> d|kr:d|kr:d|d kr4|d d }t jd|� |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 j� S )N)r Zget_stdout_text_writer)r r
r
r �_get_default_stream1 s z Formatter._get_default_streamc C s& y |j � W n tk
r Y nX d S )N)�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| j � }t|�r"|j� }n|}| j|� | jjd k rJ| jjj|�}z:y| j|||� W n" tk
r� } zW Y d d }~X nX W d | j |� X d S )N)
r r Zbuild_full_resultr r �query�search�_format_responser r )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� |jd� 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| �j|� |jdkr*tddd�| _nR|jdkrLt� }tdd|d�| _n0|jdkrnt� }tdd|d�| _ntd|j ��d S ) N�autoF�|)�initial_section�column_separatorZoff)r2 r3 �stylerZonzUnknown color option: %s) �superr/ r Zcolorr �tabler r �
ValueError)r r r6 r4 )� __class__r
r r l s
zTableFormatter.__init__c C s6 | j ||�r2y| jj|� W n tk
r0 Y nX d S )N)� _build_tabler6 Zrenderr )r r% r r 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� j|||� nZxX|D ]P}� j|�rl� j j|g� qNt� fdd�|D ��r�� j j|� qN� j d |d� qNW t|t�r�� j
||� dS )NF)� indent_levelr c 3 s | ]}� j |�V qd S )N)� _scalar_type)�.0Zel)r r
r � <genexpr>� s z.TableFormatter._build_table.<locals>.<genexpr>)�title�currentT) r6 � new_sectionr �list�dict�_build_sub_table_from_listr; �add_row�allr9 �_build_sub_table_from_dict)r r> r? r: �itemr
)r r r9 � s
zTableFormatter._build_tablec s� | j � �\}}t|�dkr8| jj|d � |d g� n*|rb| jj|� | jj� fdd�|D �� x$|D ] }| j|� | |d d� qhW d S )N� r c s g | ] }� | �qS r
r
)r< �k)r? r
r �
<listcomp>� s z=TableFormatter._build_sub_table_from_dict.<locals>.<listcomp>)r: )�_group_scalar_keys�lenr6 rD �add_row_headerr9 )r r? r: �headers�more� remainingr
)r? r rF � s
z)TableFormatter._build_sub_table_from_dictc s� | j |�\}}| jj|� d}x~|D ]v� | rN|rN| jj||d� | jj|� d}| jj� fdd�|D �� x,|D ]$}|� krr| j|� | |d d� qrW q$W d S )NT)r: Fc s g | ]}� j |d ��qS )� )�get)r< �header)�elementr
r rJ � s z=TableFormatter._build_sub_table_from_list.<locals>.<listcomp>rH )� _group_scalar_keys_from_listr6 rM r@ rD r9 )r r? r: r> rN rO �firstrP r
)rT r rC � s
z)TableFormatter._build_sub_table_from_listc C s t |ttf� S )N)r rA rB )r rT r
r
r r; � s zTableFormatter._scalar_typec C s\ t � }t � }x.|D ]&}| j|�\}}|j|� |j|� qW tt|��}tt|��}||fS )N)�setrK �updaterA �sorted)r Z
list_of_dictsrN rO rG Zcurrent_headersZ current_morer
r
r rU � s
z+TableFormatter._group_scalar_keys_from_listc C sR g }g }x0|D ](}| j || �r,|j|� q|j|� qW |j� |j� ||fS )N)r; �append�sort)r r? rO rN rT r
r
r rK � s
z!TableFormatter._group_scalar_keys)N)r )
r r r �__doc__r r$ r9 rF rC r; rU rK �
__classcell__r
r
)r8 r r/ d s
r/ c @ s e Zd Zddd�Zdd� ZdS )�
TextFormatterNc
C s� |d kr| j � }z�t|�r�|j}xZt|�D ]N\}}|dkr@i }n|j}x$|D ] }|j|�} t||j| � qLW | j||� q*W |j r�| jdd|j ii|� n| j
|� | j||� W d | j |� X d S )Nr Z NextToken) r r � result_keys� enumerateZnon_aggregate_partr# r �
expressionr$ Z resume_tokenr r )
r r% r r r_ �iZpager? Z
result_key�datar
r
r r' � s.
zTextFormatter.__call__c C s. | j jd k r | j j}|j|�}tj||� d S )N)r r" r# r Z format_text)r r r ra r
r
r r$ s
z TextFormatter._format_response)N)r r r r' r$ r
r
r
r r^ � s
"r^ c C s@ | dkrt |�S | dkr t|�S | dkr0t|�S td| ��d S )Nr r r6 zUnknown output type: %s)r( r^ r/ r7 )Z format_typer r
r
r �
get_formatter
s rd )ZloggingZbotocore.compatr Zbotocore.utilsr Zbotocore.paginater Z awscli.tabler r r Zawsclir r Z awscli.utilsr
Z getLoggerr r r �objectr r! r( r/ r^ rd r
r
r
r �<module>
s
~+