e ff @ s d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d g Z Gd d d Z Gd d d Z
d d Z d
d Z d d
Z
d d Z d d Z d d Z d d Z e d k rd d l Z y d d l Z Wn e k
rYn XGd d d e j Z e e j d k rPe j d Z n d Z yk e e Z x( e j d d D] Z e j e qyWe d d e j e j e d d e j Wn e! k
rYn Xn d S) z3Class for printing reports on profiled python code. N)
cmp_to_keyStatsc @ sw e Z d Z d Z d d d d Z d d Z d d Z d
d Z d d
Z d d Z i dO d 6dS d 6dW d 6d[ d 6d^ d 6da d 6dd d 6dg d 6dj d" 6do d$ 6ds d' 6dv d* 6dz d- 6d~ d. 6Z
d/ d0 Z d1 d2 Z d3 d4 Z
d5 d6 Z d7 d8 Z d9 d: Z d; d< Z d= d> Z d? d@ Z dA dB Z dC dD Z dE dF dG Z dH dI Z dJ dK Z d S)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). It takes an arbitrary number of
quoted strings to select the sort order. For example sort_stats('time',
'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)
streamNc G s_ | p t j | _ t | s' d } n | d } | d d } | j | | j | d S)Nr )sysstdoutr leninitadd)selfr argsarg r +/opt/alt/python34/lib64/python3.4/pstats.py__init__> s
zStats.__init__c C s d | _ g | _ d | _ d | _ d | _ d | _ d | _ t | _ i | _ i | _
| j | y | j WnB t
k
r t d | j r | j d n d d | j Yn Xd S)Nr zInvalid timing data %sr 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 H s"
'z
Stats.initc C s | d k r i | _ d St | t r t | d } t j | | _ Wd QXy- t j | } t j | j
d | } Wn Yn X| g | _ n1 t | d r | j
| j | _ i | _ n | j s t d | j | f n d S)Nrbz create_statsz.Cannot create or construct a %r object from %r)r
isinstancestropenmarshalloadosstattimeZctimest_mtimer hasattrr$ TypeError __class__)r r
fZ
file_statsr r r r [ s(
zStats.load_statsc C s x | j j D] \ } \ } } } } } | j | 7_ | j | 7_ | j | 7_ d | k rw | j j | n t t | | j k r t t | | _ q q Wd S)Njprofiler profiler)r2 r r3 )
r itemsr r r r r
r func_std_stringr )r funcccncttctcallersr r r r q s + zStats.get_top_level_statsc G sL | s
| Sx;t | D]-} t | t | k rD t | } n | j | j 7_ | j | j 7_ | j | j 7_ | j | j 7_ x! | j D] } | j j | q W| j | j k r | j | _ n d | _
xg | j j D]V \ } } | | j k r| j | } n d d d d i f } t
| | | j | r r r
r r r )r ZoldstatsZnewstatsr r6 r7 r8 r9 r: r; ZnewfuncZ
newcallersfunc2callerZold_topZnew_topr r r
strip_dirs s.
(
zStats.strip_dirsc
C s | j r
d Si | _ } x | j j D]x \ } \ } } } } } | | k r^ i | |
g g ?g ?r z6 List reduced from %r to %r due to restriction <%r>
) r% r& recompileerrorsearchr5 rb r floatr` )r ZsellistmsgZnew_listZrexr6 countr r r eval_print_amount s,
++ zStats.eval_print_amountc C s | j } | j r9 | j d d } d | j d } n t | j j } d } x) | D]! } | j | | | \ } } q[ Wt | } | s d | f St | d | j | t | j k r d } x> | D]3 } t t
| | k r t t
| } q q Wn | d | f S)Nz Ordered by:
z! Random listing order was used
r r rX ) r r ra rt r keysrw r r" r r5 )r Zsel_listwidthZ stat_listru Z selectionrv r6 r r r get_print_list/ s$
zStats.get_print_listc G sd x$ | j D] } t | d | j q
W| j rC t d | j n d } x- | j D]" } t | t | d | j qS Wt | | j d d d d | j | j | j k r t d | j d d d | j n t d | j d | j t d | j | j | \ } } | r`| j x | D] } | j
| q&Wt d | j t d | j n | S) Nr zfunction callsendz(%d primitive calls)zin %.3f secondsz ) r r" r r func_get_function_namer r r r{ print_title
print_line)r amountrC indentr6 rz rt r r r print_statsG s( "#
zStats.print_statsc G s | j | \ } } | r | j | j | d xM | D]E } | | j k rn | j | | | j | q< | j | | i q< Wt d | j t d | j n | S)Nz called...r )r{ rn print_call_headingr print_call_liner" r )r r rz rt r6 r r r
print_callees^ s
zStats.print_calleesc
G s | j | \ } } | r | j | d x@ | D]8 } | j | \ } } } } } | j | | | d q2 Wt d | j t d | j n | S)Nzwas called by...z<-r )r{ r r r r" r )
r r rz rt r6 r7 r8 r9 r: r; r r r
print_callersm s
zStats.print_callersc
C s t d j | | d | j d } xW | j j D]F \ } } } } } | r6 t t | j } t | t } Pq6 q6 W| r t d | d d | j n d S)Nz Function r Fr| z ncalls tottime cumtime) r" ljustr r valuesnextiterr% rf )
r name_sizeZ column_titleZ subheaderr7 r8 r9 r: r; valuer r r r x s %zStats.print_call_headingz->c C sk t t | j | | d d d | j | sF t d | j d St | j } d } x| D] } t | } | | } t | t r| \ }
} } }
|
| k r d |
| f } n
d |
f } d | j d d t | t
| t
|
| f } | d
} n. d | | t
| j | d f } | d } t | | | d | j d } qe Wd S)
Nr~ r| r r z%d/%dz%dz %s %s %s %srU rX r z %s(%r) %srG ) r" r5 r r sortedry r% rf rjustr f8r )r r sourceZ call_dictZarrowZclistr r6 rR r r8 r7 r9 r: ZsubstatsZ
left_widthr r r r s* ,
$
zStats.print_call_linec C s0 t d d d d | j t d d | j d S)Nz- ncalls tottime percall cumtime percallr~ r| r zfilename:lineno(function))r" r )r r r r r s zStats.print_titlec C s^ | j | \ } } } } } t | } | | k rK | d t | } n t | j d d d d | j t t | d d d | j | d k r t d d d d | j n# t t | | d d d | j t t | d d d | j | d k r t d d d d | j n# t t | | d d d | j t t | d | j d S)
N/ r~ r| r r r} z z )r r& r" r r r r5 )r r6 r7 r8 r9 r: r; cr r r r s " # #zStats.print_liner r r r )r rE r r r r )r rE r rG r r )r rH r rG r r )r rH rJ r r )r rK rJ r r )r rK rL r r )r rM rJ r r )r rK rP r r )r rQ rP r rJ r rL r r r r )r rS r r r r )r rT rU r r )r rV r rX r r )r rY r rX r r )r rY )__name__
__module__ __qualname____doc__r r r r r
rD rZ r^ rg ri rm rn rw r{ r r r r r r r r r r r r sH
c @ s. e Z d Z d Z d d Z d d Z d S)rd 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)N)comp_select_list)r r r r r r s zTupleComp.__init__c C sS xL | j D]A \ } } | | } | | } | | k r; | S| | k r
| Sq
Wd S)Nr )r )r leftrightindexZ directionlrr r r re s
zTupleComp.compareN)r r r r r re r r r r rd s rd c C s( | \ } } } t j j | | | f S)N)r* pathbasename) func_namerC rN rR r r r rj s rj c C s | d S)NrX r )r6 r r r r s r c C sc | d d d k rW | d } | j d rP | j d rP d | d d
S| Sn d | Sd S) NrX ~r <>z{%s}r z %s:%d(%s))r r r )
startswithendswith)r rR r r r r5 s
r5 c C sV | \ } } } } } | \ } } } }
} | | | | | | | |
t | | f S)z3Add together all the stats for two profile entries.) add_callers) targetr r7 r8 r9 r: r; Zt_ccZt_ncZt_ttZt_ctZ t_callersr r r r> s r> c C s i } x$ | j D] \ } } | | | s z add_callers..)r4 r% rf zip)r r Z new_callersr6 rl r r r r s !r c C s+ d } x | j D] } | | 7} q W| S)z@Sum the caller statistics to get total number of calls received.r )r )r; r8 rF r r r count_calls s r c C s d | S)Nz%8.3fr )xr r r r s r __main__c @ sK e Z d Z d d d Z d d Z d d Z d d Z d
d Z d d
Z d d Z d d Z
d d Z d d Z d d Z
d d Z d d Z d d Z d d Z d d! Z d" d# Z d$ d% Z d& d' Z d( d) Z d* d+ Z d, d- Z d. d/ Z d0 d1 Z d2 d3 Z d4 d5 Z d S)6ProfileBrowserNc C sN t j j | d | _ d | _ t j | _ | d k rJ | j | n d S)Nz% ) cmdCmdr promptr r r r do_read)r profiler r r r s zProfileBrowser.__init__c C s | j } g } x | D] } y | j t | w Wn t k
rM Yn XyQ t | } | d k su | d k r t d d | j w n | j | w Wn t k
r Yn X| j | q W| j r t | j | | n t d d | j d S)Nr r z#Fraction argument must be in [0, 1]r z No statistics object is loaded.) splitrb r`
ValueErrorrs r" r r getattr)r fnrN r Z processedZtermZfracr r r generic s,
zProfileBrowser.genericc C sv t d d | j t d d | j t d d | j t d d | j t d d | j t d d | j 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_help7 s zProfileBrowser.generic_helpc C s3 | j r | j j | n t d d | j d S)Nz No statistics object is loaded.r r )r r
r" r )r rN r r r do_add? s zProfileBrowser.do_addc C s t d d | j d S)Nz>Add profile info from given file to current statistics object.r )r" r )r r r r help_addE s zProfileBrowser.help_addc C s | j d | S)Nr )r )r rN r r r
do_calleesH s zProfileBrowser.do_calleesc C s! t d d | j | j d S)Nz6Print callees statistics from the current stat object.r )r" r r )r r r r help_calleesJ s zProfileBrowser.help_calleesc C s | j d | S)Nr )r )r rN r r r
do_callersN s zProfileBrowser.do_callersc C s! t d d | j | j d S)Nz6Print callers statistics from the current stat object.r )r" r r )r r r r help_callersP s zProfileBrowser.help_callersc C s t d d | j d S)Nr r r )r" r )r rN r r r do_EOFT s zProfileBrowser.do_EOFc C s t d d | j d S)NzLeave the profile brower.r )r" r )r r r r help_EOFW s zProfileBrowser.help_EOFc C s d S)Nr r )r rN r r r do_quitZ s zProfileBrowser.do_quitc C s t d d | j d S)NzLeave the profile brower.r )r" r )r r r r help_quit\ s zProfileBrowser.help_quitc C s | r y t | | _ Wn t k
r^ } z# t | j d d | j d SWYd d } ~ XnI t k
r } z) t | j j d | d | j d SWYd d } ~ Xn X| d | _ nK t
| j d k r | j d d } | j | n t d d | j d S) Nr r :z% rX z1No statistics object is current -- cannot reload.r ) r r OSErrorr" r r r! r0 r r r r )r rN errr r r r _ s zProfileBrowser.do_readc C s* t d d | j t d d | j 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_readp s zProfileBrowser.help_readc C s0 | j r | j j n t d d | j d S)Nz No statistics object is loaded.r r )r ri r" r )r rN r r r
do_reverset s zProfileBrowser.do_reversec C s t d d | j d S)Nz/Reverse the sort order of the profiling report.r )r" r )r r r r help_reversez s zProfileBrowser.help_reversec s | j s t d d | j d S| j j | rs t f d d | j D rs | j j | j nT t d d | j x> t j j D]- \ } } t d | | d f d | j q Wd S) Nz No statistics object is loaded.r c 3 s | ] } | k Vq d S)Nr )r r )abbrevsr r s z)ProfileBrowser.do_sort..z/Valid sort keys (unique prefixes are accepted):z%s -- %sr r )
r r" r r^ allr rg r rZ r4 )r rN r_ r r )r r do_sort} s + %zProfileBrowser.do_sortc C s* t d d | j t d d | j 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 f d d t j D S)Nc s% g | ] } | j r | q Sr )r )r a)textr r r s z0ProfileBrowser.complete_sort..)r rZ )r r r r )r r
complete_sort s z ProfileBrowser.complete_sortc C s | j d | S)Nr )r )r rN r r r do_stats s zProfileBrowser.do_statsc C s! t d d | j | j 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 s0 | j r | j j n t d d | j d S)Nz No statistics object is loaded.r )r rm r" r )r rN r r r do_strip s zProfileBrowser.do_stripc C s t d d | j d S)Nz r r r r r readline ImportErrorr r r argvZ initprofileZbrowserr r r" r ZcmdloopKeyboardInterruptr r r r sH