a DOgF@sddlmZmZddlmZddlZddlZddlZddl Z ddl Tddl m Z m Z ddZ Gdd d ZGd d d Zd d ZeGdddZeGdddeZddZGdddZGdddeZddZGdddZGdddeZGdddeZGd d!d!Zd"d#ZdS)$)SequenceIterable)total_orderingN)*)_get_object_traceback _get_tracescCsdD]|}t|dkr@|dkr@|r0d||fSd||fSt|dksT|dkrx|rhd||fSd ||fS|d }qdS) N)BZKiBZMiBZGiBTiBdrz%+.1f %sz%.1f %si(r z%+.0f %sz%.0f %si)abs)sizeZsignZunitr 0/opt/alt/python39/lib64/python3.9/tracemalloc.py _format_size src@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) Statistic tracebackr countcCs||_||_||_dSNr)selfrr rr r r__init__%szStatistic.__init__cCst|j|j|jfSr)hashrr rrr r r__hash__*szStatistic.__hash__cCs2t|tstS|j|jko0|j|jko0|j|jkSr) isinstancerNotImplementedrr rrotherr r r__eq__-s    zStatistic.__eq__cCsBd|jt|jd|jf}|jr>|j|j}|dt|d7}|S)Nz%s: size=%s, count=%iF , average=%s)rrr rrtextZaverager r r__str__4s  zStatistic.__str__cCsd|j|j|jfS)Nz)rrr r r__repr__>szStatistic.__repr__cCs|j|j|jfSr)r rrrr r r _sort_keyBszStatistic._sort_keyN __name__ __module__ __qualname__ __slots__rrrr"r#r$r r r rrs rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) StatisticDiffrr size_diffr count_diffcCs"||_||_||_||_||_dSrr+)rrr r,rr-r r rrMs zStatisticDiff.__init__cCst|j|j|j|j|jfSr)rrr r,rr-rr r rrTszStatisticDiff.__hash__cCsJt|tstS|j|jkoH|j|jkoH|j|jkoH|j|jkoH|j|jkSr)rr*rrr r,rr-rr r rrXs      zStatisticDiff.__eq__cCsPd|jt|jdt|jd|j|jf}|jrL|j|j}|dt|d7}|S)Nz %s: size=%s (%s), count=%i (%+i)FTr)rrr r,rr-r r r rr"as   zStatisticDiff.__str__cCsd|j|j|j|j|jfS)Nz9r+rr r rr#ms  zStatisticDiff.__repr__cCs t|j|jt|j|j|jfSr)r r,r r-rrrr r rr$rs  zStatisticDiff._sort_keyNr%r r r rr*Fs  r*cCsg}|D]d\}}||d}|durNt||j|j|j|j|j|j}nt||j|j|j|j}||q |D]*\}}t|d|j d|j }||qz|SNr)itemspopr*r rappend) old_group new_group statisticsrstatZpreviousr r r_compare_grouped_statsxs"   r6c@sXeZdZdZddZeddZeddZdd Zd d Z d d Z ddZ ddZ dS)Frame_framecCs ||_dSrr8rframer r rrszFrame.__init__cCs |jdSr.r8rr r rfilenameszFrame.filenamecCs |jdSNr8rr r rlinenosz Frame.linenocCst|tstS|j|jkSrrr7rr9rr r rrs z Frame.__eq__cCst|tstS|j|jkSrr@rr r r__lt__s z Frame.__lt__cCs t|jSr)rr9rr r rrszFrame.__hash__cCsd|j|jfS)Nz%s:%sr<r?rr r rr"sz Frame.__str__cCsd|j|jfS)NzrBrr r rr#szFrame.__repr__N) r&r'r(r)rpropertyr<r?rrArr"r#r r r rr7s  r7c@speZdZdZdddZeddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZdddZdS) Traceback)_frames _total_nframeNcCs"t|tt||_||_dSr)rrtuplereversedrErF)rframes total_nframer r rrs zTraceback.__init__cCs|jSr)rFrr r rrJszTraceback.total_nframecCs t|jSr)lenrErr r r__len__szTraceback.__len__cCs4t|tr"tdd|j|DSt|j|SdS)Ncss|]}t|VqdSr)r7.0tracer r r z(Traceback.__getitem__..)rslicerGrEr7rindexr r r __getitem__s zTraceback.__getitem__cCs |j|jvSr)r9rEr:r r r __contains__szTraceback.__contains__cCs t|jSr)rrErr r rrszTraceback.__hash__cCst|tstS|j|jkSrrrDrrErr r rrs zTraceback.__eq__cCst|tstS|j|jkSrrWrr r rrAs zTraceback.__lt__cCs t|dSr.)strrr r rr"szTraceback.__str__cCs8dt|}|jdur"|d7}n|d|jd7}|S)Nz z total_nframe=)rGrFrJ)rsr r rr#s   zTraceback.__repr__FcCsg}|dur2|dkr$|| d}q6|d|}n|}|rBt|}|D]@}|d|j|jft|j|j}|rF|d|qF|S)Nrz File "%s", line %sz %s)rHr1r<r? linecachegetlinestrip)rlimitZmost_recent_firstlinesZ frame_slicer;liner r rformats  zTraceback.format)N)NF)r&r'r(r)rrCrJrLrUrVrrrAr"r#rar r r rrDs  rDcCs t|}|durt|SdSdSr)rrD)objrIr r rget_object_tracebacksrcc@s\eZdZdZddZeddZeddZedd Zd d Z d d Z ddZ ddZ dS)Trace_tracecCs ||_dSrrerrOr r rrszTrace.__init__cCs |jdSr.rerr r rdomainsz Trace.domaincCs |jdSr=rerr r rr sz Trace.sizecCst|jddS)N)rDrfrr r rr#szTrace.tracebackcCst|tstS|j|jkSr)rrdrrfrr r rr's z Trace.__eq__cCs t|jSr)rrfrr r rr,szTrace.__hash__cCsd|jt|jdfS)Nz%s: %sF)rrr rr r rr"/sz Trace.__str__cCsd|jt|jd|jfS)Nz'F)rhrr rrr r rr#2szTrace.__repr__N) r&r'r(r)rrCrhr rrrr"r#r r r rrds   rdc@s<eZdZddZddZddZddZd d Zd d Zd S)_TracescCst|||_dSr)rr_traces)rtracesr r rr8s z_Traces.__init__cCs t|jSr)rKrkrr r rrL=sz_Traces.__len__cCs4t|tr"tdd|j|DSt|j|SdS)Ncss|]}t|VqdSr)rdrMr r rrPBrQz&_Traces.__getitem__..)rrRrGrkrdrSr r rrU@s z_Traces.__getitem__cCs |j|jvSr)rfrkrgr r rrVFsz_Traces.__contains__cCst|tstS|j|jkSr)rrjrrkrr r rrIs z_Traces.__eq__cCs dt|S)Nz)rKrr r rr#Nsz_Traces.__repr__N) r&r'r(rrLrUrVrr#r r r rrj7s rjcCs&tj|}|dr"|dd}|S)Nz.pyc)ospathnormcaseendswith)r<r r r_normalize_filenameRs   rrc@seZdZddZddZdS) BaseFiltercCs ||_dSr) inclusive)rrtr r rrZszBaseFilter.__init__cCstdSr)NotImplementedErrorrgr r r_match]szBaseFilter._matchN)r&r'r(rrvr r r rrsYsrscsJeZdZdfdd ZeddZddZd d Zd d Zd dZ Z S)FilterNFcs2t|||_t||_||_||_||_dSr)superrrtrr_filename_patternr? all_framesrh)rrtfilename_patternr?rzrh __class__r rrbs   zFilter.__init__cCs|jSr)ryrr r rr{kszFilter.filename_patterncCs6t|}t||jsdS|jdur(dS||jkSdS)NFT)rrfnmatchryr?rr<r?r r r_match_frame_implos  zFilter._match_frame_implcCs||||j ASr)rrtrr r r _match_framexszFilter._match_framecsHjr,tfdd|Dr"jSj Sn|d\}}||SdS)Nc3s|]\}}||VqdSr)r)rNr<r?rr rrP}sz*Filter._match_traceback..r)rzanyrtr)rrr<r?r rr_match_traceback{s   zFilter._match_tracebackcCsF|\}}}}||}|jdurB|jr4|o2||jkS|p@||jkS|Sr)rrhrt)rrOrhr rrJresr r rrvs   z Filter._match)NFN) r&r'r(rrCr{rrrrv __classcell__r r r|rrwas    rwcs0eZdZfddZeddZddZZS) DomainFiltercst|||_dSr)rxr_domain)rrtrhr|r rrs zDomainFilter.__init__cCs|jSr)rrr r rrhszDomainFilter.domaincCs|\}}}}||jk|j ASr)rhrt)rrOrhr rrJr r rrvs zDomainFilter._match)r&r'r(rrCrhrvrr r r|rrs  rc@sTeZdZddZddZeddZddZd d Zd d Z dddZ dddZ dS)SnapshotcCst||_||_dSr)rjrltraceback_limit)rrlrr r rrs zSnapshot.__init__cCs>t|d }t||tjWdn1s00YdS)Nwb)openpickledumpZHIGHEST_PROTOCOL)rr<fpr r rrs z Snapshot.dumpcCs8t|d}t|WdS1s*0YdS)Nrb)rrload)r<rr r rrs z Snapshot.loadcs@|rtfdd|DsdS|r.Fc3s|]}| VqdSrrrrr rrPsT)r)rinclude_filtersexclude_filtersrOr rr _filter_traces  zSnapshot._filter_tracecst|tstdt|j|rjgg|D] }|jrB|q,|q,fddjjD}n jj }t |j S)Nz)filters must be a list of filters, not %scsg|]}|r|qSr )rrMrrrr r s z*Snapshot.filter_traces..) rr TypeErrortyper&rtr1rlrkcopyrr)rfiltersrZ new_tracesr rr filter_tracess    zSnapshot.filter_tracesc Cs|dvrtd|f|r.|dvr.td|i}i}|s |jjD]}|\}}}} z ||} WnXty|dkr||} n(|dkr|dd} n|dddff} t| } | ||<Yn0z(|| } | j|7_| jd7_WqDtyt| |d|| <YqD0qDn|jjD]}|\}}}} |D]} z || } WnDty||dkrZ| f} n| ddff} t| } | || <Yn0z(|| } | j|7_| jd7_Wn$tyt| |d|| <Yn0q(q|S) N)rr<r?zunknown key_type: %r)r?r<z/cumulative mode cannot by used with key type %rrr?r>r) ValueErrorrlrkKeyErrorrDr rr)rkey_type cumulativeZstatsZ tracebacksrOrhr Ztrace_tracebackrJrrIr5r;r r r _group_bysZ         zSnapshot._group_byFcCs,|||}t|}|jdtjd|SNT)reversekey)rlistvaluessortrr$)rrrZgroupedr4r r rr4s  zSnapshot.statisticscCs6|||}|||}t||}|jdtjd|Sr)rr6rr*r$)rZ old_snapshotrrr3r2r4r r r compare_tos    zSnapshot.compare_toN)F)F) r&r'r(rr staticmethodrrrrr4rr r r rrs  3 rcCs$tstdt}t}t||S)NzLthe tracemalloc module must be tracing memory allocations to take a snapshot) is_tracing RuntimeErrorrZget_traceback_limitr)rlrr r r take_snapshot's r)collections.abcrr functoolsrr~r[Zos.pathrnrZ _tracemallocrrrrr*r6r7rDrcrdrjrrrsrwrrrr r r rs2 (2&M'0