o
0iF @ 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 |rd| |f S d| |f S t | dk s*|dkr<|r4d| |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
2/opt/alt/python310/lib64/python3.10/tracemalloc.py _format_size
s
r c @ D 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__* zStatistic.__hash__c C s2 t |tstS | j|jko| j|jko| 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 zStatistic._sort_keyN __name__
__module__ __qualname____doc__ __slots__r r r r$ r% r& r
r
r
r r s
r c @ r )
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 r0 r r1 r
r
r r M s
zStatisticDiff.__init__c C s t | j| j| j| j| jfS r )r r r r0 r r1 r r
r
r r T s zStatisticDiff.__hash__c C sJ t |tstS | j|jko$| j|jko$| j|jko$| j|jko$| j|jkS r )r r. r r r r0 r r1 r r
r
r r X s
zStatisticDiff.__eq__c C sP d| j t| jdt| jd| j| jf }| jr&| j| j }|dt|d 7 }|S )Nz %s: size=%s (%s), count=%i (%+i)FTr! )r r r r0 r r1 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 r0 r r1 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 ]2\}}| |d }|d ur't||j|j|j |j|j|j }n t||j|j|j|j}|| q| D ]\}}t|d|j d|j }|| q=|S Nr )itemspopr. r r append) old_group new_group
statisticsr statZpreviousr
r
r _compare_grouped_statsx s" r: 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
|| _ d S r r< r framer
r
r r
zFrame.__init__c C
| j d S r2 r< r r
r
r filename rA zFrame.filenamec C rB N r< r r
r
r lineno rA z Frame.linenoc C t |tstS | j|jkS r r r; r r= r r
r
r r
z Frame.__eq__c C t |tstS | j|jk S r rH r r
r
r __lt__ rI z Frame.__lt__c C
t | jS r )r r= r r
r
r r
zFrame.__hash__c C d| j | jf S )Nz%s:%srC rF r r
r
r r$ r' z
Frame.__str__c C rN )Nz rO r r
r
r r% r' zFrame.__repr__N)r) r* r+ r, r- r propertyrC rF r rK r r$ r% r
r
r
r r; s
r; 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 tuplereversedrR rS )r frames total_nframer
r
r r s
zTraceback.__init__c C | j S r )rS r r
r
r rW zTraceback.total_nframec C rL r )lenrR r r
r
r __len__ rM zTraceback.__len__c C 0 t |trtdd | j| D S t| j| S )Nc s | ]}t |V qd S r )r; .0tracer
r
r z(Traceback.__getitem__..)r slicerT rR r; r indexr
r
r __getitem__
zTraceback.__getitem__c C |j | jv S r )r= rR r? r
r
r __contains__ zTraceback.__contains__c C rL r )r rR r r
r
r r rM zTraceback.__hash__c C rG r r rQ r rR r r
r
r r rI zTraceback.__eq__c C rJ r rk r r
r
r rK rI zTraceback.__lt__c C s t | d S r2 )strr r
r
r r$ rj zTraceback.__str__c C s: dt | }| jd u r|d7 }|S |d| j d7 }|S )Nz z total_nframe=)rT rS rW )r sr
r
r r% s
zTraceback.__repr__Fc C s g }|d ur|dkr| | d }n | d | }n| }|r!t |}|D ] }|d|j|jf t|j|j }|rC|d| q#|S )Nr z File "%s", line %sz %s)rU r5 rC rF linecachegetlinestrip)r limitZmost_recent_firstlinesZ frame_slicer@ liner
r
r format s"
zTraceback.formatr )NF)r) r* r+ r, r- r rP rW r[ rf ri r r rK r$ r% ru r
r
r
r rQ s
rQ c C s t | }|dur t|S 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 rQ )objrV r
r
r get_object_traceback s rw 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 r> r ry r r` r
r
r r s
zTrace.__init__c C rB r2 ry r r
r
r domain rA z Trace.domainc C rB rD ry r r
r
r r rA z
Trace.sizec C s t | jdd S )N )rQ rz r r
r
r r # s zTrace.tracebackc C rG r )r rx r rz r r
r
r r ' rI z Trace.__eq__c C rL r )r rz r r
r
r r , rM 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)r| r r r r r
r
r r% 2 s zTrace.__repr__N)r) r* r+ r, r- r rP r| r r r r r$ r% r
r
r
r rx s
rx 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 rL r )rZ r r r
r
r r[ = rM z_Traces.__len__c C r\ )Nc s r] r )rx r^ r
r
r ra B rb z&_Traces.__getitem__..)r rc rT r rx rd r
r
r rf @ rg z_Traces.__getitem__c C rh r )rz r r{ r
r
r ri F rj z_Traces.__contains__c C rG r )r r~ r r r r
r
r r I rI z_Traces.__eq__c C s dt | S )Nz)rZ r r
r
r r% N rj z_Traces.__repr__N) r) r* r+ r r[ rf ri r r% r
r
r
r r~ 7 s r~ c C s& t j| } | dr| d d } | S )Nz.pyc)ospathnormcaseendswith)rC r
r
r _normalize_filenameR s
r c @ s e Zd Zdd Zdd ZdS )
BaseFilterc C r> r ) inclusive)r r r
r
r r Z rM zBaseFilter.__init__c C s t r )NotImplementedErrorr{ r
r
r _match] s zBaseFilter._matchN)r) r* r+ r r r
r
r
r r Y s r c sL e Zd Z d fdd Zedd Zdd Zd d
Zd d Zd
d Z Z
S )FilterNFc s2 t | || _t|| _|| _|| _|| _d S r )superr r r _filename_patternrF
all_framesr| )r r filename_patternrF r r| __class__r
r r b s
zFilter.__init__c C rX r )r r r
r
r r k rY zFilter.filename_patternc C s2 t |}t|| js
dS | jd u rdS || jkS )NFT)r fnmatchr rF r rC rF r
r
r _match_frame_implo s
zFilter._match_frame_implc C s | ||| j A S r )r r r r
r
r _match_framex r zFilter._match_framec sB j rt fdd|D r jS j S |d \}} ||S )Nc 3 s | ] \}} ||V qd S r )r )r_ rC rF r r
r ra } s z*Filter._match_traceback..r )r anyr r )r r rC rF r
r r _match_traceback{ s zFilter._match_tracebackc C sF |\}}}}| |}| jd ur!| jr|o|| jkS |p || jkS |S r )r r| r )r r` r| r r rW resr
r
r r s
z
Filter._match)NFN) r) r* r+ r rP r r r r r
__classcell__r
r
r r r a s
r c s0 e Zd Z fddZedd Zdd Z ZS ) DomainFilterc s t | || _d S r )r r _domain)r r r| r r
r r s
zDomainFilter.__init__c C rX r )r r r
r
r r| rY zDomainFilter.domainc C s |\}}}}|| j k| j A S r )r| r )r r` r| r r rW r
r
r r s zDomainFilter._match)r) r* r+ r rP r| r 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 )r~ r traceback_limit)r r r r
r
r r s
zSnapshot.__init__c C s@ t |d}t| |tj W d dS 1 sw Y dS )z1
Write the snapshot into a file.
wbN)openpickledumpZHIGHEST_PROTOCOL)r rC fpr
r
r r s "z
Snapshot.dumpc C s8 t | d
}t|W d S 1 sw Y dS )z.
Load a snapshot from a file.
rbN)r r load)rC r r
r
r r s $z
Snapshot.loadc s@ |rt fdd|D sdS |r t fdd|D r dS dS )Nc 3 s | ]}| V qd S r r r_ trace_filterr` r
r ra s z)Snapshot._filter_trace..Fc 3 s | ] }| V qd S r r r r r
r ra s T)r )r include_filtersexclude_filtersr` r
r r
_filter_trace s zSnapshot._filter_tracec s t |tstdt|j |r5g g |D ]}|jr!| q | q fddjjD }njj }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 r^ r r r r
r
s
z*Snapshot.filter_traces..) r r TypeErrortyper) r r5 r r copyr r )r filtersr Z
new_tracesr
r r
filter_traces s
zSnapshot.filter_tracesc
C s |dvr t d|f |r|dvrt d| i }i }|s| jjD ]`}|\}}}} z|| }
W n, ty[ |dkr=|} n|dkrH|d d } n |d d dff} t| }
|
||< Y nw z||
} | j|7 _| jd7 _W q! ty t|
|d||
< Y q!w |S | jjD ]Y}|\}}}} |D ]N}
z||
}
W n ty |dkr|
f} n|
d dff} t| }
|
||
< Y nw z||
} | j|7 _| jd7 _W q ty t|
|d||
< Y qw q|S ) N)r rC rF zunknown key_type: %r)rF rC z/cumulative mode cannot by used with key type %rr rF rE r )
ValueErrorr r KeyErrorrQ r r r )r key_type
cumulativeZstatsZ
tracebacksr` r| r Ztrace_tracebackrW r rV r9 r@ r
r
r _group_by sf 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 Zgroupedr8 r
r
r r8 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 r: r r. r& )r Z old_snapshotr r r7 r6 r8 r
r
r
compare_to s
zSnapshot.compare_toN)F)
r) r* r+ r, r r staticmethodr r r r r8 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 )r r r
r
r
take_snapshot' s
r ) Zcollections.abcr r functoolsr r ro Zos.pathr r Z _tracemallocr r r r r. r: r; rQ rw rx r~ r r r r r r r
r
r
r s4 (2&M'0