B Y6dQ @sdZddlZddlZddlZddlZddlm Z dddddd gZ d!d d ddZ d"d d ddZ ddZddZddZGdddZddZGdd d ZddZeeeeeeeeedhZddZd#ddZ ddZ!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 PrettyPrinterPF)compactcCs t|||||d}||dS)zAPretty-print a Python object to a stream [default is sys.stdout].)streamindentwidthdepthr N)rr)objectr r rrr printerr"/usr/local/lib/python3.7/pprint.pyr/scCst||||d|S)z= 0Nzdepth must be > 0zwidth must be != 0) int ValueError_depth_indent_per_level_width_stream_sysstdoutbool_compact)rr rrr r rrrrcszPrettyPrinter.__init__cCs&|||jddid|jddS)Nr )_formatr,write)rrrrrrszPrettyPrinter.pprintcCs"t}|||ddid|S)Nr) _StringIOr2getvalue)rrsiorrrrszPrettyPrinter.pformatcCs||idddS)Nrr)format)rrrrrrszPrettyPrinter.isrecursivecCs ||idd\}}}|o| S)Nr)r7)rrsreadable recursiverrrrszPrettyPrinter.isreadablec Cst|}||kr.|t|d|_d|_dS||||}|j||} t|| kr|j t |j d} | dk rd||<| |||||||d||=dSt |t rd||<|||||||d||=dS||dS)NTFr )rr3 _recursion _recursive _readable_reprr+len _dispatchgetr__repr__ isinstancedict _pprint_dict) rrr r allowancecontextlevelobjidrep max_widthprrrr2s,    zPrettyPrinter._formatc Csj|j}|d|jdkr*||jddt|}|r^t|td} || |||d|||ddS)N{r  )key})r3r*r?sorteditemsr&_format_dict_items) rrr r rFrGrHr3lengthrRrrrrEs zPrettyPrinter._pprint_dictcCslt|s|t|dS|j}||jd|t|||t|jd|d|||ddS)N(r ))r?r3repr __class__r r2listrR)rrr r rFrGrHclsrrr_pprint_ordered_dictsz"PrettyPrinter._pprint_ordered_dictcCs0|d|||||d|||ddS)N[r ])r3 _format_items)rrr r rFrGrHrrr _pprint_lists zPrettyPrinter._pprint_listcCsH|dt|dkrdnd}|||||t|||||dS)NrUr z,)rV)r3r?r^)rrr r rFrGrHendcharrrr _pprint_tuples  zPrettyPrinter._pprint_tuplec Cst|s|t|dS|j}|tkr8|dd}n&||jdd}|t|jd7}t|td}|||||t|||||dS)NrMrPz({z})r )rO) r?r3rWrXsetr rQrr^) rrr r rFrGrHtypr`rrr _pprint_sets  zPrettyPrinter._pprint_setcCs|j}t|s|t|dSg}|d} |dkrD|d7}|d7}|j|} } xt| D]\} } t| }| t| dkr| |8} t|| kr||q^td| }|st |drt | | }d}xxt|D]l\}}||}|t|dkr| t| dkr||8}tt||kr@|r:|t||}q|}qW|r^|t|q^Wt|dkrx||dS|dkr|dx8t|D],\} }| dkr|dd |||qW|dkr|d dS) NTr z\S*\s*rUrr1rNrV) r3r?rW splitlinesr+ enumerateappendrefindallAssertionErrorpop)rrr r rFrGrHr3chunkslinesZ max_width1rKilinerJpartsZ max_width2currentjpart candidaterrr _pprint_strsX      $   zPrettyPrinter._pprint_strc Cs|j}t|dkr"|t|dS|dk}|rF|d7}|d7}|dd} x:t||j||D]$} || || | s^dd|} q^W|r|ddS)Nr rUrfr1rNrV)r3r?rW_wrap_bytes_reprr+) rrr r rFrGrHr3ZparensdelimrJrrr _pprint_bytes's"  zPrettyPrinter._pprint_bytesc Cs>|j}|d|t|||d|d||d|ddS)Nz bytearray( r rV)r3r{bytes)rrr r rFrGrHr3rrr_pprint_bytearray<s zPrettyPrinter._pprint_bytearraycCs8|d||||d|d|||ddS)Nz mappingproxy( r rV)r3r2copy)rrr r rFrGrHrrr_pprint_mappingproxyEs z"PrettyPrinter._pprint_mappingproxyc Cs|j}||j7}dd|}t|d} xrt|D]f\} \} } | | k} || ||}|||d|| ||t|d| r|nd||| s2||q2WdS)Nz, rNr z: r)r3r*r?rhr>r2)rrRr r rFrGrHr3delimnl last_indexrprOentlastrJrrrrSMs    z PrettyPrinter._format_dict_itemscCsN|j}||j7}|jdkr,||jdddd|}d} |j|d} } t|} y t| } Wntk rvdSXd}x|sH| }y t| } Wn(tk rd}| |8} | |8} YnX|jr||||}t|d}| |kr| } | r|} | |kr| |8} || d} ||q~|| |} | ||||r<|nd||q~WdS) Nr rNz, rfFTrz, ) r3r*r+iternext StopIterationr0r>r?r2)rrRr r rFrGrHr3rrzrrKitZnext_entrrrJwrrrr^^sN         zPrettyPrinter._format_itemscCs4||||j|\}}}|s&d|_|r0d|_|S)NFT)r7rr)r=r<)rrrGrHrWr9r:rrrr>s zPrettyPrinter._reprcCst||||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)rrrG maxlevelsrHrrrr7szPrettyPrinter.formatc Cst|s|t|dS||j||}|j}|t|jd7}|d|j|d|f|||||d|||ddS)Nr z %s(%s, %srNrV)r?r3rWr>default_factoryrXr rE) rrr r rFrGrHZrdfrZrrr_pprint_default_dictsz"PrettyPrinter._pprint_default_dictc Cst|s|t|dS|j}||jd|jdkrN||jdd|}||||t|jd|d|||ddS)Nz({r rNrz}))r?r3rWrXr r* most_commonrS) rrr r rFrGrHrZrRrrr_pprint_counters zPrettyPrinter._pprint_counterc Cst|js|t|dS|j}||jd|t|jd7}xrt|jD]d\}} |t|jdkr|| |||d|||dqP|| ||d|||dd|qPWdS)NrUr rVz, rN)r?mapsr3rWrXr rhr2) rrr r rFrGrHrZrpmrrr_pprint_chain_maps  zPrettyPrinter._pprint_chain_mapc Cst|s|t|dS|j}||jd|t|jd7}|d|jdkrz|||||d|||dn:||||d||||j||}|dd||fdS)NrUr r\rz])z], %smaxlen=%s)rN)r?r3rWrXr maxlenr^r>) rrr r rFrGrHrZZrmlrrr _pprint_deques    zPrettyPrinter._pprint_dequec Cs||j|||||ddS)Nr )r2data)rrr r rFrGrHrrr_pprint_user_dictszPrettyPrinter._pprint_user_dictc Cs||j|||||ddS)Nr )r2r)rrr r rFrGrHrrr_pprint_user_listszPrettyPrinter._pprint_user_listc Cs||j|||||ddS)Nr )r2r)rrr r rFrGrHrrr_pprint_user_stringsz!PrettyPrinter._pprint_user_string)r r NN)2r r!r"rrrrrr2r@rErDrBr[ _collections OrderedDictr_rYratuplerdrb frozensetrwrr{r}r~ bytearrayr_typesMappingProxyTyperSr^r>r7r defaultdictrCounterrChainMaprdequerUserDictrUserListr UserStringrrrrrbsZ&      1    )       cCst|}|tkrt|ddfSt|dd}t|tr:|tjkr:|sJdSt|}|rl||krldd||kfS||krt|ddfSd||<d}d}g} | j } |d7}t } t | t d} xf| D]^\} }| | |||\}}}| ||||\}}}| d||f|o |o |}|s|rd}qW||=d d | ||fSt|trR|tjksjt|trf|tjkrft|tr|sd Sd }n"t|dkrd }n|sdSd}t|}|r||kr|dd||kfS||krt|ddfSd||<d}d}g} | j } |d7}x>|D]6}t ||||\}}}| ||s)rr r)rrrrr;1sr;cCsddl}|dkr,ddddgdddfgd }t}|}t|idd|}|||}td ||td ||dS) Nrstring)r rrx)iz _safe_repr:zpformat:)timer perf_counterrrprint)rrrLt1t2Zt3rrr _perfcheck6s rccsd}t|dd}xhtdt|dD]T}|||d}||}||krR||8}tt||krv|rpt|V|}q&|}q&W|rt|VdS)Nrxr)r?rangerW)rrrFrsrrprurvrrrryCs ry__main__)Nr r N)r r N)N)#r# collectionsrrjsysr-typesriorr4__all__rrrrrrr&rrrrr}rr'floatcomplexr/rrr;rryr rrrr#s6   D