a
ir @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl m
Z
g dZG dd d eeZe d
d G d d
d
Ze d
d G dd dZG dd dZG dd dZdd Zdd Zdd Zdd Zd d Zd d Zd d! Ze d"krddl Z z ddl Z W n e y, Y n0 G d#d$ d$e j Z!e"ej#d%kr\ej#d% Z$ndZ$zPe!e$Z%ej#d&d D ]Z&e%'e& qxe(d'e%j)d( e%* e(d)e%j)d( W n e+y Y n0 dS )*z3Class for printing reports on profiled python code. N)Enum)
cmp_to_key) dataclass)Dict)StatsSortKeyFunctionProfile StatsProfilec @ 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.9/pstats.pyr / s zSortKey.__new__N)
__name__
__module__ __qualname__ZCALLSZ
CUMULATIVEZFILENAMEZLINENAMEZNFLZPCALLSZSTDNAMEZTIMEr r! r! r! r" r $ s r T)Z unsafe_hashc @ sF e Zd ZU eed< eed< eed< eed< eed< eed< eed< dS ) r r r percall_tottimer
percall_cumtime file_name line_numberN)r# r$ r% int__annotations__floatr r! r! r! r" r 9 s
r c @ s* e Zd ZU dZeed< eeef ed< dS )r z0Class for keeping track of an item in inventory.total_tt
func_profilesN) r# r$ r% __doc__r- r, r r r r! r! r! r" r C s
r 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/ Zd0d1 Zd9d3d4Zd5d6 Zd7d8 ZdS ):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 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 )sysstdoutr1 leninitadd)selfr1 argsargr! r! r" __init__k s
zStats.__init__c C s d | _ g | _d | _d| _d| _d| _d| _t | _i | _ i | _
| | z | W n6 t
y td| jrv| jd nd | jd Y n0 d S )Nr zInvalid timing data %s file) all_calleesfilesfcn_listr. total_calls
prim_calls max_name_lenset top_levelstats
sort_arg_dict
load_statsget_top_level_stats Exceptionprintr1 )r7 r9 r! r! r" r5 u s(
z
Stats.initc C s |d u ri | _ d S t|trt|d }t|| _ W d n1 sH0 Y z"t|}t |j
d | }W n Y n0 |g| _ n t |dr|
|j | _ i |_ | j std| j|f d S )Nrbz create_statsz.Cannot create or construct a %r object from %r)rG
isinstancer openmarshalloadosstatr ctimest_mtimer@ hasattrrN TypeError __class__)r7 r9 fZ
file_statsr! r! r" rI s*
*
zStats.load_statsc C s | j D ]p\}\}}}}}| j|7 _| j|7 _| j|7 _d|v rZ| j| tt|| j kr
tt|| _ q
d S )N)Zjprofiler Zprofiler)
rG itemsrB rC r. rF r6 r4 func_std_stringrD )r7 funcccncttctcallersr! r! r" rJ 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 v r| j | }nddddi f}t
||| j |< qq| S Nr )reversedtyper r@ rB rC r. rF r6 rD rA rG r[ add_func_stats)r7 arg_listitemr] rT Z
old_func_statr! r! r" r6 s(
z Stats.addc C s< t |d }t| j| W d n1 s.0 Y dS )z:Write the profile data to a file we know how to load back.wbN)rP rQ dumprG )r7 r rZ 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))rq rm ro 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 ||v rFd||< q |||< |dd }q*q |D ]
}||= qb| j S )z)Expand all abbreviations that are unique.r Nr; )rH sort_arg_dict_defaultr[ )r7 dictZbad_listwordtupfragmentr! 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 rt rt z Can't have mixed argument typer! r< z, )keyr; )rA r4 rO r+ re rX rz sort_typer r rG r[ appendr\ sortr TupleCompcompare)r7 Zfieldr9 Z
sort_arg_defsZ
sort_tupleZ connectorrw Z
stats_listr] r^ r_ r` ra rb rA tupler! r! r"
sort_stats sF
zStats.sort_statsc C s | j r| j | S N)rA reverser7 r! r! r"
reverse_order s
zStats.reverse_orderc C s | j }i | _ }d}| D ]\}\}}}}} t|}
tt|
|krRtt|
}i } | D ]\} }
|
| t| < q^|
|v rt||
||||| f||
< q ||||| f||
< q | j}t | _}|D ]}|t| q|| _ d | _
d | _ | S rc ) rG r[ func_strip_pathr4 r\ rf rF rE r6 rD rA r? )r7 ZoldstatsZnewstatsrD r] r^ r_ r` ra rb ZnewfuncZ
newcallersfunc2callerZold_topZnew_topr! r! r"
strip_dirs s0
zStats.strip_dirsc
C st | j r
d S i | _ }| j D ]P\}\}}}}}||vr@i ||< | D ]$\}} ||vr`i ||< | || |< qHq d S r )r? rG r[ )
r7 r? r] r^ r_ r` ra rb r r r! r! r" calc_callees5 s
zStats.calc_calleesc C s |}t |trnzt|}W n( tjyD |d| 7 }||f Y S 0 g }|D ] }|t|rN|| qNnzt|}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#