a
XC?hF @ s d dl mZmZ d dlmZ d dlZd dlZd dlZd dl Z d dl
T d dl
m Z m Z dd Z
G dd d ZG d
d d Zd d
ZeG dd dZeG dd deZdd ZG dd dZG dd deZdd ZG dd dZG d d d eZG d d d eZG d d! d!Zd"d# ZdS )$ )SequenceIterable)total_orderingN)*)_get_object_traceback _get_tracesc C s dD ]|}t | dk r@|dkr@|r0d| |f S d| |f S t | dk sT|dkrx|rhd| |f S d | |f S | d
} qd S ) N)BZKiBZMiBZGiBTiBd r z%+.1f %sz%.1f %si ( r z%+.0f %sz%.0f %si )abs)sizeZsignZunit r
0/opt/alt/python39/lib64/python3.9/tracemalloc.py _format_size
s r c @ sD e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dS ) StatisticzS
Statistic difference on memory allocations between two Snapshot instance.
tracebackr countc C s || _ || _|| _d S Nr )selfr r r r
r
r __init__% s zStatistic.__init__c C s t | j| j| jfS r )hashr r r r r
r
r __hash__* s zStatistic.__hash__c C s2 t |tstS | j|jko0| j|jko0| j|jkS r )
isinstancer NotImplementedr r r r otherr
r
r __eq__- s
zStatistic.__eq__c C sB d| j t| jd| jf }| jr>| j| j }|dt|d 7 }|S )Nz%s: size=%s, count=%iF , average=%s)r r r r r textZaverager
r
r __str__4 s
zStatistic.__str__c C s d| j | j| jf S )Nz)r r r
r
r __repr__> s zStatistic.__repr__c C s | j | j| jfS r )r r r r r
r
r _sort_keyB s zStatistic._sort_keyN __name__
__module__ __qualname____doc__ __slots__r r r r" r# r$ r
r
r
r r s
r c @ sD e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dS )
StatisticDiffzd
Statistic difference on memory allocations between an old and a new
Snapshot instance.
r r size_diffr
count_diffc C s" || _ || _|| _|| _|| _d S r r, )r r r r- r r. r
r
r r M s
zStatisticDiff.__init__c C s t | j| j| j| j| jfS r )r r r r- r r. r r
r
r r T s zStatisticDiff.__hash__c C sJ t |tstS | j|jkoH| j|jkoH| j|jkoH| j|jkoH| j|jkS r )r r+ r r r r- r r. r r
r
r r X s
zStatisticDiff.__eq__c C sP d| j t| jdt| jd| j| jf }| jrL| j| j }|dt|d 7 }|S )Nz %s: size=%s (%s), count=%i (%+i)FTr )r r r r- r r. r r
r
r r" a s
zStatisticDiff.__str__c C s d| j | j| j| j| jf S )Nz9r, r r
r
r r# m s
zStatisticDiff.__repr__c C s t | j| jt | j| j| jfS r )r r- r r. r r r r
r
r r$ r s zStatisticDiff._sort_keyNr% r
r
r
r r+ F s r+ c C s g }| D ]d\}}| |d }|d urNt||j|j|j |j|j|j }nt||j|j|j|j}|| q | D ]*\}}t|d|j d|j }|| qz|S Nr )itemspopr+ r r append) old_group new_group
statisticsr statZpreviousr
r
r _compare_grouped_statsx s" r7 c @ s\ e Zd ZdZdZdd Zedd Zedd Zd d
Z d d Z
d
d Z dd Z dd Z
dS )Framez
Frame of a traceback.
_framec C s
|| _ d S r r9 r framer
r
r r s zFrame.__init__c C s
| j d S r/ r9 r r
r
r filename s zFrame.filenamec C s
| j d S N r9 r r
r
r lineno s z Frame.linenoc C s t |tstS | j|jkS r r r8 r r: r r
r
r r s
z Frame.__eq__c C s t |tstS | j|jk S r rA r r
r
r __lt__ s
z Frame.__lt__c C s
t | jS r )r r: r r
r
r r s zFrame.__hash__c C s d| j | jf S )Nz%s:%sr= r@ r r
r
r r" s z
Frame.__str__c C s d| j | jf S )Nz rC r r
r
r r# s zFrame.__repr__N)r& r' r( r) r* r propertyr= r@ r rB r r" r# r
r
r
r r8 s
r8 c @ st e Zd ZdZdZdddZedd Zdd Zd
d Z d d
Z
dd Z dd Z dd Z
dd Zdd Zd ddZdS ) Tracebackz`
Sequence of Frame instances sorted from the oldest frame
to the most recent frame.
)_frames
_total_nframeNc C s" t | tt|| _|| _d S r )r r tuplereversedrF rG )r frames total_nframer
r
r r s
zTraceback.__init__c C s | j S r )rG r r
r
r rK s zTraceback.total_nframec C s
t | jS r )lenrF r r
r
r __len__ s zTraceback.__len__c C s4 t |tr"tdd | j| D S t| j| S d S )Nc s s | ]}t |V qd S r )r8 .0tracer
r
r z(Traceback.__getitem__..)r slicerH rF r8 r indexr
r
r __getitem__ s
zTraceback.__getitem__c C s |j | jv S r )r: rF r; r
r
r __contains__ s zTraceback.__contains__c C s
t | jS r )r rF r r
r
r r s zTraceback.__hash__c C s t |tstS | j|jkS r r rE r rF r r
r
r r s
zTraceback.__eq__c C s t |tstS | j|jk S r rX r r
r
r rB s
zTraceback.__lt__c C s t | d S r/ )strr r
r
r r" s zTraceback.__str__c C s8 dt | }| jd u r"|d7 }n|d| j d7 }|S )Nz z total_nframe=)rH rG rK )r sr
r
r r# s
zTraceback.__repr__Fc C s g }|d ur2|dkr$| | d }q6| d | }n| }|rBt |}|D ]@}|d|j|jf t|j|j }|rF|d| qF|S )Nr z File "%s", line %sz %s)rI r2 r= r@ linecachegetlinestrip)r limitZmost_recent_firstlinesZ frame_slicer< liner
r
r format s
zTraceback.format)N)NF)r& r' r( r) r* r rD rK rM rV rW r r rB r" r# rb r
r
r
r rE s
rE c C s t | }|durt|S dS dS )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)r rE )objrJ r
r
r get_object_traceback s rd c @ s` e Zd ZdZdZdd Zedd Zedd Zed d
Z d d Z
d
d Z dd Z dd Z
dS )Tracez"
Trace of a memory block.
_tracec C s
|| _ d S r rf r rP r
r
r r s zTrace.__init__c C s
| j d S r/ rf r r
r
r domain s z Trace.domainc C s
| j d S r> rf r r
r
r r s z
Trace.sizec C s t | jdd S )N )rE rg r r
r
r r # s zTrace.tracebackc C s t |tstS | j|jkS r )r re r rg r r
r
r r ' s
z Trace.__eq__c C s
t | jS r )r rg r r
r
r r , s zTrace.__hash__c C s d| j t| jdf S )Nz%s: %sF)r r r r r
r
r r" / s z
Trace.__str__c C s d| j t| jd| jf S )Nz'F)ri r r r r r
r
r r# 2 s zTrace.__repr__N)r& r' r( r) r* r rD ri r r r r r" r# r
r
r
r re s
re c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )_Tracesc C s t | || _d S r )r r _traces)r tracesr
r
r r 8 s
z_Traces.__init__c C s
t | jS r )rL rl r r
r
r rM = s z_Traces.__len__c C s4 t |tr"tdd | j| D S t| j| S d S )Nc s s | ]}t |V qd S r )re rN r
r
r rQ B rR z&_Traces.__getitem__..)r rS rH rl re rT r
r
r rV @ s
z_Traces.__getitem__c C s |j | jv S r )rg rl rh r
r
r rW F s z_Traces.__contains__c C s t |tstS | j|jkS r )r rk r rl r r
r
r r I s
z_Traces.__eq__c C s dt | S )Nz)rL r r
r
r r# N s z_Traces.__repr__N) r& r' r( r rM rV rW r r# r
r
r
r rk 7 s rk c C s& t j| } | dr"| d d } | S )Nz.pyc)ospathnormcaseendswith)r= r
r
r _normalize_filenameR s
rs c @ s e Zd Zdd Zdd ZdS )
BaseFilterc C s
|| _ d S r ) inclusive)r ru r
r
r r Z s zBaseFilter.__init__c C s t d S r )NotImplementedErrorrh r
r
r _match] s zBaseFilter._matchN)r& r' r( r rw r
r
r
r rt Y s rt c sJ e Zd Zd fdd Zedd Zdd Zd d
Zd d Zd
d Z Z
S )FilterNFc s2 t | || _t|| _|| _|| _|| _d S r )superr ru rs _filename_patternr@
all_framesri )r ru filename_patternr@ r{ ri __class__r
r r b s
zFilter.__init__c C s | j S r )rz r r
r
r r| k s zFilter.filename_patternc C s6 t |}t|| jsdS | jd u r(dS || jkS d S )NFT)rs fnmatchrz r@ r r= r@ r
r
r _match_frame_implo s
zFilter._match_frame_implc C s | ||| j A S r )r ru r r
r
r _match_framex s zFilter._match_framec sH j r,t fdd|D r" jS j S n|d \}} ||S d S )Nc 3 s | ]\}} ||V qd S r )r )rO r= r@ r r
r rQ } s z*Filter._match_traceback..r )r{ anyru r )r r r= r@ r
r r _match_traceback{ s
zFilter._match_tracebackc C sF |\}}}}| |}| jd urB| jr4|o2|| jkS |p@|| jkS |S r )r ri ru )r rP ri r r rK resr
r
r rw s
z
Filter._match)NFN) r& r' r( r rD r| r r r rw
__classcell__r
r
r} r rx a s
rx c s0 e Zd Z fddZedd Zdd Z ZS ) DomainFilterc s t | || _d S r )ry r _domain)r ru ri r} r
r r s zDomainFilter.__init__c C s | j S r )r r r
r
r ri s zDomainFilter.domainc C s |\}}}}|| j k| j A S r )ri ru )r rP ri r r rK r
r
r rw s zDomainFilter._match)r& r' r( r rD ri rw r r
r
r} r r s
r c @ sX e Zd ZdZdd Zdd Zedd Zdd Zd
d Z d d
Z
dddZ dddZ dS )SnapshotzB
Snapshot of traces of memory blocks allocated by Python.
c C s t || _|| _d S r )rk rm traceback_limit)r rm r r
r
r r s
zSnapshot.__init__c C s> t |d }t| |tj W d n1 s00 Y dS )z1
Write the snapshot into a file.
wbN)openpickledumpZHIGHEST_PROTOCOL)r r= fpr
r
r r s z
Snapshot.dumpc C s8 t | d}t|W d S 1 s*0 Y dS )z.
Load a snapshot from a file.
rbN)r r load)r= r r
r
r r s z
Snapshot.loadc s@ |r t fdd|D s dS |r.Fc 3 s | ]}| V qd S r r r r r
r rQ s T)r )r include_filtersexclude_filtersrP r
r r
_filter_trace s zSnapshot._filter_tracec s t |ts tdt|j |rjg g |D ] }|jrB| q, | q, fddjjD }n jj }t
|j S )z
Create a new Snapshot instance with a filtered traces sequence, filters
is a list of Filter or DomainFilter 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 %sc s g | ]} |r|qS r
)r rN r r r r
r
s
z*Snapshot.filter_traces..) r r TypeErrortyper& ru r2 rm rl copyr r )r filtersr Z
new_tracesr
r r
filter_traces s
zSnapshot.filter_tracesc
C s |dvrt d|f |r.|dvr.t d| i }i }|s | jjD ]}|\}}}} z || }
W nX ty |dkr||} n(|dkr|d d } n|d d dff} t| }
|
||< Y n0 z(||
} | j|7 _| jd7 _W qD ty t|
|d||
< Y qD0 qDn| jjD ]}|\}}}} |D ]}
z ||
}
W nD ty| |dkrZ|
f} n|
d dff} t| }
|
||
< Y n0 z(||
} | j|7 _| jd7 _W n$ ty t|
|d||
< Y n0 q(q|S ) N)r r= r@ zunknown key_type: %r)r@ r= z/cumulative mode cannot by used with key type %rr r@ r? r )
ValueErrorrm rl KeyErrorrE r r r )r key_type
cumulativeZstatsZ
tracebacksrP ri r Ztrace_tracebackrK r rJ r6 r< r
r
r _group_by sZ
zSnapshot._group_byFc C s, | ||}t| }|jdtjd |S )zd
Group statistics by key_type. Return a sorted list of Statistic
instances.
Treversekey)r listvaluessortr r$ )r r r Zgroupedr5 r
r
r r5 s zSnapshot.statisticsc C s6 | ||}| ||}t||}|jdtjd |S )z
Compute the differences with an old snapshot old_snapshot. Get
statistics as a sorted list of StatisticDiff instances, grouped by
group_by.
Tr )r r7 r r+ r$ )r Z old_snapshotr r r4 r3 r5 r
r
r
compare_to s
zSnapshot.compare_toN)F)F)
r& r' r( r) r r staticmethodr r r r r5 r r
r
r
r r s
3
r c C s$ t stdt } t }t| |S )zI
Take a snapshot of traces of memory blocks allocated by Python.
zLthe tracemalloc module must be tracing memory allocations to take a snapshot)
is_tracing RuntimeErrorr Zget_traceback_limitr )rm r r
r
r
take_snapshot' s
r ) collections.abcr r functoolsr r r\ Zos.pathro r Z _tracemallocr r r r r+ r7 r8 rE rd re rk rs rt rx r r r r
r
r
r s2 (2&M'0