U
e5dj @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddgZ
G dd de eZ G dd dZ
G d d
d
Zd d Zd
d Zdd Zdd Zdd Zdd Zdd ZedkrddlZz ddlZW n ek
r Y nX G dd dejZe ej d krej d Z ndZ zPee Z ej d d D ]Z e !e q0e"d e j#d e $ e"d e j#d W n e%k
r~ Y nX dS )!z3Class for printing reports on profiled python code. N)Enum)
cmp_to_keyStatsSortKeyc @ s8 e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
d Z d S )
r )callsncalls)
cumulativecumtime)filenamemodulelinenamenflpcallsstdname)timetottimec G s@ |d }t | |}||_|dd D ]}|| j|< q&||_|S Nr )str__new___value__value2member_map_Z _all_values)clsvaluesvalueobjZ other_value r /usr/lib64/python3.8/pstats.pyr - s zSortKey.__new__N)
__name__
__module__ __qualname__ZCALLSZ
CUMULATIVEZFILENAMEZLINENAMEZNFLZPCALLSZSTDNAMEZTIMEr r r r r r " s c @ s e Zd ZdZddddZdd Zdd Zd
d Zd d
Zdd Z dddddddddddddd
Z
dd Z d d Z d d Z
d d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd7d1d2Zd3d4 Zd5d6 ZdS )8r 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 G sF |pt j| _t|sd }n|d }|dd }| | | j| d S r )sysstdoutr# leninitadd)selfr# argsargr r r __init__Y s
zStats.__init__c C s d | _ g | _d | _d| _d| _d| _d| _t | _i | _ i | _
| | z | W n8 t
k
r td| jrx| jd nd | jd Y nX d S )Nr zInvalid timing data %s file) all_calleesfilesfcn_listtotal_tt total_calls
prim_calls max_name_lenset top_levelstats
sort_arg_dict
load_statsget_top_level_stats Exceptionprintr# )r) r+ r r r r' c s(
z
Stats.initc C s |d kri | _ d S t|trxt|d}t|| _ W 5 Q R X z"t|}t |j
d | }W n Y nX |g| _ n t |dr|
|j | _ i |_ | j std| j|f d S )Nrbz create_statsz.Cannot create or construct a %r object from %r)r:
isinstancer openmarshalloadosstatr ctimest_mtimer2 hasattrrA TypeError __class__)r) r+ fZ
file_statsr r r r< v s*
zStats.load_statsc C s | j D ]p\}\}}}}}| j|7 _| j|7 _| j|7 _d|krZ| j| tt|| j kr
tt|| _ q
d S )N)Zjprofiler Zprofiler)
r: itemsr5 r6 r4 r9 r( r& func_std_stringr7 )r) funcccncttctcallersr r r r= s zStats.get_top_level_statsc G s |s| S t |D ]}t| t|kr,t|}| j|j7 _| j|j7 _| j|j7 _| j|j7 _|jD ]}| j| qr| j |j k r|j | _ d | _
|j D ]<\}}|| j kr| j | }nddddi f}t
||| j |< qq| S Nr )reversedtyper r2 r5 r6 r4 r9 r( r7 r3 r: rN add_func_stats)r) arg_listitemrP rG Z
old_func_statr r r r( s(
z Stats.addc C s( t |d}t| j| W 5 Q R X dS )z:Write the profile data to a file we know how to load back.wbN)rC rD dumpr: )r) r
rM 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))rd r` rb 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 st | j sni | _ }i }| j D ]>\}}|}|r |s4q ||krFd||< q |||< |dd }q*q |D ]
}||= qb| j S )z)Expand all abbreviations that are unique.r Nr- )r; sort_arg_dict_defaultrN )r) dictZbad_listwordtupZfragmentr r r get_sort_arg_defs s
zStats.get_sort_arg_defsc G s\ |sd| _ | S t|dkrBt|d trBddddd|d g}n:t|dkr||dd D ] }t|t|d krZtd qZ| }d
}d | _d }|D ]B}t|tr|j }||| d }| j||| d 7 _d }qg }| j
D ]4\}\} }
} } }
| | |
| | f| t
||f q|jtt|jd
g | _ }|D ]}| |d qB| S )Nr r r r r r )r- r r rg rg z Can't have mixed argument typer r. z, )keyr- )r3 r& rB intrX rK rl sort_typer r r: rN appendrO sortr TupleCompcompare)r) Zfieldr+ Z
sort_arg_defsZ
sort_tupleZ connectorrj Z
stats_listrP rQ rR rS rT rU r3 tupler r r
sort_stats sF
zStats.sort_statsc C s | j r| j | S N)r3 reverser) r r r
reverse_order s
zStats.reverse_orderc C s | j }i | _ }d}| D ]\}\}}}}} t|}
tt|
|krRtt|
}i } | D ]\} }
|
| t| < q^|
|krt||
||||| f||
< q ||||| f||
< q | j}t | _}|D ]}|t| q|| _ d | _
d | _ | S rV ) r: rN func_strip_pathr& rO rY r9 r8 r( r7 r3 r1 )r) ZoldstatsZnewstatsr7 rP rQ rR rS rT rU ZnewfuncZ
newcallersfunc2callerZold_topZnew_topr r r
strip_dirs s0
zStats.strip_dirsc
C st | j r
d S i | _ }| j D ]P\}\}}}}}||kr@i ||< | D ]$\}} ||kr`i ||< | || |< qHq d S rv )r1 r: rN )
r) r1 rP rQ rR rS rT rU r{ r| r r r calc_callees# s
zStats.calc_calleesc C s |}t |trpzt|}W n* tjk
rF |d| 7 }||f Y S X g }|D ] }|t|rP|| qPnzt|}t |t rd| krdk rn n t
|| d }|d | }n2t |t
rd| kr|k rn n|}|d | }t|t|kr|dt|t||f 7 }||fS )Nz#
g g ?g ?r z6 List reduced from %r to %r due to restriction <%r>
) rB r recompileerrorsearchrO rp r& floatrn )r) ZsellistmsgZnew_listZrexrP countr r r eval_print_amount6 s2
"" zStats.eval_print_amountc C s | j }| jr*| jd d }d| j d }nt| j }d}|D ]}| |||\}}q@t|}|sld|fS t|| j d |t| jk rd}|D ] }tt
||krtt
|}q|d |fS )Nz Ordered by:
z! Random listing order was used
r r/ rg ) r7 r3 ro r r: keysr r& r? r# rO )r) Zsel_listwidthZ stat_listr Z selectionr rP r r r get_print_listP s$ zStats.get_print_listc G s | j D ]}t|| jd q| j r,t| jd d}| jD ]}t|t|| jd q6t|| jdd| jd | j| jkrtd| j d| jd td| j | jd t| jd | |\}}|r| |D ]}|
| qt| jd t| jd | S )Nr/ zfunction calls endr0 z(%d primitive calls)zin %.3f seconds) r2 r? r# r9 func_get_function_namer5 r6 r4 r print_title
print_line)r) amountr
indentrP r r r r r print_statsh s(
zStats.print_statsc G sz | |\}}|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 rP r r r
print_callees s
zStats.print_calleesc
G sh | |\}}|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 rP rQ rR rS rT rU r r r
print_callers s zStats.print_callersc
C sv t d|| | jd d}| j D ]0\}}}}}|r&tt| } t| t} qXq&|rrt d| d | jd d S )Nz Function r/ Fr z ncalls tottime cumtime) r? ljustr# r: r nextiterrB rt )
r) name_sizeZ column_titleZ subheaderrQ rR rS rT rU r r r r r s
zStats.print_call_heading->c C s t t||| d| jd |s2t | jd d S t| }d}|D ]}t|}|| } t| tr| \}
} } }
|
| krd|
| f }n
d|
f }d|dd t | t
| t
|
|f }|d
}n$d || t
| j | d f }|d }t || | | jd d}qFd S )
Nr r r/ r. z%d/%dz%dz %s %s %s %srf rg r z %s(%r) %sr_ ) r? rO r r# sortedr rB rt rjustr& f8r: )r) r sourceZ call_dictZarrowZclistr rP r
r rR rQ rS rT ZsubstatsZ
left_widthr r r r s0
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/ r? r# rx r r r r s zStats.print_titlec C s | j | \}}}}}t|}||kr4|d t| }t|dd| jd tt|d| jd |dkrxtdd| jd ntt|| d| jd tt|d| jd |dkrtdd| jd ntt|| d| jd tt|| jd d S )N/ r r r r r/ )r: r r? r r# r rO )r) rP rQ rR rS rT rU cr r r r s zStats.print_line)r )r r r! __doc__r, r' r< r= r( r^ rh rl ru ry r} r~ r r r r r r r r r r r r r r 7 sF !
'
c @ s e Zd ZdZdd Zdd ZdS )rr 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 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.c C s
|| _ d S rv comp_select_list)r) r r r r r, s zTupleComp.__init__c C sF | j D ]:\}}|| }|| }||k r0| S ||kr| S qdS rV r )r) leftrightindex directionlrr r r rs s
zTupleComp.compareN)r r r! r r, rs r r r r rr s rr c C s | \}}}t j|||fS rv )rF pathbasename) func_namer
r r
r r r rz s
rz c C s | d S )Nrg r )rP r r r r s r c C sN | d d dkrB| d }| dr<|drz{%s}r r- z %s:%d(%s))
startswithendswith)r r
r r r rO s rO c C s@ |\}}}}}| \}}} }
} || || || ||
t | |fS )z3Add together all the stats for two profile entries.) add_callers) targetr rQ rR rS rT rU Zt_ccZt_ncZt_ttZt_ctZ t_callersr r r rY s
rY c C s i }| D ]\}}|||< q | D ]V\}}||krtt|trbtdd t||| D ||< q||| |7 < q&|||< q&|S )z*Combine two caller lists in a single list.c s s | ]\}}|| V qd S rv r ).0ijr r r s z add_callers..)rN rB rt zip)r r Z new_callersrP r| r r r r s
"
r c C s d}| D ] }||7 }q |S )z@Sum the caller statistics to get total number of calls received.r )r )rU rR r r r r count_calls s
r c C s d| S )Nz%8.3fr )xr r r r ( s r __main__c @ s e Zd Zd6ddZdd 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 )7ProfileBrowserNc C s6 t j| d| _d | _tj| _|d k r2| | d S )N% ) cmdCmdr, promptr: r$ r% r# do_read)r) profiler r r r, 7 s zProfileBrowser.__init__c C s | }g }|D ]}z|t| W qW n tk
r> Y nX zAdd profile info from given file to current statistics object.r/ r rx r r r help_addh s zProfileBrowser.help_addc C s | d|S )Nr r r) r r r r
do_calleesk s zProfileBrowser.do_calleesc C s t d| jd | d S )Nz6Print callees statistics from the current stat object.r/ r? r# r rx r r r help_calleesm s zProfileBrowser.help_calleesc C s | d|S )Nr r r r r r
do_callersq s zProfileBrowser.do_callersc C s t d| jd | d S )Nz6Print callers statistics from the current stat object.r/ r rx r r r help_callerss s zProfileBrowser.help_callersc C s t d| jd dS )Nr. r/ r r r r r r do_EOFw s zProfileBrowser.do_EOFc C s t d| jd d S NzLeave the profile browser.r/ r rx r r r help_EOFz s zProfileBrowser.help_EOFc C s dS )Nr r r r r r do_quit} s zProfileBrowser.do_quitc C s t d| jd d S r r rx r r r help_quit s zProfileBrowser.help_quitc
C s |rzt || _W nz tk
rN } z t|jd | jd W Y d S d }~X Y n@ tk
r } z"t|jjd || jd W Y d S d }~X Y nX |d | _ n6t
| j dkr| j d d }| | ntd| jd dS ) Nr r/ :r rg z1No statistics object is current -- cannot reload.r ) r r: r r? r* r# r> rL r r r& r )r) r errr r r r s zProfileBrowser.do_readc C s 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 rx r r r help_read s zProfileBrowser.help_readc C s$ | j r| j ntd| jd dS )Nr r/ r )r: ry r? r# r r r r
do_reverse s zProfileBrowser.do_reversec C s t d| jd d S )Nz/Reverse the sort order of the profiling report.r/ r rx r r r help_reverse s zProfileBrowser.help_reversec s | j std| jd d S | j |rRt fdd| D rR| j j| n.z/Valid sort keys (unique prefixes are accepted):z%s -- %sr r )
r: r? r# rl allr ru r rh rN )r) r rm r r r r do_sort s
zProfileBrowser.do_sortc C s t d| jd t d| jd d S )Nz.Sort profile data according to specified keys.r/ z3(Typing `sort' without arguments lists valid keys.)r rx r r r help_sort s zProfileBrowser.help_sortc s fddt jD S )Nc s g | ]}| r|qS r )r )r atextr r
s
z0ProfileBrowser.complete_sort..)r rh )r) r r* r r r
complete_sort s z ProfileBrowser.complete_sortc C s | d|S )Nr r r r r r do_stats s zProfileBrowser.do_statsc C s t d| jd | d S )Nz.Print statistics from the current stat object.r/ r rx r r r
help_stats s zProfileBrowser.help_statsc C s$ | j r| j ntd| jd d S )Nr r/ )r: r} r? r# r r r r do_strip s zProfileBrowser.do_stripc C s t d| jd d S )Nz sR