U :eB@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)sizesignZunitr0/usr/local/python38/lib/python3.8/tracemalloc.py _format_size src@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) Statistic tracebackr countcCs||_||_||_dSNr)selfrr rrrr__init__%szStatistic.__init__cCst|j|j|jfSr)hashrr rrrrr__hash__*szStatistic.__hash__cCs$|j|jko"|j|jko"|j|jkSrrrotherrrr__eq__-s    zStatistic.__eq__cCsBd|jt|jd|jf}|jr>|j|j}|dt|d7}|S)Nz%s: size=%s, count=%iF , average=%s)rrr rrtextZaveragerrr__str__2s  zStatistic.__str__cCsd|j|j|jfS)Nz)rrrrr__repr__<szStatistic.__repr__cCs|j|j|jfSr)r rrrrrr _sort_key@szStatistic._sort_keyN __name__ __module__ __qualname__ __slots__rrrr!r"r#rrrrrs rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) StatisticDiffrr size_diffr count_diffcCs"||_||_||_||_||_dSrr*)rrr r+rr,rrrrKs zStatisticDiff.__init__cCst|j|j|j|j|jfSr)rrr r+rr,rrrrrRszStatisticDiff.__hash__cCs<|j|jko:|j|jko:|j|jko:|j|jko:|j|jkSrr*rrrrrVs     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,rrrrr!]s   zStatisticDiff.__str__cCsd|j|j|j|j|jfS)Nz9r*rrrrr"is  zStatisticDiff.__repr__cCs t|j|jt|j|j|jfSr)r r+r r,rrrrrrr#ns  zStatisticDiff._sort_keyNr$rrrrr)Ds r)cCsg}|D]d\}}||d}|dk rNt||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 statisticsrstatZpreviousrrr_compare_grouped_statsts*     r5c@sXeZdZdZddZeddZeddZdd Zd d Z d d Z ddZ ddZ dS)Frame_framecCs ||_dSrr7rframerrrrszFrame.__init__cCs |jdSr-r7rrrrfilenameszFrame.filenamecCs |jdSNr7rrrrlinenosz Frame.linenocCs |j|jkSrr7rrrrrsz Frame.__eq__cCs |j|jkSrr7rrrr__lt__sz Frame.__lt__cCs t|jSr)rr8rrrrrszFrame.__hash__cCsd|j|jfS)Nz%s:%sr;r>rrrrr!sz Frame.__str__cCsd|j|jfS)Nzr@rrrrr"szFrame.__repr__N) r%r&r'r(rpropertyr;r>rr?rr!r"rrrrr6s  r6c@sbeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dddZ dS) Traceback_framescCst|tt||_dSr)rrtuplereversedrD)rframesrrrrs zTraceback.__init__cCs t|jSr)lenrDrrrr__len__szTraceback.__len__cCs4t|tr"tdd|j|DSt|j|SdS)Ncss|]}t|VqdSr)r6.0tracerrr sz(Traceback.__getitem__..) isinstanceslicerErDr6rindexrrr __getitem__s zTraceback.__getitem__cCs |j|jkSr)r8rDr9rrr __contains__szTraceback.__contains__cCs t|jSr)rrDrrrrrszTraceback.__hash__cCs |j|jkSrrCrrrrrszTraceback.__eq__cCs |j|jkSrrCrrrrr?szTraceback.__lt__cCs t|dSr-)strrrrrr!szTraceback.__str__cCsdt|fS)Nz)rErrrrr"szTraceback.__repr__NFcCsg}|dk r2|dkr$|| d}q6|d|}n|}|rBt|}|D]@}|d|j|jft|j|j}|rF|d|qF|S)Nrz File "%s", line %sz %s)rFr0r;r> linecachegetlinestrip)rlimitZmost_recent_firstlinesZ frame_slicer:linerrrformats  zTraceback.format)NF)r%r&r'r(rrIrRrSrrr?r!r"r[rrrrrBsrBcCs t|}|dk rt|SdSdSr)rrB)objrGrrrget_object_tracebacksr]c@s\eZdZdZddZeddZeddZedd Zd d Z d d Z ddZ ddZ dS)Trace_tracecCs ||_dSrr_rrLrrrrszTrace.__init__cCs |jdSr-r_rrrrdomainsz Trace.domaincCs |jdSr<r_rrrrr sz Trace.sizecCst|jdS)N)rBr`rrrrr szTrace.tracebackcCs |j|jkSrr_rrrrrsz Trace.__eq__cCs t|jSr)rr`rrrrrszTrace.__hash__cCsd|jt|jdfS)Nz%s: %sF)rrr rrrrr!sz Trace.__str__cCsd|jt|jd|jfS)Nz'F)rbrr rrrrrr"szTrace.__repr__N) r%r&r'r(rrArbr rrrr!r"rrrrr^s   r^c@s<eZdZddZddZddZddZd d Zd d Zd S)_TracescCst|||_dSr)rr_traces)rtracesrrrr s z_Traces.__init__cCs t|jSr)rHrerrrrrI%sz_Traces.__len__cCs4t|tr"tdd|j|DSt|j|SdS)Ncss|]}t|VqdSr)r^rJrrrrM*sz&_Traces.__getitem__..)rNrOrErer^rPrrrrR(s z_Traces.__getitem__cCs |j|jkSr)r`rerarrrrS.sz_Traces.__contains__cCs |j|jkSr)rerrrrr1sz_Traces.__eq__cCs dt|S)Nz)rHrrrrr"4sz_Traces.__repr__N) r%r&r'rrIrRrSrr"rrrrrds rdcCs&tj|}|dr"|dd}|S)Nz.pyc)ospathnormcaseendswith)r;rrr_normalize_filename8s   rlc@seZdZddZddZdS) BaseFiltercCs ||_dSr) inclusive)rrnrrrr@szBaseFilter.__init__cCstdSr)NotImplementedErrorrarrr_matchCszBaseFilter._matchN)r%r&r'rrprrrrrm?srmcsJeZdZdfdd ZeddZddZd d Zd d Zd dZ Z S)FilterNFcs2t|||_t||_||_||_||_dSr)superrrnrl_filename_patternr> all_framesrb)rrnfilename_patternr>rtrb __class__rrrHs   zFilter.__init__cCs|jSr)rsrrrrruQszFilter.filename_patterncCs6t|}t||jsdS|jdkr(dS||jkSdS)NFT)rlfnmatchrsr>rr;r>rrr_match_frame_implUs  zFilter._match_frame_implcCs||||j ASr)rzrnryrrr _match_frame^szFilter._match_framecsHjr,tfdd|Dr"jSj Sn|d\}}||SdS)Nc3s|]\}}||VqdSr)rz)rKr;r>rrrrMcsz*Filter._match_traceback..r)rtanyrnr{)rrr;r>rrr_match_tracebackas   zFilter._match_tracebackcCsD|\}}}||}|jdk r@|jr2|o0||jkS|p>||jkS|Sr)r}rbrn)rrLrbr rresrrrrpls   z Filter._match)NFN) r%r&r'rrArurzr{r}rp __classcell__rrrvrrqGs    rqcs0eZdZfddZeddZddZZS) DomainFiltercst|||_dSr)rrr_domain)rrnrbrvrrrxs zDomainFilter.__init__cCs|jSr)rrrrrrb|szDomainFilter.domaincCs|\}}}||jk|j ASr)rbrn)rrLrbr rrrrrps zDomainFilter._match)r%r&r'rrArbrprrrrvrrws  rc@sTeZdZddZddZeddZddZd d Zd d Z dddZ dddZ dS)SnapshotcCst||_||_dSr)rdrftraceback_limit)rrfrrrrrs zSnapshot.__init__c Cs*t|d}t||tjW5QRXdS)Nwb)openpickledumpHIGHEST_PROTOCOL)rr;fprrrrs z Snapshot.dumpc Cs,t|d}t|W5QRSQRXdS)Nrb)rrload)r;rrrrrs z Snapshot.loadcs@|rtfdd|DsdS|r.Fc3s|]}| VqdSrrrrrrrMsT)r|)rinclude_filtersexclude_filtersrLrrr _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)rrJrrrrr s z*Snapshot.filter_traces..) rNr TypeErrortyper%rnr0rfrecopyrr)rfiltersrZ new_tracesrrr filter_tracess    zSnapshot.filter_tracesc Cs|dkrtd|f|r.|dkr.td|i}i}|s|jjD]}|\}}}z ||} WnZtk r|dkr||} n(|dkr|dd} n|dddff} t| } | ||<YnXz(|| } | j|7_| jd7_WqDtk rt| |d|| <YqDXqDn|jjD]}|\}}}|D]} z || } WnFtk r~|dkr\| f} n| ddff} t| } | || <YnXz(|| } | j|7_| jd7_Wn&tk rt| |d|| <YnXq(q|S) N)rr;r>zunknown key_type: %r)r>r;z/cumulative mode cannot by used with key type %rrr>r=r) ValueErrorrfreKeyErrorrBr rr) rkey_type cumulativeZstatsZ tracebacksrLrbr Ztrace_tracebackrrGr4r:rrr _group_bysZ        zSnapshot._group_byFcCs,|||}t|}|jdtjd|SNT)reversekey)rlistvaluessortrr#)rrrZgroupedr3rrrr3s  zSnapshot.statisticscCs6|||}|||}t||}|jdtjd|Sr)rr5rr)r#)rZ old_snapshotrrr2r1r3rrr compare_tos    zSnapshot.compare_toN)F)F) r%r&r'rr staticmethodrrrrr3rrrrrrs  3 rcCs$tstdt}t}t||S)NzLthe tracemalloc module must be tracing memory allocations to take a snapshot) is_tracing RuntimeErrorrZget_traceback_limitr)rfrrrr take_snapshot s r)collections.abcrr functoolsrrxrUZos.pathrhrZ _tracemallocrrrrr)r5r6rBr]r^rdrlrmrqrrrrrrrs2 &0"?%0