o
0i|_ @ s d Z ddlZddlZddlZddlZddlZ ddl
m Z g dZ
d$dddd d
d Zd%dddd d d
ZddddZdd Zdd Zdd ZG dd dZdd ZG dd d Zeeeeee e e dhZ d d Z d&d d Z!d!d" Z"e#d#kr|e! dS 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)pprintpformat
isreadable isrecursivesaferepr
PrettyPrinterpp P FTcompact
sort_dictsunderscore_numbersc C s$ t |||||||d}|| dS )zAPretty-print a Python object to a stream [default is sys.stdout].)streamindentwidthdepthr
r r N)r r ) objectr r r r r
r r Zprinter r -/opt/alt/python310/lib64/python3.10/pprint.pyr 0 s r c C s t ||||||d| S )z Z2e2e
ej3j
< d?d@ Z4e4e
ej5j
< dAdB Z6dS )Dr r
r NFTr c C s t |}t |}|dk rtd|dur |dkr td|s"td|| _|| _|| _|dur3|| _ntj| _t|| _ || _
|| _ dS )a Handle pretty printing operations onto a stream using a set of
configured parameters.
indent
Number of spaces to indent for each level of nesting.
width
Attempted maximum number of columns in the output.
depth
The maximum depth to print out nested structures.
stream
The desired output stream. If omitted (or false), the standard
output stream available at construction will be used.
compact
If true, several items will be combined in one line.
sort_dicts
If true, dict keys are sorted.
r zindent must be >= 0Nzdepth must be > 0zwidth must be != 0) int
ValueError_depth_indent_per_level_width_stream_sysstdoutbool_compact _sort_dicts_underscore_numbers)r" r r r r r
r r r r r r# k s"
zPrettyPrinter.__init__c C s& | || jddi d | jd d S )Nr
)_formatr6 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 rA rD )r" r sreadable recursiver r r r s
zPrettyPrinter.isreadablec C s
t |}||v r|t| d| _d| _d S | |||}| j| | } t|| kr~| j t
|j d }
|
d urNd||< |
| ||||||d ||= d S t
|r~t|t
s~|jjr~t|j dr~d|j jjv r~d||< | ||||||d ||= d S || d S )NTFr
__wrapped__Z
__create_fn__)r' r?
_recursion
_recursive _readable_reprr5 len _dispatchgetr& __repr__ _dataclassesZ is_dataclass
isinstanceZ__dataclass_params__reprhasattrrI r, _pprint_dataclass) r" r r r allowancecontextlevelobjidrep max_widthpr r r r> s8
zPrettyPrinter._formatc s` j j}|t|d 7 } fddt D }||d | |||||| |d d S )Nr
c s$ g | ]}|j r|jt |jfqS r )rT namegetattr).0fr r r
s $ z3PrettyPrinter._pprint_dataclass..()) __class__r* rN rR Zfieldsr? _format_namespace_items r" r r r rW rX rY cls_nameitemsr r r rV s z PrettyPrinter._pprint_dataclassc
C sz |j }|d | jdkr|| jd d t|}|r7| jr't| td} n| } | | |||d || |d d S )N{r
key})r? r4 rN r; sortedri r0 _format_dict_items)
r" r r r rW rX rY r? Zlengthri r r r _pprint_dict s
zPrettyPrinter._pprint_dictc C sl t |s
|t| d S |j}||jd | t| ||t |j d |d || |d d S )Nrc r
rd )rN r? rT re r* r> listri )r" r r r rW rX rY 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 rW rX rY r r r _pprint_list s
zPrettyPrinter._pprint_listc C sH | d t|dkr
dnd}| ||||t| || | | d S )Nrc r
z,)rd )r? rN rw )r" r r r rW rX rY endcharr r r
_pprint_tuple s
zPrettyPrinter._pprint_tuplec C s t |s
|t| d S |j}|tu r |d d}n||jd d}|t |jd 7 }t|td}| ||||t | || || d S )Nrj rn ({})r
rl ) rN r? rT re setr* ro r rw ) r" r r r rW rX rY typrz r r r _pprint_set s
zPrettyPrinter._pprint_setc C s |j }t|s|t| d S g }|d} |dkr"|d7 }|d7 }| j| }
} t| D ]n\} }
t|
}| t| d krA|
|8 }
t||
krM|| q-td|
}| | }d}t|D ]2\}}|| }|t|d kr{| t| d kr{||8 }tt||kr|r|t| |}q_|}q_|r|t| q-t|dkr|| d S |dkr|d t|D ]\} }| dkr|dd| || q|dkr|d d S d S )
NTr
z\S*\s* rc r r= rk rd )
r? rN rT
splitlinesr5 enumerateappendrefindallpop)r" r r r rW rX rY r? ZchunkslinesZ
max_width1r\ iliner[ partsZ
max_width2currentjpart candidater r r _pprint_str sX
zPrettyPrinter._pprint_strc C s |j }t|dkr|t| d S |dk}|r#|d7 }|d7 }|d d} t|| j| |D ]}
|| ||
| s@dd| } q.|rI|d d S d S )N r
rc r r= rk rd )r? rN rT _wrap_bytes_reprr5 ) r" r r r rW rX rY r? Zparensdelimr[ r r r
_pprint_bytesC s& zPrettyPrinter._pprint_bytesc C s> |j }|d | t|||d |d ||d |d d S )Nz
bytearray(
r
rd )r? r bytes)r" r r r rW rX rY r? r r r _pprint_bytearrayX s z PrettyPrinter._pprint_bytearrayc C s8 | d | | ||d |d || | d d S )Nz
mappingproxy(
r
rd )r? r> copyrx r r r _pprint_mappingproxya s
z"PrettyPrinter._pprint_mappingproxyc C sf t |tju r
d}n|jj}|t|d 7 }|j }||d | |||||| |d d S )N namespacer
rc rd )
r& _typesSimpleNamespacere r* rN __dict__ri r? rf rg r r r _pprint_simplenamespacei s
z%PrettyPrinter._pprint_simplenamespacec C s |j }|| j7 }dd| }t|d } t|D ]3\}
\} } |
| k}
| | ||}|| |d | | ||t| d |
r@|nd|| |
sK|| qd S )N,
rk r
z: r )r? r4 rN r rM r> )r" ri r r rW rX rY r? delimnl
last_indexr rm entlastr[ r r r rp x s"
z PrettyPrinter._format_dict_itemsc C s |j }dd| }t|d } t|D ]7\}
\} } |
| k}
|| |d t| |v r0|d n| | ||t| d |
r?|nd|| |
sJ|| qd S )Nr rk r
=...)r? rN r r' r> )r" ri r r rW rX rY r? r r r rm r r r r r rf s"
z%PrettyPrinter._format_namespace_itemsc C sF |j }|| j7 }| jdkr|| jd d dd| }d} | j| d }
} t|} zt| }
W n
ty; Y d S w d}|s|
}zt| }
W n ty[ d}| |8 } |
|8 }
Y nw | jr| |||}t|d }|
|k rv| }
| rv|} |
|kr|
|8 }
|| d} || q>|| |} | ||||r|nd|| |r@d S d S ) Nr
rk r r FTr , )
r? r4 r5 iternext
StopIterationr: rM rN r> )r" ri r r rW rX rY r? r r r r\ itZnext_entr r r[ wr r r rw sV
zPrettyPrinter._format_itemsc C s4 | || | j|\}}}|sd| _|rd| _|S )NFT)rE r r3 rL rK )r" r rX rY rT rG rH r r r rM s
zPrettyPrinter._reprc C s | ||||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 rX maxlevelsrY r r r rE 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,
%srk rd )rN r? rT rM default_factoryre r* rq ) r" r r r rW rX rY Zrdfrs r r r _pprint_default_dict s z"PrettyPrinter._pprint_default_dictc C s t |s
|t| d S |j}||jd | jdkr'|| jd d | }| |||t |j d |d || |d d S )Nr| r
rk r r} )rN r? rT re r* r4 most_commonrp ) r" r r r rW rX rY rs ri r r r _pprint_counter s
z PrettyPrinter._pprint_counterc
C s t |js|t| d S |j}||jd |t |jd 7 }t|jD ]2\}} |t |jd krF| | |||d || |d q'| | ||d|| |dd| q'd S )Nrc r
rd r rk )rN mapsr? rT re r* r r> )
r" r r r rW rX rY rs 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 u r>| ||||d || |d d S | |||d|| | |j||}|dd| |f d S )Nrc r
ru r z])z],
%smaxlen=%s)rk )rN r? rT re r* maxlenrw rM ) r" r r r rW rX rY rs Zrmlr r r
_pprint_deque s"
zPrettyPrinter._pprint_dequec C | |j|||||d d S Nr
r> datarx r r r _pprint_user_dict z PrettyPrinter._pprint_user_dictc C r r r rx r r r _pprint_user_list r z PrettyPrinter._pprint_user_listc C r r r rx r r r _pprint_user_string! r z!PrettyPrinter._pprint_user_stringc C s t |}|tv rt|ddfS t|dd }t|tr0|tju r0| jr)|dddfS t|ddfS t|tr|tju r|s>dS t |}|rO||krOdd||v fS ||v rZt
|ddfS d||< d}d} g }
|
j } |d7 }| j rwt
| td} n| } | D ].\}
}| |
|||\}}}| ||||\}}}| d ||f |o|o|}|s|rd} q}||= d
d |
|| fS t|tr|tju st|trF|tju rFt|tr|sd S d
}nt|dkrd}n|sdS d}t |}|r||kr|d d||v fS ||v r t
|ddfS d||< d}d} g }
|
j } |d7 }|D ] }| ||||\}}}| | |s2d}|r7d} q ||= |d |
|| fS t|}||oS|d dfS )NTFrQ Z_d)z{}TFz{...}r
rl z%s: %sz{%s}r )z[]TFz[%s]z(%s,))z()TFz(%s)r <)r& _builtin_scalarsrT r_
issubclassr1 rQ r< dictr' rJ r r; ro ri r0 rE joinrr tuplerN
startswith)r" r rX r rY r rrZ rG rH Z
componentsr ri kvZkreprZ kreadableZkrecurZvreprZ vreadableZvrecurrE oZoreprZ oreadableZorecurr[ r r r r & s
zPrettyPrinter._safe_repr)r
r NN)7r* r+ r, r# r r r r r> rV rO rq r rQ rt _collections OrderedDictry rr r{ r r r~ frozensetr r% r r r bytearrayr r MappingProxyTyper r rp rf rw rM rE r defaultdictr Counterr ChainMapr dequer UserDictr UserListr
UserStringr r r r r r j sh ,
1
) r c C s dt | jt| f S )Nz )r& r* r' r r r r rJ z s rJ c C s dd l }| d u rddddgdddfgd } 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 r] t1t2Zt3r r r
_perfcheck s
r c c s d}t | d d }tdt | dD ]*}| ||d }|| }||kr)||8 }t t||kr;|r8t|V |}q|}q|rGt|V d S d S )N r r )rN rangerT )r r rW r r r r r r r r r s
r __main__)Nr
r N)r
r Nr! )$r- collectionsr Z dataclassesrR r sysr7 typesr ior rB __all__r r r r r r r r0 r r r% r r floatcomplexr9 r& r rJ r r r* r r r r sD