U
e5dS
@ s d dl Zd dlZd dlZd dlZd dlmZ ddddddd gZ
d$d d
dddZ d%d d
dddZ d ddd Z
dd Zdd Zdd ZG dd dZdd ZG dd dZdd Zeeeeeeee e dhZ d d Z d&d d Z d!d" Z!e"d#k re dS )' N)StringIOpprintpformat
isreadable isrecursivesaferepr
PrettyPrinterpp P FTcompact
sort_dictsc C s" t ||||||d}|| d S )N)streamindentwidthdepthr
r )r r )objectr r r r r
r Zprinter r /usr/lib64/python3.8/pprint.pyr / s c C s t |||||d| S )N)r r r r
r )r r )r r r r r
r r r r r 7 s )r c O s t | f|d|i| d S )Nr )r )r r argskwargsr r r r = s c C s t | i d ddd S )Nr T
_safe_reprr r r r r A s c C s t | i d ddd S )Nr Tr
r r r r r r E s c C s t | i d ddd S )Nr T r r r r r r I s c @ s" e Zd ZdgZdd Zdd ZdS ) _safe_keyobjc C s
|| _ d S N)r )selfr r r r __init__Y s z_safe_key.__init__c C sX z| j |j k W S tk
rR tt| j t| j ftt|j t|j fk Y S X d S r )r TypeErrorstrtypeid)r otherr r r __lt__\ s z_safe_key.__lt__N)__name__
__module__ __qualname__ __slots__r r& r r r r r M s
r c C s t | d t | d fS )Nr r
)r )tr r r _safe_tuplec s r, c @ s e Zd Zd;dddddZd d
Zd d Zd
d Zdd Zdd Zi Z dd Z
e
e e j < dd Z
e
e ejj < dd Zee ej < dd Zee ej < dd Zee ej < ee ej < d d Zee ej < d d Zee ej < d!d" Zee e j < d#d$ Z e e e j j < d%d& Z d'd( Z!d)d* Z"d+d, Z#d-d. Z$e$e ej%j < d/d0 Z&e&e ej'j < d1d2 Z(e(e ej)j < d3d4 Z*e*e ej+j < d5d6 Z,e,e ej-j < d7d8 Z.e.e ej/j < d9d: Z0e0e ej1j < dS )= 0zdepth must be > 0zwidth must be != 0) int
ValueError_depth_indent_per_level_width_stream_sysstdoutbool_compact _sort_dicts)r r r r r r
r r r r r h s
zPrettyPrinter.__init__c C s& | || jddi d | jd d S )Nr
)_formatr2 writer r r r r r s zPrettyPrinter.pprintc C s" t }| ||ddi d | S Nr ) _StringIOr9 getvalue)r r Zsior r r r s zPrettyPrinter.pformatc C s | |i ddd S )Nr r formatr; r r r r s zPrettyPrinter.isrecursivec C s | |i dd\}}}|o | S r< r? )r r sreadable 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 rd||< |
| ||||||d ||= d S t |t
rd||< | ||||||d ||= d S || d S )NTFr
)r$ r:
_recursion
_recursive _readable_reprr1 len _dispatchgetr# __repr__
isinstancedict _pprint_dict) r r r r allowancecontextlevelobjidrep max_widthpr r r r9 s0
zPrettyPrinter._formatc
C sz |j }|d | jdkr*|| jd d t|}|rn| jrNt| td} n| } | | |||d || |d d S )N{r
key})r: r0 rH r7 sorteditemsr, _format_dict_items)
r r r r rO rP rQ r: Zlengthr\ r r r rN s
zPrettyPrinter._pprint_dictc C sl t |s|t| d S |j}||jd | t| ||t |j d |d || |d d S )N(r
))rH r: repr __class__r' r9 listr\ )r r r r rO rP rQ clsr r r _pprint_ordered_dict s z"PrettyPrinter._pprint_ordered_dictc C s0 | d | ||||d || | d d S )N[r
])r:
_format_itemsr r r r rO rP rQ r r r _pprint_list s
zPrettyPrinter._pprint_listc C sH | d t|dkrdnd}| ||||t| || | | d S )Nr^ r
z,)r_ )r: rH rg )r r r r rO rP rQ endcharr r r
_pprint_tuple s
zPrettyPrinter._pprint_tuplec C s t |s|t| d S |j}|tkr8|d d}n&||jd d}|t |jd 7 }t|td}| ||||t | || || d S )NrV rZ ({})r
rX ) rH r: r` ra setr' r[ r rg ) r r r r rO rP rQ typrj 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|| qZtd|
}| | }d}t|D ]h\}}|| }|t|d kr| t| d kr||8 }tt||kr"|r |t| |}q|}q|rZ|t| qZt|dkrV|| d S |dkrh|d t|D ],\} }| dkr|dd| || qp|dkr|d d S )
NTr
z\S*\s* r^ r r8 rW r_ )
r: rH r`
splitlinesr1 enumerateappendrefindallpop)r r r r rO rP rQ r: ZchunkslinesZ
max_width1rT ilinerS partsZ
max_width2currentjpart 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^ rq r8 rW r_ )r: rH r` _wrap_bytes_reprr1 ) r r r r rO rP rQ r: ZparensdelimrS 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 rO rP rQ r: r r r _pprint_bytearrayH s z PrettyPrinter._pprint_bytearrayc C s8 | d | | ||d |d || | d d S )Nz
mappingproxy(
r
r_ )r: r9 copyrh 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,
rW r
z: r )r: r0 rH rs rG r9 )r r\ r r rO rP rQ r: delimnlZ
last_indexry rY entlastrS 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 | jr | |||}t|d }|
|k r| }
| r|} |
|kr |
|8 }
|| d} || q~|| |} | ||||r<|nd|| q~d S ) Nr
rW r rq FTr , )
r: r0 r1 iternext
StopIterationr6 rG rH r9 )r r\ r r rO rP rQ r: r r r rT itZnext_entr r rS wr r r rg j sR
zPrettyPrinter._format_itemsc C s4 | || | j|\}}}|s&d| _|r0d| _|S )NFT)r@ r r/ rF rE )r r rP rQ r` rB rC r r r rG s
zPrettyPrinter._reprc C s t ||||| jS r )r r7 )r r rP maxlevelsrQ r r r r@ s zPrettyPrinter.formatc C s t |s|t| d S | |j||}|j}|t |jd 7 }|d|j|d| f | ||||d || |d d S )Nr
z %s(%s,
%srW r_ )rH r: r` rG default_factoryra r' rN ) r r r r rO rP rQ Zrdfrc 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 )Nrl r
rW r rm )rH r: r` ra r' r0 most_commonr] ) r r r r rO rP rQ rc r\ r r r _pprint_counter s
z PrettyPrinter._pprint_counterc
C s t |js |t| d S |j}||jd |t |jd 7 }t|jD ]d\}} |t |jd kr| | |||d || |d qN| | ||d|| |dd| qNd S )Nr^ r
r_ r rW )rH mapsr: r` ra r' rs r9 )
r r r r rO rP rQ rc ry mr r r _pprint_chain_map s
z PrettyPrinter._pprint_chain_mapc C s t |s|t| d S |j}||jd |t |jd 7 }|d |jd krz| ||||d || |d n:| |||d|| | |j||}|dd| |f d S )Nr^ r
re r z])z],
%smaxlen=%s)rW )rH r: r` ra r' maxlenrg rG ) r r r r rO rP rQ rc Zrmlr r r
_pprint_deque s&
zPrettyPrinter._pprint_dequec C s | |j|||||d d S Nr
r9 datarh r r r _pprint_user_dict s z PrettyPrinter._pprint_user_dictc C s | |j|||||d d S r r rh r r r _pprint_user_list s z PrettyPrinter._pprint_user_listc C s | |j|||||d d S r r rh r r r _pprint_user_string s z!PrettyPrinter._pprint_user_string)r
r NN)2r' r( r) r r r r r r9 rI rN rM rK rd _collections OrderedDictri rb rk tuplerp rn frozensetr r" r r r bytearrayr _typesMappingProxyTyper] rg rG r@ 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|trD|tjkrD| sJdS t| }|rl||krldd||kfS ||krt| ddfS d||< d}d} g }
|
j } |d7 }|rt
| t d} n| } | D ]b\}
}t
|
||||\}}}t
|||||\}}}| d||f |o|o|}|s$|rd} q||= d d
|
|| fS t|tr\|tjkstt|trn|tjkrnt|tr| sd S d }n"t| dkrd
}n| sdS d}t| }|r||kr|d d||kfS ||krt| ddfS d||< d}d} g }
|
j } |d7 }| D ]8}t
|||||\}}}| | |sFd}|rd} q||= |d
|
|| fS t| }||o|d dfS )NTFrK )z{}TFz{...}r
rX z%s: %sz{%s}r )z[]TFz[%s]z(%s,))z()TFz(%s)z...<)r# _builtin_scalarsr` getattr
issubclassrM rK r$ rD rt r[ r\ r, r joinrb r rH
startswith)r rP r rQ r ro rrR rB rC Z
componentsrt r\ kvZkreprZ kreadableZkrecurZvreprZ vreadableZvrecurr@ oZoreprZ oreadableZorecurrS r r r r s
r c C s dt | jt| f S )Nz )r# r' r$ r r r r rD ? s rD c C s dd l }| d kr,ddddgdddfgd } 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 rU t1t2Zt3r r r
_perfcheckD s
r c c s d}t | d d }tdt | dD ]T}| ||d }|| }||krP||8 }t t||krt|rnt|V |}q$|}q$|rt|V d S )N r r )rH ranger` )r r rO r| r ry r~ r r r r r Q s
r __main__)Nr
r N)r
r N)N)# collectionsr ru sysr3 typesr ior r= __all__r r r r r r r r, r r r r" r r r- floatcomplexr5 r# r rD r r r' r r r r % sH
F