← Back
Editing: tracemalloc.cpython-310.pyc
o 0�iF � @ 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 |rd| |f S d| |f S t | �dk s*|dkr<|r4d| |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 �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| jf�S r )�hashr r r �r r r r �__hash__* � zStatistic.__hash__c C s2 t |t�stS | 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)<Statistic traceback=%r size=%i count=%i>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| jf�S r )r r r r0 r r1 r r r r r T s �zStatisticDiff.__hash__c C sJ t |t�stS | 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| jd�t| 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 )Nz9<StatisticDiff traceback=%r size=%i (%+i) count=%i (%+i)>r/ 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 )�items�popr. 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 |t�stS | j|jkS r �r r; r r= r r r r r � � z Frame.__eq__c C � t |t�stS | j|jk S r rH r r r r �__lt__� rI z Frame.__lt__c C � t | j�S r )r r= r r r r r � � zFrame.__hash__c C � d| j | jf S )Nz%s:%s�rC rF r r r r r$ � r' z Frame.__str__c C rN )Nz <Frame filename=%r lineno=%r>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 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�reversedrR 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 |t�rtdd� | j| D ��S t| j| �S )Nc s � � | ]}t |�V qd S r )r; ��.0�tracer r r � <genexpr>� � � z(Traceback.__getitem__.<locals>.<genexpr>)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� d�7 }|S )Nz <Traceback �>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 � linecache�getline�strip)r �limitZmost_recent_first�linesZ 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| 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'<Trace domain=%s size=%s, traceback=%r>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__.<locals>.<genexpr>)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<Traces len=%s>)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�| �} | �d�r| d d� } | S )Nz.pyc���)�os�path�normcase�endswith)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�|| j�s 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.<locals>.<genexpr>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� fdd�Zedd� �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 ddd�Z ddd�Z 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)�open�pickle�dumpZHIGHEST_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_filter�r` r r ra � s � �z)Snapshot._filter_trace.<locals>.<genexpr>Fc 3 s � | ] }|� � � V qd S r r� r� r� r r ra � s � �T)r� )r �include_filters�exclude_filtersr` r r� r � _filter_trace� s � �zSnapshot._filter_tracec s� t |t�stdt|�j ��|r5g �g � |D ]}|jr!��|� 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� r^ �r� r� r r r � <listcomp>� s � �z*Snapshot.filter_traces.<locals>.<listcomp>) r r � TypeError�typer) 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 q�w 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. T��reverse�key)r� �list�values�sortr 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 � 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� )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 �<module> s4 (2&M'0
Save File
Cancel