[#=@s`ddlmZmZddlmZddlZddlZddlZddl Z ddl Tddl m Z m Z ddZ Gdd d ZGd d d Zd d ZeGdddZeGdddeZddZGdddZGdddeZddZGdddZGdddZddZdS) )SequenceIterable)total_orderingN)*)_get_object_traceback _get_tracescCsxd D]}t|dkrP|dkrP|r?d||fSd||fSnt|dksn|dkr|rd ||fSd ||fSn|d }qWdS)NBKiBMiBGiBTiBdz%+.1f %sz%.1f %s iz%+.0f %sz%.0f %s)rr r r r i()abs)sizeZsignZunitr!/usr/lib/python3.4/tracemalloc.py _format_size s rc@sdeZdZdZdZddZddZd d Zd d Zd dZ ddZ dS) StatisticzS Statistic difference on memory allocations between two Snapshot instance. tracebackrcountcCs||_||_||_dS)N)rrr)selfrrrrrr__init__%s  zStatistic.__init__cCst|j|j|jfS)N)hashrrr)rrrr__hash__*szStatistic.__hash__cCs4|j|jko3|j|jko3|j|jkS)N)rrr)rotherrrr__eq__-szStatistic.__eq__cCs\d|jt|jd|jf}|jrX|j|j}|dt|d7}n|S)Nz%s: size=%s, count=%iFz , average=%s)rrrr)rtextaveragerrr__str__2s  zStatistic.__str__cCsd|j|j|jfS)Nz))rrr)rrrr__repr__<szStatistic.__repr__cCs|j|j|jfS)N)rrr)rrrr _sort_key@szStatistic._sort_keyN)z tracebackzsizezcount) __name__ __module__ __qualname____doc__ __slots__rrrrr r!rrrrrs     rc@sdeZdZdZdZddZd d Zd d Zd dZddZ ddZ dS) StatisticDiffzd Statistic difference on memory allocations between an old and a new Snapshot instance. rr size_diffr count_diffcCs1||_||_||_||_||_dS)N)rrr(rr))rrrr(rr)rrrrKs     zStatisticDiff.__init__cCs(t|j|j|j|j|jfS)N)rrrr(rr))rrrrrRszStatisticDiff.__hash__cCsX|j|jkoW|j|jkoW|j|jkoW|j|jkoW|j|jkS)N)rrr(rr))rrrrrrVs zStatisticDiff.__eq__cCsqd|jt|jdt|jd|j|jf}|jrm|j|j}|dt|d7}n|S)Nz %s: size=%s (%s), count=%i (%+i)FTz , average=%s)rrrr(rr))rrrrrrr]s  zStatisticDiff.__str__cCs&d|j|j|j|j|jfS)Nz9)rrr(rr))rrrrr iszStatisticDiff.__repr__cCs.t|j|jt|j|j|jfS)N)rr(rr)rr)rrrrr!nszStatisticDiff._sort_keyN)z tracebackzsizez size_diffzcountz count_diff) r"r#r$r%r&rrrrr r!rrrrr'Ds     r'cCsg}x|jD]\}}|j|d}|dk rrt||j|j|j|j|j|j}n$t||j|j|j|j}|j|qWxG|jD]9\}}t|d|j d|j }|j|qW|S)Nr)itemspopr'rrappend) old_group new_group statisticsrstatZpreviousrrr_compare_grouped_statsts   r1c@seZdZdZdZddZeddZeddZd d Z d d Z d dZ ddZ ddZ dS)Framez Frame of a traceback. _framecCs ||_dS)N)r3)rframerrrrszFrame.__init__cCs |jdS)Nr)r3)rrrrfilenameszFrame.filenamecCs |jdS)N)r3)rrrrlinenosz Frame.linenocCs|j|jkS)N)r3)rrrrrrsz Frame.__eq__cCs|j|jkS)N)r3)rrrrr__lt__sz Frame.__lt__cCs t|jS)N)rr3)rrrrrszFrame.__hash__cCsd|j|jfS)Nz%s:%s)r5r7)rrrrrsz Frame.__str__cCsd|j|jfS)Nz)r5r7)rrrrr szFrame.__repr__N)z_frame)r"r#r$r%r&rpropertyr5r7rr8rrr rrrrr2s      r2c@seZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ dddZdS) Tracebackz` Sequence of Frame instances sorted from the most recent frame to the oldest frame. _framescCstj|||_dS)N)rrr;)rframesrrrrs zTraceback.__init__cCs t|jS)N)lenr;)rrrr__len__szTraceback.__len__cCsBt|tr-tdd|j|DSt|j|SdS)Ncss|]}t|VqdS)N)r2).0tracerrr sz(Traceback.__getitem__..) isinstanceslicetupler;r2)rindexrrr __getitem__szTraceback.__getitem__cCs|j|jkS)N)r3r;)rr4rrr __contains__szTraceback.__contains__cCs t|jS)N)rr;)rrrrrszTraceback.__hash__cCs|j|jkS)N)r;)rrrrrrszTraceback.__eq__cCs|j|jkS)N)r;)rrrrrr8szTraceback.__lt__cCst|dS)Nr)str)rrrrrszTraceback.__str__cCsdt|fS)Nz)rD)rrrrr szTraceback.__repr__NcCsg}|dk r"|dkr"|Sxm|d|D][}|jd|j|jftj|j|jj}|r3|jd|q3q3W|S)Nrz File "%s", line %sz %s)r,r5r7 linecachegetlinestrip)rlimitlinesr4linerrrformats zTraceback.format)z_frames)r"r#r$r%r&rr>rFrGrrr8rr rOrrrrr:s          r:cCs*t|}|dk r"t|SdSdS)z Get the traceback where the Python object *obj* was allocated. Return a Traceback instance. Return None if the tracemalloc module is not tracing memory allocations or did not trace the allocation of the object. N)rr:)objr<rrrget_object_tracebacks   rQc@s|eZdZdZdZddZeddZeddZd d Z d d Z d dZ ddZ dS)Tracez" Trace of a memory block. _tracecCs ||_dS)N)rS)rr@rrrrszTrace.__init__cCs |jdS)Nr)rS)rrrrrsz Trace.sizecCst|jdS)Nr6)r:rS)rrrrrszTrace.tracebackcCs|j|jkS)N)rS)rrrrrrsz Trace.__eq__cCs t|jS)N)rrS)rrrrrszTrace.__hash__cCsd|jt|jdfS)Nz%s: %sF)rrr)rrrrr sz Trace.__str__cCsdt|jd|jfS)NzF)rrr)rrrrr szTrace.__repr__N)z_trace) r"r#r$r%r&rr9rrrrrr rrrrrRs     rRc@sXeZdZddZddZddZddZd d Zd d Zd S)_TracescCstj|||_dS)N)rr_traces)rtracesrrrrs z_Traces.__init__cCs t|jS)N)r=rU)rrrrr>sz_Traces.__len__cCsBt|tr-tdd|j|DSt|j|SdS)Ncss|]}t|VqdS)N)rR)r?r@rrrrAsz&_Traces.__getitem__..)rBrCrDrUrR)rrErrrrFsz_Traces.__getitem__cCs|j|jkS)N)rSrU)rr@rrrrG sz_Traces.__contains__cCs|j|jkS)N)rU)rrrrrr#sz_Traces.__eq__cCsdt|S)Nz)r=)rrrrr &sz_Traces.__repr__N) r"r#r$rr>rFrGrr rrrrrTs      rTcCs8tjj|}|jdr4|dd}n|S)N.pyc.pyor6)rWrX)ospathnormcaseendswith)r5rrr_normalize_filename*sr^c@sXeZdZddddZeddZddZd d Zd d ZdS) FilterNFcCs.||_t||_||_||_dS)N) inclusiver^_filename_patternr7 all_frames)rr`filename_patternr7rbrrrr2s  zFilter.__init__cCs|jS)N)ra)rrrrrc9szFilter.filename_patterncCsIt|}tj||js%dS|jdkr8dS||jkSdS)NFT)r^fnmatchrar7)rr5r7rrrZ __match_frame=s  zFilter.__match_framecCs|j|||j AS)N)_Filter__match_framer`)rr5r7rrr _match_frameFszFilter._match_framecs^jr:tfdd|Dr/jSj Sn |d\}}j||SdS)Nc3s'|]\}}j||VqdS)N)re)r?r5r7)rrrrAKsz*Filter._match_traceback..r)rbanyr`rf)rrr5r7r)rr_match_tracebackIs   zFilter._match_traceback) r"r#r$rr9rcrerfrhrrrrr_1s  r_c@seZdZdZddZddZeddZdd Zd d Z d d Z dddZ dddZ dS)SnapshotzB Snapshot of traces of memory blocks allocated by Python. cCst||_||_dS)N)rTrVtraceback_limit)rrVrjrrrrZszSnapshot.__init__c Cs2t|d}tj||tjWdQXdS)z1 Write the snapshot into a file. wbN)openpickledumpZHIGHEST_PROTOCOL)rr5fprrrrn`sz Snapshot.dumpc Cs)t|d}tj|SWdQXdS)z. Load a snapshot from a file. rbN)rlrmload)r5rorrrrqgsz Snapshot.loadcsf|d|r6tfdd|Ds6dSn|rbtfdd|DrbdSndS)Nr6c3s|]}|jVqdS)N)rh)r? trace_filter)rrrrArsz)Snapshot._filter_trace..Fc3s|]}|j VqdS)N)rh)r?rr)rrrrAvsT)rg)rinclude_filtersexclude_filtersr@r)rr _filter_traceos   zSnapshot._filter_tracecst|ts+tdt|jn|rggx4|D],}|jrcj|qDj|qDWfddjjD}njjj }t |j S)z Create a new Snapshot instance with a filtered traces sequence, filters is a list of Filter instances. If filters is an empty list, return a new Snapshot instance with a copy of the traces. z)filters must be a list of filters, not %scs+g|]!}j|r|qSr)ru)r?r@)rtrsrrr s  z*Snapshot.filter_traces..) rBr TypeErrortyper"r`r,rVrUcopyrirj)rfiltersrrZ new_tracesr)rtrsrr filter_traces{s  (zSnapshot.filter_tracesc CsM|dkr"td|fn|rG|d krGtd|ni}i}|sZx|jjD]}|\}}y||}Wnstk r|dkr|} n6|dkr|dd} n|dddff} t| }|||s*      &0 # 5 ! $