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� d��ZeG dd� de��Zdd� ZG dd� d�ZG dd� de�Zdd� ZG dd� d�ZG d d � d e�ZG d d � d e�ZG d d!� d!�Zd"d#� ZdS )$� )�Sequence�Iterable)�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)�BZKiBZMiBZGiB�TiB�d 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| jf�S r )�hashr r r �r r
r
r �__hash__* s zStatistic.__hash__c C s2 t |t�stS | 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| jf�S r )r r r r- r r. r r
r
r r T s �zStatisticDiff.__hash__c C sJ t |t�stS | 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| jd�t| 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 )�items�popr+ 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 |t�stS | j|jkS r �r r8 r r: r r
r
r r � s
z Frame.__eq__c C s t |t�stS | j|jk S r rA r r
r
r �__lt__� s
z Frame.__lt__c C s
t | j�S r )r r: r r
r
r r � s zFrame.__hash__c C s d| j | jf S )Nz%s:%s�r= 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 ZdZdZddd�Zedd� �Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd dd�ZdS ) � 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 �tuple�reversedrF 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 | j�S r )�lenrF r r
r
r �__len__� s zTraceback.__len__c C s4 t |t�r"tdd� | j| D ��S t| j| �S d S )Nc s s | ]}t |�V qd S r )r8 ��.0�tracer
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 | j�S r )r rF r r
r
r r � s zTraceback.__hash__c C s t |t�stS | j|jkS r �r rE r rF r r
r
r r � s
zTraceback.__eq__c C s t |t�stS | 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� d�7 }|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@ � linecache�getline�strip)r �limitZmost_recent_first�linesZ 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 |t�stS | j|jkS r )r re r rg r r
r
r r ' s
z Trace.__eq__c C s
t | j�S r )r rg r r
r
r r , s zTrace.__hash__c C s d| j t| jd�f 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 | j�S r )rL rl r r
r
r rM = s z_Traces.__len__c C s4 t |t�r"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 |t�stS | 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�| �} | �d�r"| d d� } | S )Nz.pyc�����)�os�path�normcase�endswith)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�|| j�sdS | 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� fdd�Zedd� �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
ddd�Z ddd�Z 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)�open�pickle�dumpZHIGHEST_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_filters�exclude_filtersrP r
r� r �
_filter_trace� s � �zSnapshot._filter_tracec s� t |t�s tdt|�j ��|rjg �g � |D ] }|jrB��|� q,� �|� q,� ��fdd��jjD �}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 � TypeError�typer& 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 t�y t|
|d�||
< Y qD0 qDn�| jjD ]�}|\}}}} |D ]�}
z ||
}
W nD t�y| |dk�rZ|
f} n|
d dff} t| �}
|
||
< Y n0 z(||
} | j|7 _| jd7 _W n$ t�y� 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.
T��reverse�key)r� �list�values�sortr 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 � std��t� } 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