B
&}| @ 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 Zedkrd dlZy d dlZW n ek
r Y nX G dd dejZee j dkre j d Z ndZ yTee Z x"e j d d D ]Z e e q.W e!d e j"d e # e!d e j"d W n e$k
r~ 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> t | }|d |_x |dd D ]}|| j|< q"W ||_|S )Nr )str__new___value__value2member_map_Z _all_values)clsvaluesobjZ other_value r +/opt/alt/python37/lib64/python3.7/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 )Nr 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 | _
| | y | 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 d Q R X y"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)r7
isinstancer openmarshalloadosstatr Zctimest_mtimer/ hasattrr> TypeError __class__)r' r) fZ
file_statsr r r r9 v s(
zStats.load_statsc C s x~| j D ]p\}\}}}}}| j|7 _| j|7 _| j|7 _d|kr\| j| tt|| j kr tt|| _ q W d S )N)Zjprofiler Zprofiler)
r7 itemsr2 r3 r1 r6 r& r$ func_std_stringr4 )r' funcccncttctcallersr r r r: s zStats.get_top_level_statsc G s |s| S xt |D ]}t| t|kr.t|}| j|j7 _| j|j7 _| j|j7 _| j|j7 _x|jD ]}| j| qvW | j |j k r|j | _ d | _
xJ|j D ]<\}}|| j kr| j | }nddddi f}t
||| j |< qW qW | S )Nr )reversedtyper r/ r2 r3 r1 r6 r& r4 r0 r7 rJ add_func_stats)r' arg_listitemrL rD Z
old_func_statr r r r& s(
z Stats.addc C s( t |d}t| j| W d Q R X d S )Nwb)r@ rA dumpr7 )r' r
rI 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)))r] r )r[ r )r\ r 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 s | j szi | _ }i }xP| j D ]B\}}|}x4|r`|s8P ||krJd||< P |||< |d d }q.W q W x|D ]
}||= qlW | j S )Nr r+ )r8 sort_arg_dict_defaultrJ )r' dictZbad_listwordtupZfragmentr r r get_sort_arg_defs s
zStats.get_sort_arg_defsc G sl |sd| _ | S t|dkrBt|d trBddddd|d g}n>t|dkrx0|dd D ] }t|t|d kr\td q\W | }d
}d | _d }xJ|D ]B}t|tr|j }||| d }| j||| d 7 _d }qW g }xB| j
D ]4\}\} }
} } }
| | |
| | f| t
||f qW |jtt|jd
g | _ }x |D ]}| |d qPW | S )Nr r r r r r )r+ r r r_ r_ z Can't have mixed argument typer r, z, )keyr+ )r0 r$ r? intrS rG rd sort_typer valuer7 rJ appendrK sortr TupleCompcompare)r' Zfieldr) Z
sort_arg_defsZ
sort_tupleZ connectorrb Z
stats_listrL rM rN rO rP rQ r0 tupler r r
sort_stats s>
zStats.sort_statsc C s | j r| j | S )N)r0 reverse)r' r r r
reverse_order s
zStats.reverse_orderc C s | j }i | _ }d}x| D ]\}\}}}}} t|}
tt|
|krTtt|
}i } x | D ]\} }
|
| t| < qbW |
|krt||
||||| f||
< q ||||| f||
< q W | j}t | _}x|D ]}|t| qW || _ d | _
d | _ | S )Nr ) r7 rJ func_strip_pathr$ rK rT r6 r5 r& r4 r0 r. )r' ZoldstatsZnewstatsr4 rL rM rN rO rP rQ ZnewfuncZ
newcallersfunc2callerZold_topZnew_topr r r
strip_dirs s.
zStats.strip_dirsc
C s| | j r
d S i | _ }xb| j D ]T\}\}}}}}||krBi ||< x0| D ]$\}} ||krdi ||< | || |< qLW q W d S )N)r. r7 rJ )
r' r. rL rM rN rO rP rQ rr rs r r r calc_callees# s
zStats.calc_calleesc C s |}t |trpyt|}W n& tjk
rB |d| 7 }||fS X g }x|D ] }|t|rN|| qNW nzt|}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>
) r? r recompileerrorsearchrK ri r$ floatrf )r' ZsellistmsgZnew_listZrexrL countr r r eval_print_amount6 s,
"" zStats.eval_print_amountc C s | j }| jr*| jd d }d| j d }nt| j }d}x |D ]}| |||\}}qBW t|}|spd|fS t|| j d |t| jk rd}x(|D ] }tt
||krtt
|}qW |d |fS )Nz Ordered by:
z! Random listing order was used
r )r- r_ ) r4 r0 rg r{ r7 keysr~ r$ r< r! rK )r' Zsel_listwidthZ stat_listr| Z selectionr} rL r r r get_print_listP s$
zStats.get_print_listc G s x | j D ]}t|| jd qW | j r0t| jd d}x"| jD ]}t|t|| jd qc C s t t||| d| jd |s2t | jd d S t| }d}x|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}qHW d S )
Nr )r r- )r- r, z%d/%dz%dz %s %s %s %sr^ r_ r z %s(%r) %srZ ) r< rK r r! sortedr r? rm rjustr$ f8r7 )r' r sourceZ call_dictZarrowZclistr rL r
rh rN rM rO rP 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 r- zfilename:lineno(function))r- )r< r! )r' 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 z )r- )r7 r r< r r! r rK )r' rL rM rN rO rP rQ cr r r r s zStats.print_line)r )r r r r* r% r9 r: r& rY r` rd rn rp rt ru r~ r r r r r r r r r r r r r 7 sB "
'
c @ s e Zd Zdd Zdd ZdS )rk c C s
|| _ d S )N)comp_select_list)r' r r r r r* s zTupleComp.__init__c C sB x<| j D ]2\}}|| }|| }||k r.| S ||kr|S qW dS )Nr )r )r' leftrightindex directionlrr r r rl s zTupleComp.compareN)r r r r* rl r r r r rk s rk c C s | \}}}t j|||fS )N)rC pathbasename) func_namer
r r
r r r rq s
rq c C s | d S )Nr_ r )rL 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 rK s rK c C s@ |\}}}}}| \}}} }
} || || || ||
t | |fS )N) add_callers) targetr rM rN rO rP rQ Zt_ccZt_ncZt_ttZt_ctZ t_callersr r r rT s rT c C s i }x | D ]\}}|||< qW xb| D ]V\}}||krzt|trhtdd t||| D ||< q|| |7 < q,|||< q,W |S )Nc s s | ]\}}|| V qd S )Nr ).0ijr r r s z add_callers..)rJ r? rm zip)r r Z new_callersrL rs r r r r s
" r c C s" d}x| D ] }||7 }qW |S )Nr )r )rQ rN 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 )Nz% ) cmdCmdr* promptr7 r" r# r! do_read)r' profiler r r r* 7 s zProfileBrowser.__init__c C s | }g }x|D ]}y|t| wW n tk
r> Y nX y8t|}|dksZ|dk rjtd| jd w|| wW n tk
r Y nX || qW | jrt| j|| ntd| jd dS )Nr r z#Fraction argument must be in [0, 1])r- z No statistics object is loaded.) splitri rf
ValueErrorrz r< r! r7 getattr)r' fnr r( Z processedZtermZfracr r r generic? s,
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 r generic_helpW s zProfileBrowser.generic_helpc
C sd | j rRy| j | W q` tk
rN } ztd||f | jd W d d }~X Y q`X ntd| jd dS )Nz$Failed to load statistics for %s: %s)r- z No statistics object is loaded.r )r7 r& OSErrorr< r! )r' r er r r do_add_ s *zProfileBrowser.do_addc C s t d| jd d S )Nz>Add profile info from given file to current statistics object.)r- )r< r! )r' 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 )r' r r r help_calleesm s zProfileBrowser.help_calleesc C s | d|S )Nr )r )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< r! r )r' r r r help_callerss s zProfileBrowser.help_callersc C s t d| jd dS )Nr, )r- r )r< r! )r' r r r r do_EOFw s zProfileBrowser.do_EOFc C s t d| jd d S )NzLeave the profile brower.)r- )r< r! )r' r r r help_EOFz s zProfileBrowser.help_EOFc C s dS )Nr r )r' r r r r do_quit} s zProfileBrowser.do_quitc C s t d| jd d S )NzLeave the profile brower.)r- )r< r! )r' r r r help_quit s zProfileBrowser.help_quitc
C s |ryt || _W nn tk
rH } zt|jd | jd d S d }~X Y n: tk
r } z t|jjd || jd d S d }~X Y nX |d | _ n6t
| j dkr| j d d }| | ntd| jd dS ) Nr )r- :z% r_ z1No statistics object is current -- cannot reload.r ) r r7 r r< r( r! r; rH 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< r! )r' r r r help_read s zProfileBrowser.help_readc C s$ | j r| j ntd| jd dS )Nz No statistics object is loaded.)r- r )r7 rp r< 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< r! )r' r r r help_reverse s zProfileBrowser.help_reversec s | j std| jd d S | j |rRt fdd| D rR| j j| n@td| jd x0tj D ]"\}}td||d f | jd qlW dS ) Nz No statistics object is loaded.)r- c 3 s | ]}| kV qd S )Nr )r r )abbrevsr r r s z)ProfileBrowser.do_sort..z/Valid sort keys (unique prefixes are accepted):z%s -- %sr r )
r7 r< r! rd allr rn r r` rJ )r' r re rh 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< r! )r' r r r help_sort s zProfileBrowser.help_sortc s fddt jD S )Nc s g | ]}| r|qS r )r )r a)textr r
s z0ProfileBrowser.complete_sort..)r r` )r' r r( r )r r
complete_sort s z ProfileBrowser.complete_sortc C s | d|S )Nr )r )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< r! r )r' r r r
help_stats s zProfileBrowser.help_statsc C s$ | j r| j ntd| jd d S )Nz No statistics object is loaded.)r- )r7 rt r< r! )r' r r r r do_strip s zProfileBrowser.do_stripc C s t d| jd d S )Nz sP