3
\g @ sf d Z ddlZddlZddlZddlZddlZddlmZ dgZG dd dZ G dd dZ
dd Z d
d Z d d
Z
dd Zdd Zdd Zdd ZedkrbddlZy ddlZW n ek
r Y nX G dd dejZeejdkrejd ZndZyTeeZx"ejdd D ]Z ej e qW e dej d ej e d ej d W n e!k
r` Y nX dS ) z3Class for printing reports on profiled python code. N)
cmp_to_keyStatsc @ s e Zd ZdZddddZdd Zdd Zd
d Zd d
Zdd Z dBdFdJdNdQdTdWdZd]dbdfdidmdqd!Z
d"d# Z d$d% Z d&d' Z
d(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zdrd9d:Zd;d< Zd=d> ZdS )sr 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 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)
N)streamc G sF |pt j| _t|sd }n|d }|dd }| j| | j| d S )Nr )sysstdoutr leninitadd)selfr argsarg r /usr/lib64/python3.6/pstats.py__init__A s
zStats.__init__c C s d | _ g | _d | _d| _d| _d| _d| _t | _i | _ i | _
| j | y | j W n8 t
k
r td| jrx| jd nd | jd Y nX d 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 K s"
z
Stats.initc C s |d kri | _ d S t|trxt|d}tj|| _ W d Q R X y"tj|}tj |j
d | }W n Y nX |g| _ n t |dr|j
|j | _ i |_ | j std| j|f 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 ]p\}\}}}}}| j|7 _| j|7 _| j|7 _d|kr\| jj| tt|| j kr tt|| _ q W d S )Njprofiler profiler)r2 r r3 )
r itemsr r r r r
r func_std_stringr )r funcccncttctcallersr r r r t 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 ]}| jj| qvW | j |j k r|j | _ d | _
xJ|j j D ]<\}}|| j kr| j | }nddddi f}t
||| j |< qW qW | S )Nr )reversedtyper r r r r r r
r r r r4 add_func_stats)r arg_listitemr6 r+ Z
old_func_statr r r r
~ s(
z Stats.addc
C s( t |d}tj| j| W dQ R X dS )z:Write the profile data to a file we know how to load back.wbN)r' r( dumpr )r filenamer1 r r r
dump_stats s zStats.dump_statsr
call count cumulative time file name line number
function namename/file/liner primitive call count
standard name
internal time)callsZncallsZcumtime
cumulativer rC linemodulenameZnflZpcallsstdnamer, Ztottimec C s | j szi | _ }i }xP| jj D ]B\}}|}x4|r`|s8P ||krJd||< P |||< |dd }q.W q W x|D ]
}||= qlW | j S )z)Expand all abbreviations that are unique.r Nr r )r sort_arg_dict_defaultr4 )r dictZbad_listwordtupZfragmentr r r get_sort_arg_defs s
zStats.get_sort_arg_defsc G s" |sd| _ | S t|dkrFt|d trFd dddddddi|d g}| j }f }d| _d}x:|D ]2}||| d }| j||| d 7 _d }qbW g }xB| jj D ]4\}\}} }
} } |j|| |
| f| t ||f qW |j
t t |j
d
g | _ }
x |D ]}|
j|d qW | S )
Nr r rY rT r, rR rU r z, )keyr r )r r r% intr^ sort_typer r4 appendr5 sortr TupleCompcompare)r ZfieldZ
sort_arg_defsZ
sort_tupleZ connectorr\ Z
stats_listr6 r7 r8 r9 r: r; r tupler r r
sort_stats s2
zStats.sort_statsc C s | j r| j j | S )N)r reverse)r r r r
reverse_order s
zStats.reverse_orderc C s | j }i | _ }d}x|j D ]\}\}}}}} t|}
tt|
|krTtt|
}i } x | j D ]\} }
|
| t| < qbW |
|krt||
||||| f||
< q ||||| f||
< q W | j}t | _}x|D ]}|jt| qW || _ d | _
d | _ | S )Nr ) r r4 func_strip_pathr r5 r> 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 S i | _ }xb| jj D ]T\}\}}}}}||krBi ||< x0|j D ]$\}} ||krdi ||< | || |< qLW q W d S )N)r r r4 )
r r r6 r7 r8 r9 r: r; rk rl r r r calc_callees s
zStats.calc_calleesc
C s |}t |trpytj|}W n& tjk
rB |d| 7 }||fS X g }x|D ] }|jt|rN|j| qNW nzt|}t |t rd| kodk n rt
|| d }|d | }n2t |t
rd| ko|k n r|}|d | }t|t|kr|dt|t||f 7 }||fS )Nz#