3
\A � @ 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 sz xtd
D ]l}t | �dk r:|dkr:|r.d| |f S d| |f S t | �dk sN|dkrj|r^d | |f S d | |f S | d
} qW d S )N�B�KiB�MiB�GiB�TiB�d z%+.1f %sz%.1f %s�
i z%+.0f %sz%.0f %s)r r r
r r i ( )�abs)�sizeZsignZunit� r �#/usr/lib64/python3.6/tracemalloc.py� _format_size
s
r c @ s@ e Zd ZdZdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )� Statistic� tracebackr �countc C s || _ || _|| _d S )N)r r r )�selfr r r r r r �__init__% s zStatistic.__init__c C s t | j| j| jf�S )N)�hashr r r )r r r r �__hash__* s zStatistic.__hash__c C s$ | j |j ko"| j|jko"| j|jkS )N)r 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=%iFz , average=%s)r r r r )r �text�averager r r �__str__2 s
zStatistic.__str__c C s d| j | j| jf S )Nz))r r r )r r r r �__repr__< s zStatistic.__repr__c C s | j | j| jfS )N)r r r )r r r r � _sort_key@ s zStatistic._sort_keyN)r r r )
�__name__�
__module__� __qualname__� __slots__r r r r r r! r r r r r s
r c @ s@ e Zd ZdZdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dS )�
StatisticDiffr r � size_diffr �
count_diffc C s" || _ || _|| _|| _|| _d S )N)r r r' r r( )r r r r' r r( r r r r K s
zStatisticDiff.__init__c C s t | j| j| j| j| jf�S )N)r r r r' r r( )r r r r r R s zStatisticDiff.__hash__c C s< | j |j ko:| j|jko:| j|jko:| j|jko:| j|jkS )N)r r r' r r( )r r r r r r V 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)FTz , average=%s)r r r r' r r( )r r r r r r r ] s
zStatisticDiff.__str__c C s d| j | j| j| j| jf S )Nz9)r r r' r r( )r r r r r i s zStatisticDiff.__repr__c C s t | j�| jt | j�| j| jfS )N)r r' r r( r r )r r r r r! n s zStatisticDiff._sort_keyN)r r r' r r( )
r" r# r$ r% r r r r r r! r r r r r& D s r& c C s� g }xp|j � D ]d\}}| j|d �}|d k rPt||j|j|j |j|j|j �}nt||j|j|j|j�}|j|� qW x6| j � D ]*\}}t|d|j d|j �}|j|� q�W |S )Nr )�items�popr& r r �append)� old_group� new_group�
statisticsr �statZpreviousr r r �_compare_grouped_statst s r0 c @ sX e Zd ZdZdd� Zedd� �Zedd� �Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dS )�Frame�_framec C s
|| _ d S )N)r2 )r �framer r r r � s zFrame.__init__c C s
| j d S )Nr )r2 )r r r r �filename� s zFrame.filenamec C s
| j d S )N� )r2 )r r r r �lineno� s z Frame.linenoc C s | j |j kS )N)r2 )r r r r r r � s z Frame.__eq__c C s | j |j k S )N)r2 )r r r r r �__lt__� s z Frame.__lt__c C s
t | j�S )N)r r2 )r r r r r � s zFrame.__hash__c C s d| j | jf S )Nz%s:%s)r4 r6 )r r r r r � s z
Frame.__str__c C s d| j | jf S )Nz )r4 r6 )r r r r r � s zFrame.__repr__N)r2 )
r" r# r$ r% r �propertyr4 r6 r r7 r r r r r r r r1 � s r1 c @ sb e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z ddd�Z
dS )� Traceback�_framesc C s t j| � || _d S )N)r r r: )r �framesr r r r � s
zTraceback.__init__c C s
t | j�S )N)�lenr: )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 )N)r1 )�.0�tracer r r � � s z(Traceback.__getitem__..)�
isinstance�slice�tupler: r1 )r �indexr r r � __getitem__� s
zTraceback.__getitem__c C s |j | jkS )N)r2 r: )r r3 r r r � __contains__� s zTraceback.__contains__c C s
t | j�S )N)r r: )r r r r r � s zTraceback.__hash__c C s | j |j kS )N)r: )r r r r r r � s zTraceback.__eq__c C s | j |j k S )N)r: )r r r r r r7 � s zTraceback.__lt__c C s t | d �S )Nr )�str)r r r r r � s zTraceback.__str__c C s dt | �f S )Nz)rC )r r r r r � s zTraceback.__repr__Nc C sn g }|d k r|dk r|S xP| d |� D ]@}|j d|j|jf � tj|j|j�j� }|r&|j d| � q&W |S )Nr z File "%s", line %sz %s)r+ r4 r6 � linecache�getline�strip)r �limit�linesr3 �liner r r �format� s zTraceback.format)r: )N)r" r# r$ r% r r= rE rF r r r7 r r rN r r r r r9 � s r9 c C s t | �}|d k rt|�S d S d S )N)r r9 )�objr; r r r �get_object_traceback� s rP c @ s\ e Zd ZdZdd� Zedd� �Zedd� �Zedd � �Zd
d � Z d d
� Z
dd� Z dd� Z dS )�Trace�_tracec C s
|| _ d S )N)rR )r r? r r r r � s zTrace.__init__c C s
| j d S )Nr )rR )r r r r �domain� s z Trace.domainc C s
| j d S )Nr5 )rR )r r r r r � s z
Trace.sizec C s t | jd �S )N� )r9 rR )r r r r r s zTrace.tracebackc C s | j |j kS )N)rR )r r r r r r s z Trace.__eq__c C s
t | j�S )N)r rR )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)rS r r r )r r r r r s zTrace.__repr__N)rR )
r" r# r$ r% r r8 rS r r r r r r r r r r rQ � s rQ c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�_Tracesc C s t j| � || _d S )N)r r �_traces)r �tracesr r r r s
z_Traces.__init__c C s
t | j�S )N)r< rV )r r r r r= 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 )N)rQ )r> r? r r r r@ s z&_Traces.__getitem__..)rA rB rC rV rQ )r rD r r r rE s
z_Traces.__getitem__c C s |j | jkS )N)rR rV )r r? r r r rF $ s z_Traces.__contains__c C s | j |j kS )N)rV )r r r r r r ' s z_Traces.__eq__c C s dt | � S )Nz)r< )r r r r r * s z_Traces.__repr__N) r" r# r$ r r= rE rF r r r r r r rU s rU c C s& t jj| �} | jd�r"| d d� } | S )Nz.pycr5 �����)�os�path�normcase�endswith)r4 r r r �_normalize_filename. s
r] c @ s e Zd Zdd� Zdd� ZdS )�
BaseFilterc C s
|| _ d S )N)� inclusive)r r_ r r r r 6 s zBaseFilter.__init__c C s t �d S )N)�NotImplementedError)r r? r r r �_match9 s zBaseFilter._matchN)r" r# r$ r ra r r r r r^ 5 s r^ c sJ e Zd Zd� fdd� Zedd� �Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�FilterNFc s2 t � j|� || _t|�| _|| _|| _|| _d S )N)�superr r_ r] �_filename_patternr6 �
all_framesrS )r r_ �filename_patternr6 re rS )� __class__r r r > s
zFilter.__init__c C s | j S )N)rd )r r r r rf G s zFilter.filename_patternc C s6 t |�}tj|| j�sdS | jd kr(dS || jkS d S )NFT)r] �fnmatchrd r6 )r r4 r6 r r r �_match_frame_implK s
zFilter._match_frame_implc C s | j ||�| j A S )N)ri r_ )r r4 r6 r r r � _match_frameT s zFilter._match_framec sH � j r,t� fdd�|D ��r"� jS � j S n|d \}}� j||�S d S )Nc 3 s | ]\}}� j ||�V qd S )N)ri )r> r4 r6 )r r r r@ Y s z*Filter._match_traceback..r )re �anyr_ rj )r r r4 r6 r )r r �_match_tracebackW s
zFilter._match_tracebackc C sD |\}}}| j |�}| jd k r@| jr2|o0|| jkS |p>|| jkS |S )N)rl rS r_ )r r? rS r r �resr r r ra b s
z
Filter._match)NFN) r" r# r$ r r8 rf ri rj rl ra �
__classcell__r r )rg r rb = s rb c s0 e Zd Z� fdd�Zedd� �Zdd� Z� ZS )� DomainFilterc s t � j|� || _d S )N)rc r �_domain)r r_ rS )rg r r r n s zDomainFilter.__init__c C s | j S )N)rp )r r r r rS r s zDomainFilter.domainc C s |\}}}|| j k| j A S )N)rS r_ )r r? rS r r r r r ra v s
zDomainFilter._match)r" r# r$ r r8 rS ra rn r r )rg r ro m s ro c @ sT e Zd Zdd� Zdd� Zedd� �Zdd� Zd d
� Zd d � Z ddd�Z
ddd�Z dS )�Snapshotc C s t |�| _|| _d S )N)rU rW �traceback_limit)r rW rr r r r r � s
zSnapshot.__init__c C s* t |d��}tj| |tj� W d Q R X d S )N�wb)�open�pickle�dumpZHIGHEST_PROTOCOL)r r4 �fpr r r rv � s z
Snapshot.dumpc C s t | d�� }tj|�S Q R X d S )N�rb)rt ru �load)r4 rw r r r ry � s z
Snapshot.loadc s@ |r t � fdd�|D ��s dS |r � trace_filter)r? r r r@ � s z)Snapshot._filter_trace..Fc 3 s | ]}|j � � V qd S )N)ra )r> rz )r? r r r@ � s T)rk )r �include_filters�exclude_filtersr? r )r? r �
_filter_trace� s
zSnapshot._filter_tracec s� t |t�s tdt|�j ��|rng �g � x(|D ] }|jrD�j|� q.� j|� q.W � ��fdd��jjD �}n �jjj � }t
|�j �S )Nz)filters must be a list of filters, not %sc s g | ]}�j �� |�r|�qS r )r} )r> r? )r| r{ r r r �
� s z*Snapshot.filter_traces..) rA r � TypeError�typer" r_ r+ rW rV �copyrq rr )r �filtersrz Z
new_tracesr )r| r{ r r �
filter_traces� s
zSnapshot.filter_tracesc
C s� |dkrt d|f ��|r.|d kr.t d| ��i }i }|�sx�| jjD ]�}|\}}}y || } W nZ tk
r� |dkr~|}
n(|dkr�|d d� }
n|d d dff}
t|
�} | ||< Y nX y(|| } | j|7 _| jd7 _W qF tk
�r
t| |d�|| < Y qFX qFW n�x�| jjD ]�}|\}}}x�|D ]�} y || } W nF tk
�r� |dk�rd| f}
n| d dff}
t|
�} | || < Y nX y(|| } | j|7 _| jd7 _W n& tk
�r� t| |d�|| < Y nX �q0W �q W |S )
Nr r4 r6 zunknown key_type: %rz/cumulative mode cannot by used with key type %rr5 r )r r4 r6 )r6 r4 )�
ValueErrorrW rV �KeyErrorr9 r r r )
r �key_type�
cumulativeZstatsZ
tracebacksr? rS r Ztrace_tracebackr r; r/ r3 r r r � _group_by� sX
"zSnapshot._group_byFc C s, | j ||�}t|j� �}|jdtjd� |S )NT)�reverse�key)r� �list�values�sortr r! )r r� r� Zgroupedr. r r r r. � s zSnapshot.statisticsc C s6 | j ||�}|j ||�}t||�}|jdtjd� |S )NT)r� r� )r� r0 r� r& r! )r Z old_snapshotr� r� r- r, r. r r r �
compare_to� s
zSnapshot.compare_toN)F)F) r" r# r$ r rv � staticmethodry r} r� r� r. r� r r r r rq { s 3
rq c C s$ t � std��t� } t� }t| |�S )NzLthe tracemalloc module must be tracing memory allocations to take a snapshot)Z
is_tracing� RuntimeErrorr Zget_traceback_limitrq )rW rr r r r �
take_snapshot s
r� ) � collectionsr r � functoolsr rh rH Zos.pathrY ru Z _tracemallocr r r r r& r0 r1 r9 rP rQ rU r] r^ rb ro rq r� r r r r � s0 &0# 5%0