o
6��f�r � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z g d�Z
G dd� dee�Ze
d d
�G d d � d ��Ze
d d
�G d
d� d��ZG dd� d�ZG dd� d�Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zed!kr�d dl Z zd dl Z W n e y� Y nw G d"d#� d#e j �Z e!e j"�d$kr�e j"d$ Z#ndZ#z(e e#�Z$e j"d%d� D ]Z%e$�&e%� q�e'd&e$j(d'� e$�)� e'd(e$j(d'� W dS e*y� Y dS w dS ))� N)�Enum)�
cmp_to_key)� dataclass)�Dict)�Stats�SortKey�FunctionProfile� StatsProfilec @ s8 e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
d � Z d S )
r )�calls�ncalls)�
cumulative�cumtime)�filename�module�line�name�nfl�pcalls�stdname)�time�tottimec G s@ |d }t �| |�}||_|dd � D ]}|| j|< q||_|S �Nr � )�str�__new__�_value_�_value2member_map_Z _all_values)�cls�values�value�objZ other_value� r! �-/opt/alt/python310/lib64/python3.10/pstats.pyr / s zSortKey.__new__N)
�__name__�
__module__� __qualname__ZCALLSZ
CUMULATIVEZFILENAMEZLINE�NAMEZNFLZPCALLSZSTDNAMEZTIMEr r! r! r! r" r $ s r T)Z unsafe_hashc @ sF e Zd ZU eed< eed< eed< eed< eed< eed< eed< dS ) r r r �percall_tottimer
�percall_cumtime� file_name� line_numberN)r# r$ r% r �__annotations__�float�intr! r! r! r" r 9 s
r c @ s( e Zd ZU eed< eeef ed< dS )r �total_tt�
func_profilesN)r# r$ r% r, r+ r r r r! r! r! r" r C s
r c @ s� e Zd Z dd�dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdddddddddddddd�
Z dd� Z
dd � Z d d � Z d d � Z
d!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd8d2d3�Zd4d5� Zd6d7� ZdS )9r N)�streamc G sF |pt j| _t|�s
d }n
|d }|dd � }| �|� | j|� d S r )�sys�stdoutr0 �len�init�add)�selfr0 �args�argr! r! r" �__init__k s
zStats.__init__c C s� d | _ g | _d | _d| _d| _d| _d| _t� | _i | _ i | _
| � |� z| � � W d S t
yD td| jr<| jd nd | jd� � w )Nr zInvalid timing data %s���� ��file)� all_callees�files�fcn_listr. � total_calls�
prim_calls� max_name_len�set� top_level�stats�
sort_arg_dict�
load_stats�get_top_level_stats� Exception�printr0 )r6 r8 r! r! r" r4 u s*
���z
Stats.initc C s� |d u r i | _ d S t|t�rDt|d��}t�|�| _ W d � n1 s$w Y zt�|�}t� |j
�d | }W n Y |g| _ nt |d�rT|�
� |j | _ i |_ | j s`td| j|f ��d S )N�rbz � create_statsz.Cannot create or construct a %r object from %r)rF �
isinstancer �open�marshal�load�os�statr �ctime�st_mtimer? �hasattrrM � TypeError� __class__)r6 r8 �fZ
file_statsr! r! r" rH � s,
�
�zStats.load_statsc C s� | j �� D ]8\}\}}}}}| j|7 _| j|7 _| j|7 _d|v r-| j�|� tt|��| j kr=tt|��| _ qd S )N)Zjprofiler Zprofiler)
rF �itemsrA rB r. rE r5 r3 �func_std_stringrC )r6 �func�cc�nc�tt�ct�callersr! r! r" rI � s ��zStats.get_top_level_statsc G s� |s| S t |�D ]k}t| �t|�krt|�}| j|j7 _| j|j7 _| j|j7 _| j|j7 _|jD ]}| j�|� q9| j |j k rL|j | _ d | _
|j � � D ] \}}|| j v rc| j | }nddddi f}t
||�| j |< qTq| S �Nr )�reversed�typer r? rA rB r. rE r5 rC r@ rF rZ �add_func_stats)r6 �arg_list�itemr\ rS Z
old_func_statr! r! r" r5 � s*
�z Stats.addc C s@ t |d��}t�| j|� W d � d S 1 sw Y d S )N�wb)rO rP �dumprF )r6 r rY r! r! r" �
dump_stats� s "�zStats.dump_stats)))r r: z
call count)))� r: zcumulative time))�� r z file name))�� r z line number))�� r z
function name))rp rl rn zname/file/line)))r r: zprimitive call count)))� r z
standard name)))� r: z
internal time)
r
r r
r r r r r r r r r r c C sx | j s9i | _ }i }| j�� D ] \}}|}|r0|sn||v r$d||< n |||< |d d� }|sq|D ]}||= q3| j S )Nr r: )rG �sort_arg_dict_defaultrZ )r6 �dictZbad_list�word�tupZfragmentr! r! r" �get_sort_arg_defs� s&
��zStats.get_sort_arg_defsc G sZ |sd| _ | S t|�dkr!t|d t�r!ddddd�|d g}n t|�dkr>|dd � D ]}t|�t|d �kr=td ��q-| �� }d
}d | _d }|D ]!}t|t�rU|j }||| d }| j||| d 7 _d }qKg }| j
� � D ]\}\} }
} } }
|� | |
| | f| t
|�|f � qt|jtt|�j�d
� g | _ }|D ] }|� |d � q�| S )Nr r r r
r r )r: r r rs rs z Can't have mixed argument typer! r; z, )�keyr: )r@ r3 rN r- rd rW rx � sort_typer r rF rZ �appendr[ �sortr � TupleComp�compare)r6 Zfieldr8 Z
sort_arg_defsZ
sort_tupleZ connectorrv Z
stats_listr\ r] r^ r_ r` ra r@ �tupler! r! r" �
sort_stats� sH �� �
�
zStats.sort_statsc C s | j r| j �� | S �N)r@ �reverse�r6 r! r! r" �
reverse_order s
zStats.reverse_orderc C s� | j }i | _ }d}|�� D ]H\}\}}}}} t|�}
tt|
��|kr)tt|
��}i } | �� D ]
\} }
|
| t| �< q/|
|v rMt||
||||| f�||
< q||||| f||
< q| j}t� | _}|D ] }|�t|�� qb|| _ d | _
d | _ | S rb ) rF rZ �func_strip_pathr3 r[ re rE rD r5 rC r@ r> )r6 ZoldstatsZnewstatsrC r\ r] r^ r_ r` ra ZnewfuncZ
newcallers�func2�callerZold_topZnew_topr! r! r" �
strip_dirs s0
� zStats.strip_dirsc
C st | j rd S i | _ }| j�� D ](\}\}}}}}||vr i ||< |�� D ]\}} ||vr0i ||< | || |< q$qd S r� )r> rF rZ )
r6 r> r\ r] r^ r_ r` ra r� r� r! r! r" � calc_callees5 s
�zStats.calc_calleesc C s |}t |t�r7zt�|�}W n tjy" |d| 7 }||f Y S w g }|D ]}|�t|��r5|�|� q'n=t|�}t |t �r[d| krJdk r[n nt
|| d �}|d |� }nt |t
�rtd| krj|k rtn n|}|d |� }t|�t|�kr�|dt|�t|�|f 7 }||fS )Nz# <Invalid regular expression %r>
g g �?g �?r z6 List reduced from %r to %r due to restriction <%r>
) rN r �re�compile�error�searchr[ r{ r3 r, r- )r6 Zsel�list�msgZnew_listZrexr\ �countr! r! r" �eval_print_amountH s4
�
��"" �zStats.eval_print_amountc C s
| j r | j d d � nt| j�� �}|stdi �S tt| j��}i }t||�}|D ]Y}| j| \}}}} }
|\} } }
||krBt|�n t|�d t|� }tt|��}|dkrXdntt|| ��}tt| ��}|dkrldntt| | ��}t |||||| | �}|||
< q)|S )Nr �/r: )
r@ r� rF �keysr r, �f8r. r r )r6 Z func_listr. r/ Z
stats_profiler\ r] r^ r_ r` ra r) r* � func_namer r r' r
r( Z func_profiler! r! r" �get_stats_profileb s4 "
$ �
zStats.get_stats_profilec C s� | j }| jr| jd d � }d| j d }n t| j�� �}d}|D ] }| �|||�\}}q t|�}|s6d|fS t|| j d� |t| j�k rYd}|D ]}tt
|��|krXtt
|��}qH|d |fS )Nz Ordered by: �
z! Random listing order was used
r r< rs ) rC r@ rz r� rF r� r� r3 rK r0 r[ )r6 Zsel_list�widthZ stat_listr� Z selectionr� r\ r! r! r" �get_print_list� s& � zStats.get_print_listc G s� | j D ] }t|| jd� q| j rt| jd� d}| jD ] }t|t|�| jd� qt|| jdd| jd� | j| jkrDtd| j d| jd� td| j | jd� t| jd� | �|�\}}|rw| � � |D ]}| �
|� qct| jd� t| jd� | S )Nr< � zfunction calls� ��endr= z(%d primitive calls)zin %.3f seconds) r? rK r0 rE �func_get_function_namerA rB r. r� � print_title�
print_line)r6 �amountr �indentr\ r� r� r! r! r" � print_stats� s(
zStats.print_statsc G sz | � |�\}}|r;| �� | �|d� |D ]}|| jv r'| �||| j| � q| �||i � qt| jd� t| jd� | S )Nz called...r< )r� r� �print_call_headingr> �print_call_linerK r0 )r6 r� r� r� r\ r! r! r" �
print_callees� s
zStats.print_calleesc
G sh | � |�\}}|r2| �|d� |D ]}| j| \}}}}} | �||| d� qt| jd� t| jd� | S )Nzwas called by...z<-r< )r� r� rF r� rK r0 )
r6 r� r� r� r\ r] r^ r_ r` ra r! r! r" �
print_callers� s zStats.print_callersc
C sz t d�|�| | jd� d}| j�� D ]\}}}}}|r+tt|�� ��} t| t�} nq|r;t d| d | jd� d S d S )Nz Function r< Fr� z ncalls tottime cumtime) rK �ljustr0 rF r �next�iterrN r )
r6 � name_sizeZ column_titleZ subheaderr] r^ r_ r` ra r r! r! r" r� � s
��zStats.print_call_heading�->c C s t t|��|�| d| jd� |st | jd� d S t|�� �}d}|D ]^}t|�}|| } t| t�rb| \}
} } }
|
| krCd|
| f }nd|
f }d|�dd t |� �t
| �t
|
�|f }|d
}nd || t
| j | d �f }|d }t || | | jd� d}q#d S )
Nr� r� r< r; z%d/%dz%dz %s %s %s %srr rs r z %s(%r) %srk ) rK r[ r� r0 �sortedr� rN r �rjustr3 r� rF )r6 r� �sourceZ call_dictZarrowZclistr� r\ r r r^ r] r_ r` ZsubstatsZ
left_widthr! r! r" r� � s.
�
�zStats.print_call_linec C s" t dd| jd� t d| jd� d S )Nz- ncalls tottime percall cumtime percallr� r� zfilename:lineno(function)r< �rK r0 r� r! r! r" r� � s zStats.print_titlec C s� | j | \}}}}}t|�}||kr|d t|� }t|�d�d| jd� tt|�d| jd� |dkr<tdd| jd� n tt|| �d| jd� tt|�d| jd� |dkr_tdd| jd� n tt|| �d| jd� tt|�| jd� d S )Nr� � r� r� r r� r< )rF r rK r� r0 r� r[ )r6 r\ r] r^ r_ r` ra �cr! r! r" r� � s zStats.print_line)r� )r# r$ r% r9 r4 rH rI r5 rj rt rx r� r� r� r� r� r� r� r� r� r� r� r� r� r� r! r! r! r" r I sJ !
�' #
r c @ s e Zd Z dd� Zdd� ZdS )r} c C s
|| _ d S r� ��comp_select_list)r6 r� r! r! r" r9 s
zTupleComp.__init__c C sF | j D ] \}}|| }|| }||k r| S ||kr | S qdS rb r� )r6 �left�right�indexZ direction�l�rr! r! r" r~ s
�zTupleComp.compareN)r# r$ r% r9 r~ r! r! r! r" r} s r} c C s | \}}}t j�|�||fS r� )rR �path�basename)r� r r r r! r! r" r� " s
r� c C s | d S )Nrs r! )r\ r! r! r" r� & � r� c C sH | d d� dkr | d }|� d�r |�d�r d|dd� S |S d| S ) Nrs )�~r �<�>z{%s}r r: z %s:%d(%s))�
startswith�endswith)r� r r! r! r" r[ ) s r[ c C sB |\}}}}}| \}}} }
} || || || ||
t | |�fS r� )� add_callers) �targetr� r] r^ r_ r` ra Zt_ccZt_ncZt_ttZt_ctZ t_callersr! r! r" re : s �re c C s� i }| � � D ]\}}|||< q|� � D ]+\}}||v r;t|t�r2tdd� t||| �D ��||< q|| |7 < q|||< q|S )Nc s s � | ] \}}|| V qd S r� r! )�.0�i�jr! r! r" � <genexpr>J s � z add_callers.<locals>.<genexpr>)rZ rN r �zip)r� r� Z new_callersr\ r� r! r! r" r� A s
"
r� c C s d}| � � D ]}||7 }q|S rb )r )ra r^ r
r! r! r" � count_callsR s
r� c C s d| S )Nz%8.3fr! )�xr! r! r" r� ] r� r� �__main__c @ s� e Zd Zd6dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z dS )7�ProfileBrowserNc C s: t j�| � d| _d | _tj| _|d ur| �|� d S d S )N�% ) �cmd�Cmdr9 �promptrF r1 r2 r0 �do_read)r6 �profiler! r! r" r9 l s �zProfileBrowser.__init__c C s� |� � }g }|D ]@}z |�t|�� W q ty Y nw z t|�}|dks*|dk r3td| jd� W q|�|� W q tyB Y nw |�|� q| jrVt| j|�|� dS td| jd� dS )Nr r z#Fraction argument must be in [0, 1]r< � No statistics object is loaded.) �splitr{ r- �
ValueErrorr, rK r0 rF �getattr)r6 �fnr r7 Z processedZtermZfracr! r! r" �generict s2 �
� �zProfileBrowser.genericc C sX t d| jd� t d| jd� t d| jd� t d| jd� t d| jd� t d| jd� d S )NzArguments may be:r< z0* An integer maximum number of entries to print.z:* A decimal fractional number between 0 and 1, controllingz- what fraction of selected entries to print.z8* A regular expression; only entries with function namesz that match it are printed.r� r� r! r! r" � generic_help� s zProfileBrowser.generic_helpc
C sh | j r+z | j �|� W dS ty* } ztd||f | jd� W Y d }~dS d }~ww td| jd� dS )Nz$Failed to load statistics for %s: %sr< r� r )rF r5 �OSErrorrK r0 )r6 r �er! r! r" �do_add� s � ��zProfileBrowser.do_addc C � t d| jd� d S )Nz>Add profile info from given file to current statistics object.r< r� r� r! r! r" �help_add� � zProfileBrowser.help_addc C � | � d|�S )Nr� �r� �r6 r r! r! r" �
do_callees� � zProfileBrowser.do_calleesc C � t d| jd� | �� d S )Nz6Print callees statistics from the current stat object.r< �rK r0 r� r� r! r! r" � help_callees� � zProfileBrowser.help_calleesc C r� )Nr� r� r� r! r! r" �
do_callers� r� zProfileBrowser.do_callersc C r� )Nz6Print callers statistics from the current stat object.r< r� r� r! r! r" � help_callers� r� zProfileBrowser.help_callersc C s t d| jd� dS )Nr; r< r r� r� r! r! r" �do_EOF� s zProfileBrowser.do_EOFc C r� �NzLeave the profile browser.r< r� r� r! r! r" �help_EOF� r� zProfileBrowser.help_EOFc C s dS )Nr r! r� r! r! r" �do_quit� s zProfileBrowser.do_quitc C r� r� r� r� r! r! r" � help_quit� r� zProfileBrowser.help_quitc
C s� |rLzt |�| _W n; ty& } zt|jd | jd� W Y d }~d S d }~w tyD } zt|jjd || jd� W Y d }~d S d }~ww |d | _ dS t
| j �dkra| j d d� }| � |� dS td| jd� dS ) Nr r< �:r� rs ���z1No statistics object is current -- cannot reload.r ) r rF r� rK r7 r0 rJ rX r# r� r3 r� )r6 r �errr! r! r" r� � s( ���
�
�zProfileBrowser.do_readc C � t d| jd� t d| jd� d S )Nz+Read in profile data from a specified file.r< z*Without argument, reload the current file.r� r� r! r! r" � help_read� � zProfileBrowser.help_readc C s& | j r
| j �� dS td| jd� dS )Nr� r< r )rF r� rK r0 r� r! r! r" �
do_reverse� s
�zProfileBrowser.do_reversec C r� )Nz/Reverse the sort order of the profiling report.r< r� r� r! r! r" � help_reverse� r� zProfileBrowser.help_reversec s� | j s td| jd� d S | j �� � |r*t� fdd�|�� D ��r*| j j|�� � dS td| jd� tj� � D ]\}}td||d f | jd� q6dS ) Nr� r< c 3 s � | ]}|� v V qd S r� r! )r� r� �Zabbrevsr! r" r� � s � z)ProfileBrowser.do_sort.<locals>.<genexpr>z/Valid sort keys (unique prefixes are accepted):z%s -- %sr r )
rF rK r0 rx �allr� r� r rt rZ )r6 r ry r r! r� r" �do_sort� s
� zProfileBrowser.do_sortc C r� )Nz.Sort profile data according to specified keys.r< z3(Typing `sort' without arguments lists valid keys.)r� r� r! r! r" � help_sort� r� zProfileBrowser.help_sortc s � fdd�t jD �S )Nc s g | ] }|� � �r|�qS r! )r� )r� �a��textr! r" �
<listcomp>� s z0ProfileBrowser.complete_sort.<locals>.<listcomp>)r rt )r6 r� r7 r! r� r" �
complete_sort� s z ProfileBrowser.complete_sortc C r� )Nr� r� r� r! r! r" �do_stats� r� zProfileBrowser.do_statsc C r� )Nz.Print statistics from the current stat object.r< r� r� r! r! r" �
help_stats� r� zProfileBrowser.help_statsc C s& | j r
| j �� d S td| jd� d S )Nr� r< )rF r� rK r0 r� r! r! r" �do_strip� s zProfileBrowser.do_stripc C r� )Nz<Strip leading path information from filenames in the report.r< r� r� r! r! r" �
help_strip� r� zProfileBrowser.help_stripc C r� )Nz Show help for a given command.r< r� r� r! r! r" � help_help� r� zProfileBrowser.help_helpc C s |r|S d S r� r! )r6 �stopr r! r! r" �postcmd� s zProfileBrowser.postcmdr� ) r# r$ r% r9 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r! r! r! r" r� k s6
r� r rs z*Welcome to the profile statistics browser.r< zGoodbye.)+r1 rR r rP r� �enumr � functoolsr Z dataclassesr �typingr �__all__r r r r r r} r� r� r[ re r� r� r� r# r� �readline� ImportErrorr� r� r3 �argvZ initprofileZbrowserr� r� rK r0 Zcmdloop�KeyboardInterruptr! r! r! r" �<module> sh C � � ��