U e5dS @sdZddlZddlZddlZddlZddlm Z dddddd d gZ d%d ddddZ d&d ddddZ d ddd ZddZddZddZGdddZddZGdd d ZddZeeeeeeeeedhZddZ 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 PrettyPrinterppPFTcompact sort_dictscCs"t||||||d}||dS)zAPretty-print a Python object to a stream [default is sys.stdout].)streamindentwidthdepthr rN)rr)objectrrrrr rZprinterr/usr/lib64/python3.8/pprint.pyr/scCst|||||d|S)z= 0Nzdepth must be > 0zwidth must be != 0) int ValueError_depth_indent_per_level_width_stream_sysstdoutbool_compact _sort_dicts)rrrrrr rrrrr hs  zPrettyPrinter.__init__cCs&|||jddid|jddS)Nr )_formatr3writerrrrrrszPrettyPrinter.pprintcCs"t}|||ddid|SNr) _StringIOr:getvalue)rrZsiorrrrszPrettyPrinter.pformatcCs||idddS)Nrrformatr<rrrrszPrettyPrinter.isrecursivecCs ||idd\}}}|o| Sr=r@)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 )r$r; _recursion _recursive _readable_reprr2len _dispatchgetr#__repr__ isinstancedict _pprint_dict) rrrr allowancecontextlevelobjidrep max_widthprrrr:s0   zPrettyPrinter._formatc Csz|j}|d|jdkr*||jddt|}|rn|jrNt|td} n|} || |||d|||ddS)N{r  key})r;r1rIr8sorteditemsr-_format_dict_items) rrrrrPrQrRr;Zlengthr]rrrrOs zPrettyPrinter._pprint_dictcCslt|s|t|dS|j}||jd|t|||t|jd|d|||ddS)N(r ))rIr;repr __class__r'r:listr])rrrrrPrQrRclsrrr_pprint_ordered_dictsz"PrettyPrinter._pprint_ordered_dictcCs0|d|||||d|||ddS)N[r ])r; _format_itemsrrrrrPrQrRrrr _pprint_lists  zPrettyPrinter._pprint_listcCsH|dt|dkrdnd}|||||t|||||dS)Nr_r z,)r`)r;rIrh)rrrrrPrQrRendcharrrr _pprint_tuples zPrettyPrinter._pprint_tuplec Cst|s|t|dS|j}|tkr8|dd}n&||jdd}|t|jd7}t|td}|||||t|||||dS)NrWr[({})r rY) rIr;rarbsetr'r\rrh) rrrrrPrQrRtyprkrrr _pprint_sets   zPrettyPrinter._pprint_setcCs|j}t|s|t|dSg}|d} |dkrD|d7}|d7}|j|} } t| D]\} } t| }| t| dkr| |8} t|| kr||qZtd| }|st |drt | | }d}t|D]l\}}||}|t|dkr| t| dkr||8}tt||kr:|r4|t||}q|}q|rZ|t|qZt|dkrn||dS|dkr|dt|D],\} }| dkr|dd |||q|dkr|d dS) NTr z\S*\s*r_rr9rXr`) r;rIra splitlinesr2 enumerateappendrefindallAssertionErrorpop)rrrrrPrQrRr;ZchunkslinesZ max_width1rUilinerTpartsZ max_width2currentjpart candidaterrr _pprint_strsX      $    zPrettyPrinter._pprint_strc Cs|j}t|dkr"|t|dS|dk}|rF|d7}|d7}|dd} t||j||D]$} || || | s\dd|} q\|r|ddS)Nr r_rsr9rXr`)r;rIra_wrap_bytes_reprr2) rrrrrPrQrRr;ZparensdelimrTrrr _pprint_bytes3s"  zPrettyPrinter._pprint_bytesc Cs>|j}|d|t|||d|d||d|ddS)Nz bytearray( r r`)r;rbytes)rrrrrPrQrRr;rrr_pprint_bytearrayHszPrettyPrinter._pprint_bytearraycCs8|d||||d|d|||ddS)Nz mappingproxy( r r`)r;r:copyrirrr_pprint_mappingproxyQs  z"PrettyPrinter._pprint_mappingproxyc Cs|j}||j7}dd|}t|d} t|D]f\} \} } | | k} || ||}|||d|| ||t|d| r|nd||| s0||q0dS)N, rXr z: r)r;r1rIrurHr:)rr]rrrPrQrRr;delimnlZ last_indexr|rZentlastrTrrrr^Ys     z PrettyPrinter._format_dict_itemscCsL|j}||j7}|jdkr,||jdddd|}d} |j|d} } t|} z t| } Wntk rxYdSXd}|sH| }z t| } Wn(tk rd}| |8} | |8} YnX|jr||||}t|d}| |kr| } | r|} | |kr| |8} || d} ||q~|| |} | ||||r<|nd||q~dS) Nr rXrrsFTr, ) r;r1r2iternext StopIterationr7rHrIr:)rr]rrrPrQrRr;rrrrUitZnext_entrrrTwrrrrhjsR         zPrettyPrinter._format_itemscCs4||||j|\}}}|s&d|_|r0d|_|S)NFT)rArr0rGrF)rrrQrRrarCrDrrrrHs  zPrettyPrinter._reprcCst|||||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. )rr8)rrrQ maxlevelsrRrrrrAszPrettyPrinter.formatc Cst|s|t|dS||j||}|j}|t|jd7}|d|j|d|f|||||d|||ddS)Nr z %s(%s, %srXr`)rIr;rarHdefault_factoryrbr'rO) rrrrrPrQrRZrdfrdrrr_pprint_default_dictsz"PrettyPrinter._pprint_default_dictc Cst|s|t|dS|j}||jd|jdkrN||jdd|}||||t|jd|d|||ddS)Nrmr rXrrn)rIr;rarbr'r1 most_commonr^) rrrrrPrQrRrdr]rrr_pprint_counters zPrettyPrinter._pprint_counterc Cst|js|t|dS|j}||jd|t|jd7}t|jD]d\}} |t|jdkr|| |||d|||dqN|| ||d|||dd|qNdS)Nr_r r`rrX)rImapsr;rarbr'rur:) rrrrrPrQrRrdr|mrrr_pprint_chain_maps  zPrettyPrinter._pprint_chain_mapc Cst|s|t|dS|j}||jd|t|jd7}|d|jdkrz|||||d|||dn:||||d||||j||}|dd||fdS)Nr_r rfrz])z], %smaxlen=%s)rX)rIr;rarbr'maxlenrhrH) rrrrrPrQrRrdZrmlrrr _pprint_deques&    zPrettyPrinter._pprint_dequec Cs||j|||||ddSNr r:datarirrr_pprint_user_dictszPrettyPrinter._pprint_user_dictc Cs||j|||||ddSrrrirrr_pprint_user_listszPrettyPrinter._pprint_user_listc Cs||j|||||ddSrrrirrr_pprint_user_stringsz!PrettyPrinter._pprint_user_string)r r NN)2r'r(r)r rrrrr:rJrOrNrLre _collections OrderedDictrjrcrltuplerqro frozensetrr"rrr bytearrayr_typesMappingProxyTyper^rhrHrAr defaultdictrCounterrChainMaprdequerUserDictrUserListr UserStringrrrrrgs^ +      1    )       cCst|}|tkrt|ddfSt|dd}t|trD|tjkrD|sJdSt|}|rl||krldd||kfS||krt|ddfSd||<d}d} g} | j } |d7}|rt | t d} n| } | D]b\} }t | ||||\}}}t |||||\}}}| d||f|o|o|}|s$|rd} q||=d d | || fSt|tr\|tjkstt|trn|tjkrnt|tr|sd Sd }n"t|dkrd }n|sdSd}t|}|r||kr|dd||kfS||krt|ddfSd||<d}d} g} | j } |d7}|D]8}t |||||\}}}| ||sFd}|rd} q||=|d | || fSt|}||o|d dfS)NTFrL)z{}TFz{...}r rYz%s: %sz{%s}r)z[]TFz[%s]z(%s,))z()TFz(%s)z...<)r#_builtin_scalarsragetattr issubclassrNrLr$rErvr\r]r-rjoinrcrrI startswith)rrQrrRrrprrSrCrDZ componentsrvr]kvZkreprZ kreadableZkrecurZvreprZ vreadableZvrecurrAoZoreprZ oreadableZorecurrTrrrrs      rcCsdt|jt|fS)Nz)r#r'r$rrrrrE?srEcCsddl}|dkr,ddddgdddfgd }t}|}t|iddd |}|||}td ||td ||dS) Nrstring)r rr)iTz _safe_repr:zpformat:)timer perf_counterrrprint)rrrVt1t2Zt3rrr _perfcheckDs rccsd}t|dd}tdt|dD]T}|||d}||}||krP||8}tt||krt|rnt|V|}q$|}q$|rt|VdS)Nrr)rIrangera)rrrPrrr|rrrrrrQs r__main__)Nr r N)r r N)N)$r* collectionsrrwsysr4typesriorr>__all__rrr rrrrr-rrrr"rrr.floatcomplexr6r#rrErrr'rrrr sJ    F