U

e5d�j�@s�
dZd
dl
Z
d
dlZd
dlZd
dlZd
dlZd
dlmZ
d
dlm	Z	
ddgZ
Gdd�dee�ZGdd�d�Z
Gd	d
�d
�Zdd�Zd
d�Zdd�Zdd�Zdd�Zdd�Zdd�Zedk�
r�d
dlZzd
dlZWnek
r�


YnXGdd�dej�Zee
j�
dk�
re
jdZndZzPee�
Ze
jdd�D]Z e�!e �

�
q0e"dej#d�
e�$�
e"d ej#d�
Wne%k
�
r~


YnXdS)!z3Class for printing reports on profiled python code.�N)
�Enum)
�
cmp_to_key�Stats�SortKeyc@s8eZ
dZd
ZdZdZdZdZdZdZ	dZ
d	Zd
d�ZdS)
r)�calls�ncalls)�
cumulative�cumtime)�filename�module�line�name�nfl�pcalls�stdname)�time�tottimec
Gs@|
d
}t�
||�}||_|
dd�D]}||j|<q&|
|_|S�Nr
�
)�str�__new__�_value_�_value2member_map_Z_all_values)�cls�values�value�objZother_value�r�/usr/lib64/python3.8/pstats.pyr-s






zSortKey.__new__N)
�__name__�
__module__�__qualname__ZCALLSZ
CUMULATIVEZFILENAMEZLINE�NAMEZNFLZPCALLSZSTDNAMEZTIMErrrrrr"s








c@s�eZ
dZd
Zdd�
dd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dddddddddddddd�
Z
dd�Zdd�Zdd�Z
d d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd7d1d2�
Zd3d4�Zd5d6�ZdS)8ra�This class is used for creating reports from data generated by the
    Profile class.  It is a "friend" of that class, and imports data either
    by direct access to members of Profile class, or by reading in a dictionary
    that was emitted (via marshal) from the Profile class.

    The big change from the previous Profiler (in terms of raw functionality)
    is that an "add()" method has been provided to combine Stats from
    several distinct profile runs.  Both the constructor and the add()
    method now take arbitrarily many file names as arguments.

    All the print methods now take an argument that indicates how many lines
    to print.  If the arg is a floating point number between 0 and 1.0, then
    it is taken as a decimal percentage of the available lines to be printed
    (e.g., .1 means print 10% of all available lines).  If it is an integer,
    it is taken to mean the number of lines of data that you wish to have
    printed.

    The sort_stats() method now processes some additional options (i.e., in
    addition to the old -1, 0, 1, or 2 that are respectively interpreted as
    'stdname', 'calls', 'time', and 'cumulative').  It takes either an
    arbitrary number of quoted strings or SortKey enum to select the sort
    order.

    For example sort_stats('time', 'name') or sort_stats(SortKey.TIME,
    SortKey.NAME) sorts on the major key of 'internal function time', and on
    the minor key of 'the name of the function'.  Look at the two tables in
    sort_stats() and get_sort_arg_defs(self) for more examples.

    All methods return self, so you can string together commands like:
        Stats('foo', 'goo').strip_dirs().sort_stats('calls').                            print_stats(5).print_callers(5)
    N)
�streamc

GsF|
ptj
|_t|�
sd}n|d
}|dd�}|�|�

|j|�
dSr)�sys�stdoutr#�len�init�add)�selfr#�args�argrrr�__init__Ys






zStats.__init__cCs�d|_g|_
d|_d
|_d
|_d
|_d
|_t�|_i|_	i|_
|�|
�

z|��
Wn8t
k
r�


td|j
rx|j
dnd|jd�
�YnXdS)Nr
zInvalid timing data %s�������
�file)�all_callees�files�fcn_list�total_tt�total_calls�
prim_calls�max_name_len�set�	top_level�stats�
sort_arg_dict�
load_stats�get_top_level_stats�	Exception�printr#)r)r+rrrr'cs(
















�
�z
Stats.initc	Cs�|
dkri|_dSt
|
t�rxt|
d
��}t�|�
|_W5QRXz"t�|
�
}t�	|j
�
d|
}
Wn


YnX|
g
|_n t|
d�r�|
�
�
|
j|_i|
_|js�td|j|
f�
�
dS)N�rbz    �create_statsz.Cannot create or construct a %r object from %r)r:�
isinstancer�open�marshal�load�os�statr�ctime�st_mtimer2�hasattrrA�	TypeError�	__class__)r)r+�
fZ
file_statsrrrr<vs*






















�zStats.load_statsc
Cs�|j�
�D]p\}
\}}}}}|j|7_|j|7_|j|7_d
|krZ|j�|
�

tt|
�
�
|j	kr
tt|
�
�
|_	q
dS)N)Zjprofiler
Zprofiler)
r:�itemsr5r6r4r9r(r&�func_std_stringr7)r)�func�cc�nc�tt�ct�callersrrrr=�s







zStats.get_top_level_statsc
Gs�|
s|St|
�
D]�}t
|�
t
|�
kr,t|�
}|j|j7_|j|j7_|j|j7_|j|j7_|jD]}|j�|�

qr|j	|j	kr�|j	|_	d|_
|j��D]<\}}||jkr�|j|}nd
d
d
d
if}t
||�|j|<q�q|S�Nr
)�reversed�typerr2r5r6r4r9r(r7r3r:rN�add_func_stats)r)�arg_list�itemrPrGZ
old_func_statrrrr(�s(

















z	Stats.addc	Cs(t|
d
��}t
�|j|�
W5QRXdS)z:Write the profile data to a file we know how to load back.�wbN)rCrD�dumpr:)r)r
rMrrr�
dump_stats�s
zStats.dump_stats))
)rr-z
call count))
)�r-zcumulative time))
��rz	file name))
��rzline number))
��rz
function name))rdr`rbzname/file/line))
)r
r-zprimitive call count))
)�rz
standard name))
)�r-z
internal time)
rrr	rr
rrr
rrrrrc
Cst|jsni|_}
i}|j
��D]>\}}|}|r|s4q||
krFd
||<q||
|<|dd�}q*q|D]
}|
|=qb|jS)z)Expand all abbreviations that are unique.r
Nr-)r;�sort_arg_dict_defaultrN)r)�dictZbad_list�word�tupZfragmentrrr�get_sort_arg_defs�s 















zStats.get_sort_arg_defsc
Gs\
|
sd
|_|St
|
�
dkrBt|
d
t�rBddddd�|
d
g
}
n:t
|
�
dkr||
dd�D] }t|�
t|
d
�
krZtd	�
�
qZ|��}d
}d|_d}|
D]B}t|t�r�|j	}|||d
}|j|||d7_d}q�g}|j
��D]4\}\}	}
}}}
|�|	|
||f|t
|�
|f�

q�|jtt|�
j�
d
�

g|_}|D]}|�|d�

�
qB|S)Nr
rrrrr)r-r
rrgrgzCan't have mixed argument typerr.z, )
�keyr-)r3r&rB�intrXrKrl�	sort_typerrr:rN�appendrO�sortr�	TupleComp�compare)r)Zfieldr+Z
sort_arg_defsZ
sort_tupleZ	connectorrjZ
stats_listrPrQrRrSrTrUr3�tuplerrr�
sort_stats�sF






��
















�



zStats.sort_statsc

Cs|jr|j�
�
|S�
N)r3�reverse�
r)rrr�
reverse_order

s



zStats.reverse_orderc
Cs�|j}
i|_}d
}|
�
�D]�\}\}}}}}	t|�
}
tt|
�
�
|krRtt|
�
�
}i}|	�
�D]\}}
|
|t|�
<q^|
|kr�t||
|||||f�||
<q|||||f||
<q|j}t�|_}|D]}|�t|�
�

q�||_	d|_
d|_|SrV)r:rN�func_strip_pathr&rOrYr9r8r(r7r3r1)r)ZoldstatsZnewstatsr7rPrQrRrSrTrUZnewfuncZ
newcallers�func2�callerZold_topZnew_toprrr�
strip_dirs
s0













�






zStats.strip_dirsc

Cst|jr
dSi|_}
|j
��D]P\}\}}}}}||
kr@i|
|<|��D]$\}}	||
kr`i|
|<|	|
||<qHqdSrv)r1r:rN)
r)r1rPrQrRrSrTrUr{r|rrr�calc_callees#
s











zStats.calc_calleescCs
|}t|
t
�rpzt�|
�
}Wn*tjk
rF


|d
|
7}||fYSXg}|D]}|�t|�
�
rP|�|�

qPnzt|�
}t|
t	�r�d|
k
r�dkr�n
nt
||
d�
}|d|�}n2t|
t
�r�d|
k
r�|kr�n
n|
}|d|�}t|�
t|�
k�
r|dt|�
t|�
|
f7}||fS)Nz#   <Invalid regular expression %r>
gg�?g�?r
z6   List reduced from %r to %r due to restriction <%r>
)rBr�re�compile�error�searchrOrpr&�floatrn)r)Zsel�list�msgZnew_listZrexrP�countrrr�eval_print_amount6
s2












"


"




�zStats.eval_print_amountcCs�|j}|j
r*|j
dd�}d
|jd}nt|j���
}d}|
D]}|�|||�\}}q@t|�
}|sld|fSt||j	d�
|t|j�
kr�d}|D] }tt
|�
�
|kr�tt
|�
�
}q�|d|fS)Nz   Ordered by: �

z!   Random listing order was used
r
r/rg)r7r3ror�r:�keysr�r&r?r#rO)r)Zsel_list�widthZ	stat_listr�Z	selectionr�rPrrr�get_print_listP
s$













zStats.get_print_listc
Gs�|jD]}t
||jd
�
q|jr,t
|jd
�

d}|jD]}t
|t|�
|jd
�
q6t
||jdd|jd�
|j|jkr�t
d|jd|jd�
t
d|j|jd
�
t
|jd
�

|�|
�
\}}|r�|�	�
|D]}|�
|�

q�t
|jd
�

t
|jd
�

|S)Nr/�        zfunction calls�
 ��endr0z(%d primitive calls)zin %.3f seconds)r2r?r#r9�func_get_function_namer5r6r4r��print_title�
print_line)r)�amountr
�indentrPr�r�rrr�print_statsh
s(




















zStats.print_statsc
Gsz|�|
�
\}}|rv|�
�
|�|d
�
|D]2}||jkrN|�|||j|�
q*|�||i�
q*t|jd�

t|jd�

|S)Nz	called...r/)r�r~�print_call_headingr1�print_call_liner?r#)r)r�r�r�rPrrr�
print_callees
s









zStats.print_calleesc

Gsh|�|
�
\}}|rd|�
|d
�
|D](}|j|\}}}}}	|�|||	d�
q"t|jd�

t|jd�

|S)Nzwas called by...z<-r/)r�r�r:r�r?r#)
r)r�r�r�rPrQrRrSrTrUrrr�
print_callers�
s








zStats.print_callersc
Csvtd
�
|
�
||jd�
d}|j��D]0\}}}}}|r&tt|���
�
}	t|	t�}
qXq&|rrtd|
d|jd�
dS)Nz	Function r/Fr�z    ncalls  tottime  cumtime)	r?�ljustr#r:r�next�iterrBrt)
r)�	name_sizeZcolumn_titleZ	subheaderrQrRrSrTrUrrrrr��
s








zStats.print_call_heading�->cCs
tt
|�
�|
�
|d
|jd�
|s2t|jd�

dSt|���
}d}|D]�}t
|�
}||}	t|	t�r�|	\}
}}}
|
|kr�d|
|f}n
d|
f
}d|�dd	t	|�
�
t
|�
t
|
�
|f}|
d
}n$d||	t
|j|d�
f}|
d}t||||jd�
d
}qFdS)
Nr�r�r/r.z%d/%dz%dz%s %s %s  %srfrgrz	%s(%r) %sr_)r?rOr�r#�sortedr�rBrt�rjustr&�f8r:)r)r��sourceZ	call_dictZarrowZclistr�rPr
rrRrQrSrTZsubstatsZ
left_widthrrrr��
s0
















�



zStats.print_call_linec

Cs"td
d|j
d�
td|j
d�
dS)Nz-   ncalls  tottime  percall  cumtime  percallr�r�zfilename:lineno(function)r/�r?r#rxrrrr��
s

zStats.print_titlecCs�|j|
\}}}}}t
|�
}||kr4|d
t
|�
}t|�d�
d|jd�
tt|�
d|jd�
|dkrxtdd|jd�
ntt||�
d|jd�
tt|�
d|jd�
|dkr�tdd|jd�
ntt||�
d|jd�
tt|
�
|jd�
dS)N�
/�	r�r�r
r�r/)r:rr?r�r#r�rO)r)rPrQrRrSrTrU�
crrrr��
s











zStats.print_line)
r�)rr r!�__doc__r,r'r<r=r(r^rhrlruryr}r~r�r�r�r�r�r�r�r�r�rrrrr7sF
!













�'
c@s eZ
dZd
Zdd�Zdd�ZdS)rra�
This class provides a generic function for comparing any two tuples.
    Each instance records a list of tuple-indices (from most significant
    to least significant), and sort direction (ascending or decending) for
    each tuple-index.  The compare functions can then be used as the function
    argument to the system sort() function when a list of tuples need to be
    sorted in the instances order.cCs
|
|_dSrv�
�comp_select_list)r)r�rrrr,�
s
zTupleComp.__init__cCsF|jD]:\}}|
|}||}||kr0|
S||kr|
Sqd
SrVr�)r)�left�right�index�	direction�
l�
rrrrrs�
s









zTupleComp.compareN)rr r!r�r,rsrrrrrr�
s
rrc
Cs|\}
}}tj
�|
�
||fSrv)rF�path�basename)�	func_namer
rr
rrrrz�
s


rzc

Cs|d
S)Nrgr)
rPrrrr��
s
r�c
CsN|dd
�dkrB|d
}
|
�d�
r<|
�
d�
r<d|
dd�S|
Snd|SdS)	Nrg)�
~r
�
<�
>z{%s}rr-z	%s:%d(%s))�
startswith�endswith)r�r
rrrrO�
s


rOcCs@|
\}}}}}|\}}}	}
}||||||	||
t||�fS)
z3Add together all the stats for two profile entries.)
�add_callers)�targetr�rQrRrSrTrUZt_ccZt_ncZt_ttZt_ctZ	t_callersrrrrYs



�rYcCs�i}|��D]\}}|||<q|
��D]V\}}||krtt
|t�rbtd
d�t|||�D�
�
||<q||||7<q&|||<q&|S)z*Combine two caller lists in a single list.c
ss|]\}
}|
|V
qdSrvr)�.0�
i�
jrrr�	<genexpr>szadd_callers.<locals>.<genexpr>)rNrBrt�zip)r�r�Znew_callersrPr|rrrr�s






"

r�c
Csd
}
|��D]}|
|7}
q|
S)z@Sum the caller statistics to get total number of calls received.r
)
r)rUrRrrrr�count_callss



r�c

Csd
|S)Nz%8.3fr)
�
xrrrr�(s
r��__main__c@s�eZ
dZd6dd�
Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd
S)7�ProfileBrowserNcCs6tj
�|�

d
|_d|_tj|_|
dk	r2|�|
�

dS)N�% )	�cmd�Cmdr,�promptr:r$r%r#�do_read)r)�profilerrrr,7s





zProfileBrowser.__init__c	Cs�|��}g}|D]�}z|�
t|�
�

WqWntk
r>


YnXz<t|�
}|d
ksZ|dkrltd|jd�
Wq|�
|�

WqWntk
r�


YnX|�
|�

q|jr�t|j|
�|�
ntd|jd�
dS)Nrr
z#Fraction argument must be in [0, 1]r/�No statistics object is loaded.)	�splitrprn�
ValueErrorr�r?r#r:�getattr)r)�fnrr*Z	processedZtermZfracrrr�generic?s,





















zProfileBrowser.genericc

CsXtd
|j
d�
td|j
d�
td|j
d�
td|j
d�
td|j
d�
td|j
d�
dS)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�rxrrr�generic_helpWs





zProfileBrowser.generic_helpc
Csd|jrRz|j�
|
�

Wq`tk
rN
}
ztd
|
|f|jd�
W5d}~XYq`Xntd|jd�
dS)Nz$Failed to load statistics for %s: %sr/r�r
)r:r(�OSErrorr?r#)r)r�
errr�do_add_s




*
zProfileBrowser.do_addc

Cstd
|j
d�
dS)Nz>Add profile info from given file to current statistics object.r/r�rxrrr�help_addhs
zProfileBrowser.help_addcCs|�d
|
�S)Nr��
r��r)rrrr�
do_calleesks
zProfileBrowser.do_calleesc

Cstd
|j
d�
|��
dS)Nz6Print callees statistics from the current stat object.r/�r?r#r�rxrrr�help_calleesms

zProfileBrowser.help_calleescCs|�d
|
�S)Nr�r�r�rrr�
do_callersqs
zProfileBrowser.do_callersc

Cstd
|j
d�
|��
dS)Nz6Print callers statistics from the current stat object.r/r�rxrrr�help_callersss

zProfileBrowser.help_callerscCstd
|j
d�
dS)Nr.r/rr�r�rrr�do_EOFws

zProfileBrowser.do_EOFc

Cstd
|j
d�
dS�NzLeave the profile browser.r/r�rxrrr�help_EOFzs
zProfileBrowser.help_EOFc
Csd
S)Nrrr�rrr�do_quit}s
zProfileBrowser.do_quitc

Cstd
|j
d�
dSr�r�rxrrr�	help_quits
zProfileBrowser.help_quitc
Cs�|
r�zt|
�
|_
Wnztk
rN
}
zt|jd
|jd�
WY�dSd}~XYn@tk
r�
}
z"t|jjd||jd�
WY�dSd}~XYnX|
d|_	n6t
|j	�
dkr�|j	dd�}
|�|
�

ntd|jd�
dS)	Nrr/�
:r�rg�����z1No statistics object is current -- cannot reload.r
)rr:r�r?r*r#r>rLrr�r&r�)r)r�errrrrr��s













zProfileBrowser.do_readc

Cs td
|j
d�
td|j
d�
dS)Nz+Read in profile data from a specified file.r/z*Without argument, reload the current file.r�rxrrr�	help_read�s

zProfileBrowser.help_readcCs$|jr|j�
�
ntd
|jd�
dS)Nr�r/r
)r:ryr?r#r�rrr�
do_reverse�s


zProfileBrowser.do_reversec

Cstd
|j
d�
dS)Nz/Reverse the sort order of the profiling report.r/r�rxrrr�help_reverse�s
zProfileBrowser.help_reversecs�|jst
d
|jd�
dS|j���|
rRt�f
dd�|
��D�
�
rR|jj|
���
n<t
d|jd�
tj�	�D]"\}}t
d||df|jd�
qjdS)	Nr�r/c
3s|]}
|
�kV
qdSrvr)r�r��
Zabbrevsrrr��sz)ProfileBrowser.do_sort.<locals>.<genexpr>z/Valid sort keys (unique prefixes are accepted):z%s -- %srr
)
r:r?r#rl�allr�rurrhrN)r)rrmrrr�r�do_sort�s









zProfileBrowser.do_sortc

Cs td
|j
d�
td|j
d�
dS)Nz.Sort profile data according to specified keys.r/z3(Typing `sort' without arguments lists valid keys.)r�rxrrr�	help_sort�s

zProfileBrowser.help_sortcs�f
d
d�tj
D�
S)Nc
sg|]}
|
���
r|
�qSr)
r�)r��
a�
�textrr�
<listcomp>�s
z0ProfileBrowser.complete_sort.<locals>.<listcomp>)rrh)r)r�r*rr�r�
complete_sort�s
zProfileBrowser.complete_sortcCs|�d
|
�S)Nr�r�r�rrr�do_stats�s
zProfileBrowser.do_statsc

Cstd
|j
d�
|��
dS)Nz.Print statistics from the current stat object.r/r�rxrrr�
help_stats�s

zProfileBrowser.help_statscCs$|jr|j�
�
ntd
|jd�
dS)Nr�r/)r:r}r?r#r�rrr�do_strip�s

zProfileBrowser.do_stripc

Cstd
|j
d�
dS)Nz<Strip leading path information from filenames in the report.r/r�rxrrr�
help_strip�s
zProfileBrowser.help_stripc

Cstd
|j
d�
dS)NzShow help for a given command.r/r�rxrrr�	help_help�s
zProfileBrowser.help_helpc
Cs|
r|
SdSrvr)r)�stoprrrr�postcmd�s


zProfileBrowser.postcmd)
N)rr 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�rrrrr�6s4

	r�rrgz*Welcome to the profile statistics browser.r/zGoodbye.)&r�r$rFrrDr�enumr�	functoolsr�__all__rrrrrrzr�rOrYr�r�r�rr��readline�ImportErrorr�r�r&�argvZinitprofileZbrowserr�r�r?r#Zcmdloop�KeyboardInterruptrrrr�<module>
sR