U
e5dS
@ s d Z ddlZddlZddlZddlZddlm Z
dddddd d
gZ d%d
ddddZ d&d
ddddZ
d
ddd
Zdd Zdd Zdd ZG dd dZdd ZG dd d Zd d Zeeeeeee e e dhZ d d Z d'd d!Z!d"d# Z"e#d$k re! 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)StringIOpprintpformat
isreadable isrecursivesaferepr
PrettyPrinterpp P FTcompact
sort_dictsc C s" t ||||||d}|| dS )zAPretty-print a Python object to a stream [default is sys.stdout].)streamindentwidthdepthr
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_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
)_formatr3 writer 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 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_reprr2 len _dispatchgetr# __repr__
isinstancedict _pprint_dict) r r r r allowancecontextlevelobjidrep max_widthpr 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 sorteditemsr- _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_itemsr 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 |jd 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|| qZtd|
}|st |d rt |
| }d}t|D ]l\}}|| }|t|d kr| t| d kr||8 }tt||kr:|r4|t| |}q|}q|rZ|t| qZt|dkrn|| d S |dkr|d t|D ],\} }| dkr|dd | || q|dkr|d
d S ) NTr
z\S*\s* r_ r r9 rX r` ) r; rI ra
splitlinesr2 enumerateappendrefindallAssertionErrorpop)r r r r rP rQ rR r; ZchunkslinesZ
max_width1rU ilinerT partsZ
max_width2currentjpart candidater r r _pprint_str sX
$
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_ rs r9 rX r` )r; rI ra _wrap_bytes_reprr2 ) r r r r rP rQ rR r; ZparensdelimrT 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 ru rH r: )r r] r r rP rQ rR r; delimnlZ
last_indexr| rZ entlastrT 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
rX r rs FTr , )
r; r1 r2 iternext
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 ||||| jS )zFormat 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 |jd 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 |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 rX )rI mapsr; ra rb r' ru r: )
r r r r rP rQ rR rd r| 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
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 _typesMappingProxyTyper^ 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|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 )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 rv r\ r] r- r joinrc r rI
startswith)r rQ r rR r rp rrS rC rD Z
componentsrv r] kvZkreprZ 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,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 rV 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 )rI rangera )r r rP r r r| r r r r r r Q s
r __main__)Nr
r N)r
r N)N)$r* collectionsr rw sysr4 typesr ior r> __all__r r r r r r r r- r r r r" r r r. floatcomplexr6 r# r rE r r r' r r r r sJ
F