3
\|Q �
@ s� d Z ddlZddlZddlZddlZddlm Z
dddddd gZ d!d d
�dd�Z d"d d
�dd�Z
dd� Zdd� Zdd� ZG dd� d�Zdd� ZG dd � d �Zdd� Zeeeeeeee e d�h�Z dd� Z d#d d �Z d d � Z!e"d kr�e � dS )$a/ Support to pretty-print lists, tuples, & dictionaries recursively.
Very simple, but useful, especially in debugging data structures.
Classes
-------
PrettyPrinter()
Handle pretty-printing operations onto a stream using a configured
set of formatting parameters.
Functions
---------
pformat()
Format a Python object into a pretty-printed representation.
pprint()
Pretty-print a Python object to a stream [default is sys.stdout].
saferepr()
Generate a 'standard' repr()-like value, but protect against recursive
data structures.
� N)�StringIO�pprint�pformat�
isreadable� isrecursive�saferepr�
PrettyPrinter� �P F)�compactc C s t |||||d�}|j| � dS )zAPretty-print a Python object to a stream [default is sys.stdout].)�stream�indent�width�depthr N)r r )�objectr r
r r r Zprinter� r � /usr/lib64/python3.6/pprint.pyr / s c C s t ||||d�j| �S )z= 0Nzdepth must be > 0zwidth must be != 0)
�int�
ValueError�_depth�_indent_per_level�_width�_stream�_sys�stdout�bool�_compact)r r
r r r r r r r r c s zPrettyPrinter.__init__c C s& | j || jddi d� | jjd� d S )Nr �
)�_formatr+ �write)r r r r r r � s zPrettyPrinter.pprintc C s" t � }| j||ddi d� |j� S )Nr )� _StringIOr1 �getvalue)r r Zsior r r r � s zPrettyPrinter.pformatc C s | j |i dd�d S )Nr r )�format)r r r r r r � s zPrettyPrinter.isrecursivec C s | j |i dd�\}}}|o | S )Nr )r5 )r r �s�readable� recursiver r r r � s zPrettyPrinter.isreadablec C s� t |�}||kr.|jt|�� d| _d| _d S | j|||�}| j| | } t|�| kr�| jj t
|�j d �}
|
d k r�d||< |
| ||||||d � ||= d S t |t
�r�d||< | j||||||d � ||= d S |j|� d S )NTFr )r r2 �
_recursion�
_recursive� _readable�_reprr* �len� _dispatch�getr �__repr__�
isinstance�dict� _pprint_dict) r r r r
� allowance�context�level�objid�rep� max_width�pr r r r1 � s,
zPrettyPrinter._formatc
C sj |j }|d� | jdkr*|| jd d � t|�}|r^t|j� td�} | j| |||d ||� |d� d S )N�{r � )�key�})r2 r) r= �sorted�itemsr% �_format_dict_items)
r r r r
rD rE rF r2 ZlengthrP r r r rC � s
zPrettyPrinter._pprint_dictc C sl t |�s|jt|�� d S |j}|j|jd � | jt|j� �||t |j� d |d ||� |jd� d S )N�(r �))r= r2 �repr� __class__r r1 �listrP )r r r r
rD rE rF �clsr r r �_pprint_ordered_dict� s z"PrettyPrinter._pprint_ordered_dictc C s0 |j d� | j||||d ||� |j d� d S )N�[r �])r2 �
_format_items)r r r r
rD rE rF r r r � _pprint_list� s
zPrettyPrinter._pprint_listc C sH |j d� t|�dkrdnd}| j||||t|� ||� |j |� d S )NrR r z,)rS )r2 r= r[ )r r r r
rD rE rF �endcharr r r �
_pprint_tuple� s
zPrettyPrinter._pprint_tuplec C s� t |�s|jt|�� d S |j}|tkr8|jd� d}n&|j|jd � d}|t |j�d 7 }t|td�}| j||||t |� ||� |j|� d S )NrK rN z({z})r )rM ) r= r2 rT rU �setr rO r r[ ) r r r r
rD rE rF �typr] r r r � _pprint_set� s
zPrettyPrinter._pprint_setc C s� |j }t|�s |t|�� d S g }|jd�} |dkrD|d7 }|d7 }| j| }
} x�t| �D ]�\} }
t|
�}| t| �d kr�|
|8 }
t|�|
kr�|j|� q\tjd|
�}|j � | }d}xtt|�D ]h\}}|| }|t|�d kr�| t| �d kr�||8 }tt|��|k�r&|�r |jt|�� |}q�|}q�W |r\|jt|�� q\W t|�dk�r^||� d S |dk�rp|d� x8t|�D ],\} }| dk�r�|dd| � ||� �qzW |dk�r�|d � d S )
NTr z\S*\s*� rR r r0 rL rS )
r2 r= rT �
splitlinesr* � enumerate�append�re�findall�pop)r r r r
rD rE rF r2 Zchunks�linesZ
max_width1rI �i�linerH �partsZ
max_width2�current�j�part� candidater r r � _pprint_str� sT
zPrettyPrinter._pprint_strc C s� |j }t|�dkr"|t|�� d S |dk}|rF|d7 }|d7 }|d� d} x:t|| j| |�D ]$}
|| � ||
� | s^dd| } q^W |r�|d� d S )N� r rR rb r0 rL rS )r2 r= rT �_wrap_bytes_reprr* ) r r r r
rD rE rF r2 Zparens�delimrH r r r �
_pprint_bytes' s" zPrettyPrinter._pprint_bytesc C s> |j }|d� | jt|�||d |d ||d � |d� d S )Nz
bytearray(�
r rS )r2 ru �bytes)r r r r
rD rE rF r2 r r r �_pprint_bytearray< s
z PrettyPrinter._pprint_bytearrayc C s8 |j d� | j|j� ||d |d ||� |j d� d S )Nz
mappingproxy(�
r rS )r2 r1 �copy)r r r r
rD rE rF r r r �_pprint_mappingproxyE s
z"PrettyPrinter._pprint_mappingproxyc C s� |j }|| j7 }dd| }t|�d } xrt|�D ]f\}
\} } |
| k}
| j| ||�}||� |d� | j| ||t|� d |
r�|nd||� |
s2||� q2W d S )Nz,
rL r z: r )r2 r) r= rd r< r1 )r rP r r
rD rE rF r2 �delimnlZ
last_indexrj rM �ent�lastrH r r r rQ M s
z PrettyPrinter._format_dict_itemsc C sN |j }|| j7 }| jdkr,|| jd d � dd| }d} | j| d }
} t|�} y t| �}
W n tk
rv d S X d}x�|�sH|
}y t| �}
W n( tk
r� d}| |8 } |
|8 }
Y nX | j�r | j|||�}t|�d }|
|k r�| }
| r�|} |
|k�r |
|8 }
|| � d} ||� q~|| � |} | j ||||�r<|nd||� q~W d S ) Nr rL z,
rb FTr z, )
r2 r) r* �iter�next�
StopIterationr/ r< r= r1 )r rP r r
rD rE rF r2 r| rt r rI �itZnext_entr~ r} rH �wr r r r[ ^ sN
zPrettyPrinter._format_itemsc C s4 | j ||j� | j|�\}}}|s&d| _|r0d| _|S )NFT)r5 rz r( r; r: )r r rE rF rT r7 r8 r r r r< � s zPrettyPrinter._reprc C s t ||||�S )z�Format object for a specific context, returning a string
and flags indicating whether the representation is 'readable'
and whether the object represents a recursive construct.
)r )r r rE � maxlevelsrF r r r r5 � s zPrettyPrinter.formatc C s� t |�s|jt|�� d S | j|j||�}|j}|t |j�d 7 }|jd|j|d| f � | j||||d ||� |jd� d S )Nr z %s(%s,
%srL rS )r= r2 rT r< �default_factoryrU r rC ) r r r r
rD rE rF ZrdfrW r r r �_pprint_default_dict� s z"PrettyPrinter._pprint_default_dictc C s� t |�s|jt|�� d S |j}|j|jd � | jdkrN|j| jd d � |j� }| j|||t |j� d |d ||� |jd� d S )Nz({r rL r z}))r= r2 rT rU r r) � most_commonrQ ) r r r r
rD rE rF rW rP r r r �_pprint_counter� s
z PrettyPrinter._pprint_counterc
C s� t |j�s |jt|�� d S |j}|j|jd � |t |j�d 7 }xrt|j�D ]d\}} |t |j�d kr�| j| |||d ||� |jd� qP| j| ||d||� |jdd| � qPW d S )NrR r rS z,
rL )r= �mapsr2 rT rU r rd r1 )
r r r r
rD rE rF rW rj �mr r r �_pprint_chain_map� s
z PrettyPrinter._pprint_chain_mapc C s� t |�s|jt|�� d S |j}|j|jd � |t |j�d 7 }|jd� |jd krz| j||||d ||� |jd� n:| j|||d||� | j|j||�}|jdd| |f � d S )NrR r rY r z])z],
%smaxlen=%s)rL )r= r2 rT rU r �maxlenr[ r< ) r r r r
rD rE rF rW Zrmlr r r �
_pprint_deque� s
zPrettyPrinter._pprint_dequec C s | j |j|||||d � d S )Nr )r1 �data)r r r r
rD rE rF r r r �_pprint_user_dict� s z PrettyPrinter._pprint_user_dictc C s | j |j|||||d � d S )Nr )r1 r� )r r r r
rD rE rF r r r �_pprint_user_list� s z PrettyPrinter._pprint_user_listc C s | j |j|||||d � d S )Nr )r1 r� )r r r r
rD rE rF r r r �_pprint_user_string� s z!PrettyPrinter._pprint_user_string)r r
NN)2r r r! r r r r r r1 r> rC rB r@ rX � _collections� OrderedDictr\ rV r^ �tuplera r_ � frozensetrq r ru rw rx � bytearrayr{ �_types�MappingProxyTyperQ r[ r< r5 r� � defaultdictr� �Counterr� �ChainMapr� �dequer� �UserDictr� �UserListr� �
UserStringr r r r r b sZ &
1
) c C s� t | �}|tkr t| �ddfS t|dd �}t|t�o<|tjk�r8| sHdS t| �}|rj||krjdd||kfS ||kr�t| �ddfS d||< d}d}g } | j }
|d7 }t
} t | j � t
d�} xf| D ]^\}
}| |
|||�\}}}| ||||�\}}}|
d||f � |�o
|�o
|}|�s|r�d}q�W ||= d d
j| � ||fS t|t��rP|tjk�sht|t��rd|tjk�rdt|t��r�| �s~dS d }n"t| �dk�r�d
}n| �s�dS d}t| �}|�r�||k�r�|d d||kfS ||k�r�t| �ddfS d||< d}d}g } | j }
|d7 }x>| D ]6}t
||||�\}}}|
|� |�s:d}|�rd}�qW ||= |d
j| � ||fS t| �}||�o~|jd� dfS )NTFr@ �{}z{...}r )rM z%s: %sz{%s}z, �[]z[%s]z(%s,)�()z(%s)z...�<)r� TF)r� TF)r� TF)r �_builtin_scalarsrT �getattr�
issubclassrB r@ r r9 re r rO rP r% �joinrV r� r= �
startswith)r rE r� rF r` �rrG r7 r8 Z
componentsre r rP �k�vZkreprZ kreadableZkrecurZvreprZ vreadableZvrecurr5 �oZoreprZ oreadableZorecurrH r r r r � s|
r c C s dt | �jt| �f S )Nz )r r r )r r r r r9 1 s r9 c C s� dd l }| d kr,dd
ddgddd �fgd
} t� }|j � }t| i d d� |j � }|j| � |j � }td || � td || � d S )Nr �stringr r � rr � � )� � i�� z _safe_repr:zpformat:)r r )�timer r r �print)r r� rJ Zt1Zt2Zt3r r r �
_perfcheck6 s
r� c c s� d}t | �d d }xhtdt | �d�D ]T}| ||d � }|| }||krR||8 }t t|��|krv|rpt|�V |}q&|}q&W |r�t|�V d S )N� rr r )r= �rangerT )r r rD rm r~ rj ro rp r r r rs C s
rs �__main__)Nr r
N)r r
N)N)#r" � collectionsr� rf �sysr, �typesr� �ior r3 �__all__r r r r r r r% r r r� r rw r� r& �float�complexr. r r� r9 r� rs r r r r r �# s6
D