֦iir d Z ddlZddlZddlZddlZddlZddlmZmZ ddl m
Z
ddl m Z ddl
mZ g dZ ee G d d Z e d
G d d
Z e d
G d d Z G d d Z G d d Zd Zd Zd Zd Zd Zd Zd Ze dk( rddl Z ddl Z G d d e j@ Z! e"ejF d kD rejF d Z$ndZ$ e!e$ Z%ejF d d D ] Z&e%jO e& e(d e%jR ! e%jU e(d"e%jR ! yy# e $ r Y w xY w# e+$ r Y yw xY w)#z3Class for printing reports on profiled python code. N)StrEnum _simple_enum)
cmp_to_key) dataclass)Dict)StatsSortKeyFunctionProfile StatsProfilec 6 e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z y ) r )callsncalls)
cumulativecumtime)filenamemodulelinenamenflpcallsstdname)timetottimec |d }t j | | }||_ |dd D ] }|| j |< ||_ |S Nr )str__new___value__value2member_map_ _all_values)clsvaluesvalueobj other_values /usr/lib64/python3.12/pstats.pyr zSortKey.__new__0 sN q kk#u% !!":K25C " "; / &
N)
__name__
__module__ __qualname__CALLS
CUMULATIVEFILENAMELINENAMENFLPCALLSSTDNAMETIMEr r( r' r r $ s4 E(J#H D D
C
FG Dr( r T) unsafe_hashc T e Zd ZU eed< eed< eed< eed< eed< eed< eed< y) r
r r percall_tottimer percall_cumtime file_name line_numberN)r) r* r+ r __annotations__floatintr5 r( r' r
r
: s( K
N
NNr( r
c 0 e Zd ZU dZeed< eeef ed< y)r z0Class for keeping track of an item in inventory.total_tt
func_profilesN) r) r* r+ __doc__r= r< r r r
r5 r( r' r r D s :O_,--r( r c e Zd ZdZdddZd Zd Zd Zd Zd Z d
d
d d d d
d ddddddd
Z
d Z d Z d Z
d Zd Zd Zd Zd Zd Zd Zd Zd Zd#d Zd! Zd" Zy)$r a 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 |xs t j | _ t | sd }n
|d }|dd }| j | | j
| y r )sysstdoutrD leninitadd)selfrD argsargs r' __init__zStats.__init__l sJ *
4yCq'C8D #$r( c j d | _ g | _ d | _ d| _ d| _ d| _ d| _ t | _ i | _ i | _
| j | | j y # t $ r7 t d| j r| j d ndz | j w xY w)Nr zInvalid timing data %s file) all_calleesfilesfcn_listr@ total_calls
prim_calls max_name_lenset top_levelstats
sort_arg_dict
load_statsget_top_level_stats ExceptionprintrD )rK rM s r' rI z
Stats.initv s
$ $ & *%)ZZ4::b>R9?C{{
L s
!A2 2A B2c |i | _ y t |t rst |d 5 }t j
| | _ d d d t
j | }t j |j dz |z }|g| _ n4t |d r(|j |j | _ i |_ | j st d| j d| y # 1 sw Y xY w# Y uxY w)Nrbz create_statsz Cannot create or construct a z
object from )r\
isinstancer openmarshalloadosstatr ctimest_mtimerU hasattrrd TypeError __class__)rK rM f
file_statss r' r^ zStats.load_stats s ;DJ
S
!c4 A $\\!_
!
WWS\
jj!4!45>D DJ
S.
) DJCIzz #~~s 4 5
5 !
s C* :C6 *C36C:c | j j D ] \ }\ }}}}}| xj |z
c_ | xj |z
c_ | xj |z
c_ d|v r| j
j
| t t | | j kD st t | | _ y )N)jprofiler profiler)
r\ itemsrW rX r@ r[ rJ rH func_std_stringrY )rK funcccncttctcallerss r' r_ zStats.get_top_level_stats s /3zz/?/?/A +D+2r2r7 " OO "O MM "M*g5""4(?4()D,=,==$'(=$>! 0Br( c |s| S t | D ]^ }t | t | k7 r t | }| xj |j z
c_ | xj |j z
c_ | xj
|j
z
c_ | xj |j z
c_ |j D ] }| j j | | j |j k r|j | _ d | _
|j j D ]C \ }}|| j v r| j | }nddddi f}t || | j |< E a | S Nr )reversedtyper rU rW rX r@ r[ rJ rY rV r\ ru add_func_stats)rK arg_listitemrw rj
old_func_stats r' rJ z Stats.add s+ K X&DDzT$Z' T{ JJ$** $J 0 0 0 OOt .O MMT]] *M ""4( ' 4#4#44$($5$5! DM "jj . . 0
d4::%$(JJt$4M%&1a$5M#1-#F
4 1 '* r( c t |d 5 }t j | j | ddd y# 1 sw Y yxY w)z:Write the profile data to a file we know how to load back.wbN)rf rg dumpr\ )rK r rp s r'
dump_statszStats.dump_stats s,
(D
!Q LLQ ' "
!
!s !7A )))r rP z
call count))) rP zcumulative time)) r z file name)) r z line number)) r z
function name))r r r zname/file/line)))r rP zprimitive call count))) r z
standard name))) rP z
internal time)
r
r r r r r r r r r r r r c | j sSi x| _ }i }| j j D ]! \ }}|}|s ||v rd||< |||< |dd }|r# |D ] }||= | j S )z)Expand all abbreviations that are unique.r NrP )r] sort_arg_dict_defaultru )rK dictbad_listwordtupfragments r' get_sort_arg_defszStats.get_sort_arg_defs s !!(* *D H ! 7 7 = = ? c 4'-.* %(DN '}H @ !J !!!!r( c |s d| _ | S t | dk( r"t |d t rddddd|d g}n=t | dk\ r/|dd D ]' }t | t |d k7 s t d | j
}d
}d | _ d }|D ]I }t |t r |j }||| d z }| xj ||| d z z
c_ d }K g }| j j D ]1 \ }\ } }
} } }
|j | |
| | f|z t | |fz 3 |j t t! | j"
g x| _ }|D ] }|j |d | S )Nr r r r
r r )rP r r r r z Can't have mixed argument typer5 rQ z, )keyrP )rV rH re r> r rn r sort_typer r$ r\ ru appendrv sortr TupleCompcompare)rK fieldrM
sort_arg_defs
sort_tuple connectorr
stats_listrw rx ry rz r{ r| rV tuples r'
sort_statszStats.sort_stats s DMK u:?z%(C 8$"!') */q3 5E Z1_ QRy9U1X.#$DEE ! ..0
D$(zz#mD&9!&<
g g ?g ?r z List reduced from z to z due to restriction
) re r recompileerrorsearchrv r rH r= r> )rK sellistmsgnew_listrexrw counts r' eval_print_amountzStats.eval_print_amountG s c3
%jjo H ::od 34 OOD) IE#u%#*:s*:ECK" ,- <C%!s*:U*: < t9H
% D 3x=#/
/C } ' 88
%=CC }$
%s D D"!D"c
| j r| j dd n"t | j j }|s t di S t t
| j }i }t || }|D ] }| j | \ }}}} }
|\ } } }
||k( r t | nt | dz t | z }t t
| }|dk( rdnt t
||z }t t
| }|dk( rdnt t
| |z }t |||||| | }|||
< |S )a? This method returns an instance of StatsProfile, which contains a mapping
of function names to instances of FunctionProfile. Each FunctionProfile
instance holds information related to the function's profile such as how
long the function took to run, how many times it was called, etc...
Nr /rP )
rV r r\ keysr r= f8r@ r r
)rK func_listr@ rA
stats_profilerw rx ry rz r{ r| r: r; func_namer r r8 r r9 func_profiles r' get_stats_profilezStats.get_stats_profilea s0 )-
DMM!$4
@Q;R 2& &DMM*+
$X}=
D&*jj&6 #BBG04 -I{I "bSWs2w}s2w/FFBrFmG$&!Gbr"R%y1AOBrFmG$&!Gbr"R%y1AO* L (4M) $# & r( c | j }| j r"| j d d }d| j z dz }n%t | j j }d}|D ] }| j
||| \ }} t | }|sd|fS t || j |t | j k r5d}|D ]. }t t | |kD st t | }0 |dz |fS )Nz Ordered by:
z! Random listing order was used
r rR r ) rY rV r r r\ r r rH ra rD rv )rK sel_listwidth stat_listr selectionr rw s r' get_print_listzStats.get_print_list s !! ==
a(I#dnn4t;C TZZ__ ./I6C!I ! 3 3Iy# NNIs " Ii<
c $ 3tzz? "E!-.6 5 6E " Qw !!r( c | j D ] }t || j | j rt | j d}| j D ]# }t |t | | j % t || j
dd| j | j
| j k7 r%t d| j z d| j t d| j z | j t | j | j | \ }}|rT| j |D ] }| j | t | j t | j | S )NrR zfunction calls endrS z(%d primitive calls)zin %.3f seconds) rU ra rD r[ func_get_function_namerW rX r@ r print_title
print_line)rK amountr indentrw r r s r' print_statszStats.print_stats s
H ( - # :: t{{ #NND &06T[[ I # fd&&(8c T t . (4??:$++ V
$--/dkkB
4;; ))&1 t % t{{ # t{{ # r( c ` | j | \ }}|r| j | j |d |D ]D }|| j v r!| j ||| j | 2| j ||i F t | j t | j | S )Nz called...rR )r r print_call_headingrT print_call_linera rD )rK r r r rw s r'
print_calleeszStats.print_callees s ))&1 t # #E; 7 4+++((d6F6Ft6LM((b9
t{{ # t{{ # r( c | j | \ }}|rn| j |d |D ]+ }| j | \ }}}}} | j ||| d - t | j
t | j
| S )Nzwas called by...z<-rR )r r r\ r ra rD )
rK r r r rw rx ry rz r{ r| s
r'
print_callerszStats.print_callers s ))&1 t # #E+= > *.**T*:'BB$$UD'4@
t{{ # t{{ # r( c P t dj | |z | j d}| j j D ]> \ }}}}}|s t t
|j } t | t } n |r t d|z dz | j y y )Nz Function rR Fr z ncalls tottime cumtime) ra ljustrD r\ r# nextiterre r )
rK name_size column_title subheaderrx ry rz r{ r| r$ s
r' r zStats.print_call_heading s
k *\9 L '+zz'8'8': #BBG T'.."2 34 &ue 4 (;
#i-"@@t{{ S r( c h t t | j | |z d| j |st | j y t |j }d}|D ] }t | }|| } t
| t r]| \ }
} } }
|
| k7 rd|
| fz }nd|
fz }|j ddt | z z dt | dt |
d |}|d
z }n(|d | d t | j | d
}|d
z }t ||z |z | j d} y )Nr r rR rQ z%d/%dz%dr r z r (z) r ) ra rv r rD sortedr re r rjustrH r r\ )rK r source call_dictarrowclistr rw r r$ ry rx rz r{ substats
left_widths r' r zStats.print_call_line s
of%++I6>CdkkZ t{{ # y~~'(D"4(D dOE%'!&BB8 &"b 1H #re|H-5^^AaF mO-L-/VRVT C &]
*.r$**T:J1:M7NO &]
&#h.T[[ AF! r( c b t dd| j t d| j y )Nz- ncalls tottime percall cumtime percallr r zfilename:lineno(function)rR ra rD r s r' r zStats.print_title s"
=3T[[Y
) )r) r* r+ rB rN rI r^ r_ rJ r r r r r r r r r r r r r r r r r r5 r( r' r r J s B &* &,?4( D C H H B D B F G M F F F ""%N
: & 4! F"0.
T2=7r( r c e Zd ZdZd Zd Zy)r a 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 descending) 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.c || _ y r comp_select_list)rK r s r' rN zTupleComp.__init__ s
0 r( c f | j D ]" \ }}|| }|| }||k r| c S ||kD s |c S yr~ r )rK leftrightindex directionlrs r' r zTupleComp.compare sJ $ 5 5 E9U Ae A1u!z!1u
!6 r( N)r) r* r+ rB rN r r5 r( r' r r s &1r( r c R | \ }}}t j j | ||fS r )ri pathbasename)r r r r s r' r r ! s* $HdD
77 H %tT 11r( c | d S )Nr r5 )rw s r' r r % s 7Nr( c ~ | d d dk( r1| d }|j d r|j d rd|dd z S |S d| z S ) Nr )~r <>z{%s}r rP z %s:%d(%s))
startswithendswith)r r s r' rv rv ( sQ !} | ??3 DMM#$6D2J& &KY&&r( c \ |\ }}}}}| \ }}} }
} ||z ||z || z ||
z t | | fS )z3Add together all the stats for two profile entries.) add_callers) targetr rx ry rz r{ r| t_cct_nct_ttt_ct t_callerss r' r r 9 sO $BBG(.%D$dI tGRWbgr$w)W- / /r( c i }| j D ]
\ }}|||< |j D ]O \ }}||v rAt |t r#t d t ||| D ||< =||xx |z
cc< K|||< Q |S )z*Combine two caller lists in a single list.c 3 , K | ] \ }}||z y wr r5 ).0ijs r' z add_callers..I s )[ d}| j D ] }||z
} |S )z@Sum the caller statistics to get total number of calls received.r )r# )r| ry r
s r' count_callsr# Q s&
B !
e "
Ir( c d| z S )Nz%8.3fr5 )xs r' r r \ s Q;r( __main__c e Zd Zd dZd Zd Zd Zd Zd Zd Z d Z
d
Z d Z d Z
d
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z y) ProfileBrowserNc t j j | d| _ d | _ t
j | _ || j | y y )N% ) cmdCmdrN promptr\ rF rG rD do_read)rK profiles r' rN zProfileBrowser.__init__k sD GG T " DK DJ **DK" W% #r( c |j }g }|D ] } |j t | | j r t | j | | yt d| j y# t $ r Y nw xY w t | }|dkD s|dk rt d| j |j | # t $ r Y nw xY w|j | )Nr r z#Fraction argument must be in [0, 1]rR No statistics object is loaded.) splitr r>
ValueErrorr= ra rD r\ getattr)rK fnr rL processedtermfracs r' genericzProfileBrowser.generics s ::A?,C0C C
Cc t d| j t d| j t d| j t d| j t d| j t d| j y )NzArguments may be:rR 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 s r' generic_helpzProfileBrowser.generic_help sb %DKK 8 D4;; W NUYU`U` a A T LSWS^S^ _ 0t{{ Cr( c | j r | j j | yt d| j y# t $ r'}t d|d|| j Y d }~yd }~ww xY w)Nz Failed to load statistics for z: rR r1 r )r\ rJ OSErrorra rD )rK r es r' do_addzProfileBrowser.do_add sj zz`JJNN4(
7dkkJ `D!LSWS^S^__ `s A A2 A--A2c 2 t d| j y )Nz>Add profile info from given file to current statistics object.rR r r s r' help_addzProfileBrowser.help_add s RY]YdYd er( c & | j d| S )Nr r9 rK r s r'
do_calleeszProfileBrowser.do_callees <<6 6r( c R t d| j | j y )Nz6Print callees statistics from the current stat object.rR ra rD r; r s r' help_calleeszProfileBrowser.help_callees JQUQ\Q\ ] r( c & | j d| S )Nr rC rD s r'
do_callerszProfileBrowser.do_callers rF r( c R t d| j | j y )Nz6Print callers statistics from the current stat object.rR rH r s r' help_callerszProfileBrowser.help_callers rJ r( c 2 t d| j y)NrQ rR r r rD s r' do_EOFzProfileBrowser.do_EOF s "4;; 'r( c 2 t d| j y NzLeave the profile browser.rR r r s r' help_EOFzProfileBrowser.help_EOF .T[[ Ar( c y)Nr r5 rD s r' do_quitzProfileBrowser.do_quit s r( c 2 t d| j y rR r r s r' help_quitzProfileBrowser.help_quit rT r( c |r t | | _ |dz | _ yt | j dkD r!| j d d }| j | yt d| j
y# t $ r.}t |j d | j
Y d }~y d }~wt $ r9}t |j j dz || j
Y d }~y d }~ww xY w) Nr rR :r* r z1No statistics object is current -- cannot reload.r ) r r\ r= ra rL rD r` ro r) r- rH r. )rK r errs r' r. zProfileBrowser.do_read s !&tDJ #Tk T[[!A%{{3B' T" IPTP[P[\ #((1+DKK8 #--0036$++Ns# A0 0 C'9$B"" C'./C""C'c ` t d| j t d| j y )Nz+Read in profile data from a specified file.rR z*Without argument, reload the current file.r r s r' help_readzProfileBrowser.help_read s ?dkk R >T[[ Qr( c | j r| j j yt d| j y)Nr1 rR r )r\ r ra rD rD s r'
do_reversezProfileBrowser.do_reverse s2 zz
((* 7dkkJr( c 2 t d| j y )Nz/Reverse the sort order of the profiling report.rR r r s r' help_reversezProfileBrowser.help_reverse s C$++ Vr( c | j st d| j y | j j |rJt fd|j D r( | j j |j yt d| j t j j D ]$ \ }}t |d|d | j & y)Nr1 rR c 3 & K | ] }|v
y wr r5 )r r% abbrevss r' r z)ProfileBrowser.do_sort.. s ALqQ'\Ls z/Valid sort keys (unique prefixes are accepted):z -- r r )
r\ ra rD r allr2 r r r ru )rK r r r$ re s @r' do_sortzProfileBrowser.do_sort s ::7dkkJjj224GADJJLAA%
%%tzz|4
GdkkZ$)$?$?$E$E$GLS%U1X6T[[I %Hr( c ` t d| j t d| j y )Nz.Sort profile data according to specified keys.rR z3(Typing `sort' without arguments lists valid keys.)r r s r' help_sortzProfileBrowser.help_sort s B U Gdkk Zr( c l t j D cg c] }|j | s| c}S c c}w r )r r r )rK textrL as r'
complete_sortz ProfileBrowser.complete_sort s- $ : :Q :!all4>PA :Q QQs 11c & | j d| S )Nr rC rD s r' do_statszProfileBrowser.do_stats s <<
t4 4r( c R t d| j | j y )Nz.Print statistics from the current stat object.rR rH r s r'
help_statszProfileBrowser.help_stats s B U r( c | j r| j j y t d| j y )Nr1 rR )r\ r ra rD rD s r' do_stripzProfileBrowser.do_strip s( zz
%%'7dkkJr( c 2 t d| j y )Nzr s 9, & !
A
g * t t . . .
|7 |7|
22 '"/" z
P Pd 388}qhhqk
-xx|G NN7 # $
: P
jw~~.G
@
s% )E *A%E EE E$#E$