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 @ B e Zd Z dZdd Zdd Zdd Zdd Zd
d Zd d
Z dS ) Statistic 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__ __slots__r r r r$ r% r& r
r
r
r r s
r c @ r )
StatisticDiffr r size_diffr
count_diffc C s" || _ || _|| _|| _|| _d S r r. )r r r r/ r r0 r
r
r r M s
zStatisticDiff.__init__c C s t | j| j| j| j| jfS r )r r r r/ r r0 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 r/ r r0 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 r/ r r0 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 r0 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" r9 c @ sZ e Zd Z dZdd Zedd Zedd Zdd Zd
d Z d d
Z
dd Z dd Z dS )Frame_framec C
|| _ d S r r; r framer
r
r r
zFrame.__init__c C
| j d S r1 r; r r
r
r filename r@ zFrame.filenamec C rA N r; r r
r
r lineno r@ 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 rG r r
r
r __lt__ rH 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:%srB rE r r
r
r r$ r' z
Frame.__str__c C rM )Nz rN r r
r
r r% r' zFrame.__repr__N)
r) r* r+ r, r propertyrB rE r rJ r r$ r% r
r
r
r r: s
r: c @ sr e Zd Z dZdddZedd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dd ZdddZdS ) Traceback)_frames
_total_nframeNc C s" t | tt|| _|| _d S r )r r tuplereversedrQ rR )r frames total_nframer
r
r r s
zTraceback.__init__c C | j S r )rR r r
r
r rV zTraceback.total_nframec C rK r )lenrQ r r
r
r __len__ rL 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 slicerS rQ r: r indexr
r
r __getitem__
zTraceback.__getitem__c C |j | jv S r )r< rQ r> r
r
r __contains__ zTraceback.__contains__c C rK r )r rQ r r
r
r r rL zTraceback.__hash__c C rF r r rP r rQ r r
r
r r rH zTraceback.__eq__c C rI r rj r r
r
r rJ rH zTraceback.__lt__c C s t | d S r1 )strr r
r
r r$ ri zTraceback.__str__c C s: dt | }| jd u r|d7 }|S |d| j d7 }|S )Nz z total_nframe=)rS rR rV )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)rT r4 rB rE linecachegetlinestrip)r limitZmost_recent_firstlinesZ frame_slicer? liner
r
r format s"
zTraceback.formatr )NF)r) r* r+ r, r rO rV rZ re rh r r rJ r$ r% rt r
r
r
r rP s
rP c C s t | }|d ur
t|S d S r )r rP )objrU r
r
r get_object_traceback s
rv c @ s^ e Zd Z dZdd Zedd Zedd Zedd Zd
d Z d d
Z
dd Z dd Z dS )Trace_tracec C r= r rx r r_ r
r
r r s
zTrace.__init__c C rA r1 rx r r
r
r domain r@ z Trace.domainc C rA rC rx r r
r
r r r@ z
Trace.sizec C s t | jdd S )N )rP ry r r
r
r r # s zTrace.tracebackc C rF r )r rw r ry r r
r
r r ' rH z Trace.__eq__c C rK r )r ry r r
r
r r , rL 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 rO r{ r r r r r$ r% r
r
r
r rw s
rw 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 rK r )rY r~ r r
r
r rZ = rL z_Traces.__len__c C r[ )Nc s r\ r )rw r] r
r
r r` B ra z&_Traces.__getitem__..)r rb rS r~ rw rc r
r
r re @ rf z_Traces.__getitem__c C rg r )ry r~ rz r
r
r rh F ri z_Traces.__contains__c C rF r )r r} r r~ r r
r
r r I rH z_Traces.__eq__c C s dt | S )Nz)rY r r
r
r r% N ri z_Traces.__repr__N) r) r* r+ r rZ re rh r r% r
r
r
r r} 7 s r} c C s& t j| } | dr| d d } | S )Nz.pyc)ospathnormcaseendswith)rB 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 rL zBaseFilter.__init__c C s t r )NotImplementedErrorrz 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_patternrE
all_framesr{ )r r filename_patternrE r r{ __class__r
r r b s
zFilter.__init__c C rW r )r r r
r
r r k rX zFilter.filename_patternc C s2 t |}t|| js
dS | jd u rdS || jkS )NFT)r fnmatchr rE r rB rE 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^ rB rE r r
r r` } s z*Filter._match_traceback..r )r anyr r )r r rB rE 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 rV resr
r
r r s
z
Filter._match)NFN) r) r* r+ r rO 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 rW r )r r r
r
r r{ rX zDomainFilter.domainc C s |\}}}}|| j k| j A S r )r{ r )r r_ r{ r r rV r
r
r r s zDomainFilter._match)r) r* r+ r rO r{ r r r
r
r r r s
r c @ sV e Zd Z dd Zdd Zedd Zdd Zd d
Zd d Z dddZ
dddZ dS )Snapshotc C s t || _|| _d S r )r} r traceback_limit)r r r r
r
r r s
zSnapshot.__init__c C sB t |d}t| |tj W d d S 1 sw Y d S )Nwb)openpickledumpZHIGHEST_PROTOCOL)r rB fpr
r
r r s "z
Snapshot.dumpc C s: t | d
}t|W d S 1 sw Y d S )Nrb)r r load)rB 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 r` s z)Snapshot._filter_trace..Fc 3 s | ] }| V qd S r r r r r
r r` s T)r )r include_filtersexclude_filtersr_ r
r r
_filter_trace s zSnapshot._filter_tracec s t |tstdt|j |r6g g |D ]}|jr"| q | q fddjjD }njj }t
|j S )Nz)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 r4 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 rB rE zunknown key_type: %r)rE rB z/cumulative mode cannot by used with key type %rr rE rD r )
ValueErrorr r~ KeyErrorrP r r r )r key_type
cumulativeZstatsZ
tracebacksr_ r{ r Ztrace_tracebackrV r rU r8 r? r
r
r _group_by sf zSnapshot._group_byFc C s. | ||}t| }|jdtjd |S NT)reversekey)r listvaluessortr r& )r r r Zgroupedr7 r
r
r r7 s
zSnapshot.statisticsc C s8 | ||}| ||}t||}|jdtjd |S r )r r9 r r- r& )r Z old_snapshotr r r6 r5 r7 r
r
r
compare_to s
zSnapshot.compare_toN)F) r) r* r+ r r staticmethodr r r r r7 r r
r
r
r r s
3
r c C s& t stdt } t }t| |S )NzLthe 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 rn Zos.pathr r Z _tracemallocr r r r r- r9 r: rP rv rw r} r r r r r r r
r
r
r s4 (2&M'0