U
e5d�S �
@ s d Z ddlZddlZddlZddlZddlm Z
dddddd d
gZ d%d
dd�dd�Z d&d
dd�dd�Z
d
d�dd
�Zdd� Zdd� Zdd� ZG dd� d�Zdd� ZG dd � d �Zd d � Zeeeeeee e e d�h�Z d d � Z d'd d!�Z!d"d#� Z"e#d$k� r�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�pp� �P FT��compact�
sort_dictsc C s" t ||||||d�}|�| � dS )zAPretty-print a Python object to a stream [default is sys.stdout].)�stream�indent�width�depthr
r N)r r )�objectr r r r r
r Zprinter� r � /usr/lib64/python3.8/pprint.pyr / s �c C s t |||||d��| �S )z= 0Nzdepth must be > 0zwidth must be != 0) �int�
ValueError�_depth�_indent_per_level�_width�_stream�_sys�stdout�bool�_compact� _sort_dicts)r r r r r r
r r r r r h s
zPrettyPrinter.__init__c C s& | � || jddi d� | j�d� d S )Nr �
)�_formatr3 �write�r r r r r r � s zPrettyPrinter.pprintc C s" t � }| �||ddi d� |�� S �Nr )� _StringIOr: �getvalue)r r Zsior r r r � s zPrettyPrinter.pformatc C s | � |i dd�d S )Nr r ��formatr< r r r r � s zPrettyPrinter.isrecursivec C s | � |i dd�\}}}|o | S r= r@ )r r �s�readable� recursiver r r r � s zPrettyPrinter.isreadablec C s� t |�}||kr.|�t|�� d| _d| _d S | �|||�}| j| | } t|�| kr�| j� t
|�j d �}
|
d k r�d||< |
| ||||||d � ||= d S t |t
�r�d||< | �||||||d � ||= d S |�|� d S )NTFr
)r$ r; �
_recursion�
_recursive� _readable�_reprr2 �len� _dispatch�getr# �__repr__�
isinstance�dict� _pprint_dict) r r r r � allowance�context�level�objid�rep� max_width�pr r r r: � s0
�zPrettyPrinter._formatc
C sz |j }|d� | jdkr*|| jd d � t|�}|rn| jrNt|�� td�} n|�� } | �| |||d ||� |d� d S )N�{r
� ��key�})r; r1 rI r8 �sorted�itemsr- �_format_dict_items)
r r r r rP rQ rR r; Zlengthr] r r r rO � s
�zPrettyPrinter._pprint_dictc C sl t |�s|�t|�� d S |j}|�|jd � | �t|�� �||t |j� d |d ||� |�d� d S )N�(r
�))rI r; �repr� __class__r' r: �listr] )r r r r rP rQ rR �clsr r r �_pprint_ordered_dict� s �z"PrettyPrinter._pprint_ordered_dictc C s0 |� d� | �||||d ||� |� d� d S )N�[r
�])r; �
_format_items�r r r r rP rQ rR r r r � _pprint_list� s
�zPrettyPrinter._pprint_listc C sH |� d� t|�dkrdnd}| �||||t|� ||� |� |� d S )Nr_ r
z,)r` )r; rI rh )r r r r rP rQ rR �endcharr r r �
_pprint_tuple� s
�zPrettyPrinter._pprint_tuplec C s� t |�s|�t|�� d S |j}|tkr8|�d� d}n&|�|jd � d}|t |j�d 7 }t|td�}| �||||t |� ||� |�|� d S )NrW r[ �({�})r
rY ) rI r; ra rb �setr' r\ r rh ) r r r r rP rQ rR �typrk r r r � _pprint_set� s
�zPrettyPrinter._pprint_setc C s� |j }t|�s |t|�� d S g }|�d�} |dkrD|d7 }|d7 }| j| }
} t| �D ]�\} }
t|
�}| t| �d kr�|
|8 }
t|�|
kr�|�|� qZt�d|
�}|� � | }d}t|�D ]h\}}|| }|t|�d kr�| t| �d kr�||8 }tt|��|k�r"|�r |�t|�� |}q�|}q�|rZ|�t|�� qZt|�dk�rV||� d S |dk�rh|d� t|�D ],\} }| dk�r�|dd| � ||� �qp|dk�r�|d � d S )
NTr
z\S*\s*� r_ r r9 rX r` )
r; rI ra �
splitlinesr2 � enumerate�append�re�findall�pop)r r r r rP rQ rR r; Zchunks�linesZ
max_width1rU �i�linerT �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} t|| j| |�D ]$}
|| � ||
� | s\dd| } q\|r�|d� d S )N� r
r_ rr r9 rX r` )r; rI ra �_wrap_bytes_reprr2 ) r r r r rP rQ rR r; Zparens�delimrT r r r �
_pprint_bytes3 s" zPrettyPrinter._pprint_bytesc C s> |j }|d� | �t|�||d |d ||d � |d� d S )Nz
bytearray(�
r
r` )r; r� �bytes)r r r r rP rQ rR r; r r r �_pprint_bytearrayH s �z PrettyPrinter._pprint_bytearrayc C s8 |� d� | �|�� ||d |d ||� |� d� d S )Nz
mappingproxy(�
r
r` )r; r: �copyri r r r �_pprint_mappingproxyQ s
�z"PrettyPrinter._pprint_mappingproxyc C s� |j }|| j7 }dd| }t|�d } t|�D ]f\}
\} } |
| k}
| �| ||�}||� |d� | �| ||t|� d |
r�|nd||� |
s0||� q0d S )N�,
rX r
z: r )r; r1 rI rt rH r: )r r] r r rP rQ rR r; �delimnlZ
last_indexrz rZ �ent�lastrT r r r r^ Y s
�z PrettyPrinter._format_dict_itemsc C sL |j }|| j7 }| jdkr,|| jd d � dd| }d} | j| d }
} t|�} z t| �}
W n tk
rx Y d S X d}|�sH|
}z t| �}
W n( tk
r� d}| |8 } |
|8 }
Y nX | j�r | �|||�}t|�d }|
|k r�| }
| r�|} |
|k�r |
|8 }
|| � d} ||� q~|| � |} | � ||||�r<|nd||� q~d S ) Nr
rX r� rr FTr �, )
r; r1 r2 �iter�next�
StopIterationr7 rH rI r: )r r] r r rP rQ rR r; r� r� r rU �itZnext_entr� r� rT �wr r r rh j sR
�zPrettyPrinter._format_itemsc C s4 | � ||�� | j|�\}}}|s&d| _|r0d| _|S )NFT)rA r� r0 rG rF )r r rQ rR ra rC rD r r r rH � s �
zPrettyPrinter._reprc C s t ||||| j�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 r8 )r r rQ � maxlevelsrR r r r rA � s zPrettyPrinter.formatc C s� t |�s|�t|�� d S | �|j||�}|j}|t |j�d 7 }|�d|j|d| f � | �||||d ||� |�d� d S )Nr
z %s(%s,
%srX r` )rI r; ra rH �default_factoryrb r' rO ) r r r r rP rQ rR Zrdfrd r r r �_pprint_default_dict� s z"PrettyPrinter._pprint_default_dictc C s� t |�s|�t|�� d S |j}|�|jd � | jdkrN|�| jd d � |�� }| �|||t |j� d |d ||� |�d� d S )Nrm r
rX r rn )rI r; ra rb r' r1 � most_commonr^ ) r r r r rP rQ rR rd r] r r r �_pprint_counter� s
�z PrettyPrinter._pprint_counterc
C s� t |j�s |�t|�� d S |j}|�|jd � |t |j�d 7 }t|j�D ]d\}} |t |j�d kr�| �| |||d ||� |�d� qN| �| ||d||� |�dd| � qNd S )Nr_ r
r` r� rX )rI �mapsr; ra rb r' rt r: )
r r r r rP rQ rR rd rz �mr r r �_pprint_chain_map� s
z PrettyPrinter._pprint_chain_mapc C s� t |�s|�t|�� d S |j}|�|jd � |t |j�d 7 }|�d� |jd krz| �||||d ||� |�d� n:| �|||d||� | �|j||�}|�dd| |f � d S )Nr_ r
rf r z])z],
%smaxlen=%s)rX )rI r; ra rb r' �maxlenrh rH ) r r r r rP rQ rR rd Zrmlr r r �
_pprint_deque� s&
� �zPrettyPrinter._pprint_dequec C s | � |j|||||d � d S �Nr
�r: �datari r r r �_pprint_user_dict� s z PrettyPrinter._pprint_user_dictc C s | � |j|||||d � d S r� r� ri r r r �_pprint_user_list� s z PrettyPrinter._pprint_user_listc C s | � |j|||||d � d S r� r� ri r r r �_pprint_user_string� s z!PrettyPrinter._pprint_user_string)r
r NN)2r' r( r) r r r r r r: rJ rO rN rL re � _collections� OrderedDictrj rc rl �tuplerq ro � frozensetr� r" r� r� r� � bytearrayr� �_types�MappingProxyTyper^ rh rH rA r� � defaultdictr� �Counterr� �ChainMapr� �dequer� �UserDictr� �UserListr� �
UserStringr r r r r g s^ � +
1
) c C s� t | �}|tkr t| �ddfS t|dd �}t|t��rD|tjk�rD| sJdS t| �}|rl||krldd||kfS ||kr�t| �ddfS d||< d}d} g }
|
j } |d7 }|r�t
| � � t d�} n| � � } | D ]b\}
}t
|
||||�\}}}t
|||||�\}}}| d||f � |�o|�o|}|�s$|r�d} q�||= d d
�|
� || fS t|t��r\|tjk�stt|t��rn|tjk�rnt|t��r�| �s�d S 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 }| D ]8}t
|||||�\}}}| |� |�sFd}|�rd} �q||= |d
�|
� || fS t| �}||�o�|�d� dfS )NTFrL )z{}TFz{...}r
rY z%s: %sz{%s}r� )z[]TFz[%s]z(%s,))z()TFz(%s)z...�<)r# �_builtin_scalarsra �getattr�
issubclassrN rL r$ rE ru r\ r] r- r �joinrc r� rI �
startswith)r rQ r� rR r rp �rrS rC rD Z
componentsru r] �k�vZkreprZ kreadableZkrecurZvreprZ vreadableZvrecurrA �oZoreprZ oreadableZorecurrT r r r r � s�
��
r c C s dt | �jt| �f S )Nz )r# r' r$ r r r r rE ? s �rE c C s� dd l }| d kr,ddddgddd�fgd } t� }|�� }t| i d dd
� |�� }|�| � |�� }td || � td || � d S )
Nr �string)r
r � r� � � )� � i�� Tz _safe_repr:zpformat:)�timer � perf_counterr r �print)r r� rV �t1�t2Zt3r r r �
_perfcheckD s
r� c c s� d}t | �d d }tdt | �d�D ]T}| ||d � }|| }||krP||8 }t t|��|krt|rnt|�V |}q$|}q$|r�t|�V d S )N� r� r )rI �rangera )r r rP r} r� rz r r� r r r r� Q s
r� �__main__)Nr
r N)r
r N)N)$r* � collectionsr� rv �sysr4 �typesr� �ior r> �__all__r r r r r r r r- r r r� r" r� r� r. �float�complexr6 r# r� rE r� r� r' r r r r � sJ
� � � F �