U
e5dj @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ ddgZ G dd de
eZ G dd dZ G dd d Z
d
d Zd d
Zdd Zdd Zdd Zdd Zdd Zedkr|d dlZz d dlZW n ek
r Y nX G dd dejZee j dkre j d Z ndZ zPee Z e j d d D ]Z e e q,e!d e j"d e # e!d e j"d W n e$k
rz Y nX dS ) 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 ZddddZdd 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. Zd6d0d1Zd2d3 Zd4d5 ZdS )7r 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 d S )Nwb)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 |||< |d d }q*q |D ]
}||= qb| j S )Nr r- )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! 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 sD "
'
c @ s e Zd Zdd Zdd ZdS )rr 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, 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 rv ) 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 )Nc 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 rV )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 sP