3
\|Q
@ s d Z ddlZddlZddlZddlZddlm Z
dddddd gZ d!d d
ddZ d"d d
ddZ
dd Zdd Zdd ZG dd dZdd ZG dd d Zdd Zeeeeeeee e dhZ dd Z d#d d Z d d Z!e"d kre 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
PrettyPrinter P F)compactc C s t |||||d}|j| dS )zAPretty-print a Python object to a stream [default is sys.stdout].)streamindentwidthdepthr N)r r )objectr r
r r r Zprinter r /usr/lib64/python3.6/pprint.pyr / s c C s t ||||dj| S )z= 0Nzdepth must be > 0zwidth must be != 0)
int
ValueError_depth_indent_per_level_width_stream_sysstdoutbool_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 ddd S )Nr r )format)r r r r r r s zPrettyPrinter.isrecursivec C s | j |i dd\}}}|o | S )Nr )r5 )r r sreadable 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 rd||< |
| ||||||d ||= d S t |t
rd||< | j||||||d ||= d S |j| d S )NTFr )r r2
_recursion
_recursive _readable_reprr* len _dispatchgetr __repr__
isinstancedict _pprint_dict) r r r r
allowancecontextlevelobjidrep max_widthpr 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= sorteditemsr% _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 |jd 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|
}|st |d
st |j
| }d}xxt|D ]l\}}|| }|t|d kr| t| d kr||8 }tt||krD|r>|jt| |}q|}qW |r^|jt| q^W t|dkr||| d S |dkr|d x8t|D ],\} }| dkr|dd| || qW |dkr|d d S ) NTr z\S*\s* rR r r0 rL rS ) r2 r= rT
splitlinesr* enumerateappendrefindallAssertionErrorpop)r r r r
rD rE rF r2 ZchunkslinesZ
max_width1rI ilinerH 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} 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 ZparensdelimrH 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 rw 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= re r< r1 )r rP r r
rD rE rF r2 delimnlZ
last_indexrl rM entlastrH 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 | jr | j|||}t|d }|
|k r| }
| r|} |
|kr |
|8 }
|| d} || q~|| |} | j ||||r<|nd|| q~W d S ) Nr rL z,
rb FTr z, )
r2 r) r* iternext
StopIterationr/ r< r= r1 )r rP r r
rD rE rF r2 r~ rv 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 r| r( r; r: )r r rE rF rT r7 r8 r r r r< s zPrettyPrinter._reprc C s t ||||S )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 )r r rE maxlevelsrF r r r r5 s zPrettyPrinter.formatc C s t |s|jt| d S | j|j||}|j}|t |jd 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 |js |jt| d S |j}|j|jd |t |jd 7 }xrt|jD ]d\}} |t |jd kr| j| |||d || |jd qP| j| ||d|| |jdd| qPW d S )NrR r rS z,
rL )r= mapsr2 rT rU r re r1 )
r r r r
rD rE rF rW rl mr r r _pprint_chain_map s
z PrettyPrinter._pprint_chain_mapc C s t |s|jt| d S |j}|j|jd |t |jd 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_ frozensetrs r rw ry rz bytearrayr} _typesMappingProxyTyperQ 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|to<|tjkr8| sHdS t| }|rj||krjdd||kfS ||krt| ddfS d||< d}d}g } | j }
|d7 }t
} t | j t
d} xf| D ]^\}
}| |
|||\}}}| ||||\}}}|
d||f |o
|o
|}|s|rd}qW ||= d d
j| ||fS t|trP|tjksht|trd|tjkrdt|tr| s~dS d }n"t| dkrd
}n| sdS d}t| }|r||kr|d d||kfS ||krt| 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 rf r rO rP r% joinrV r r=
startswith)r rE r rF r` rrG r7 r8 Z
componentsrf r rP kvZkreprZ 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 rt ) 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 | dD ]T}| ||d }|| }||krR||8 }t t||krv|rpt|V |}q&|}q&W |rt|V d S )N rt r )r= rangerT )r r rD ro r rl rq rr r r r ru C s
ru __main__)Nr r
N)r r
N)N)#r" collectionsr rg sysr, typesr ior r3 __all__r r r r r r r% r r r r ry r r& floatcomplexr. r r r9 r ru r r r r r # s6
D