a
ir @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z g dZ
G dd deeZe
d d
G d d d Ze
d d
G d
d dZG dd dZG dd dZdd Zdd Zdd Zdd Zdd Zd d Zd d Zed!krd dl Z z d dl Z W n e y( Y n0 G d"d# d#e j Z e!e j"d$krXe j"d$ Z#ndZ#zPe e#Z$e j"d%d D ]Z%e$&e% qte'd&e$j(d' e$) e'd(e$j(d' W n e*y Y n0 dS )) 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 eed< eeef ed< dS )r total_tt
func_profilesN)r# r$ r% r- r, r r r r! r! r! r" r C s
r 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. Zd/d0 Zd8d2d3Zd4d5 Zd6d7 ZdS )9r N)streamc G sF |pt j| _t|sd }n|d }|dd }| | | j| d S r )sysstdoutr0 leninitadd)selfr0 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 Exceptionprintr0 )r6 r8 r! r! r" r4 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)rF
isinstancer openmarshalloadosstatr ctimest_mtimer? hasattrrM TypeError __class__)r6 r8 fZ
file_statsr! r! r" rH 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)
rF itemsrA rB r. rE r5 r3 func_std_stringrC )r6 funcccncttctcallersr! r! r" rI 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? rA rB r. rE r5 rC r@ rF rZ add_func_stats)r6 arg_listitemr\ rS Z
old_func_statr! r! r" r5 s(
z Stats.addc C s< t |d }t| j| W d n1 s.0 Y d S )Nwb)rO rP dumprF )r6 r rY 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))rp rl rn 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 |||< |d d }q*q |D ]
}||= qb| j S )Nr r: )rG sort_arg_dict_defaultrZ )r6 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 rs rs z Can't have mixed argument typer! r; z, )keyr: )r@ r3 rN r+ rd rW ry sort_typer r rF rZ appendr[ sortr TupleCompcompare)r6 Zfieldr8 Z
sort_arg_defsZ
sort_tupleZ connectorrv Z
stats_listr\ r] r^ r_ r` ra r@ tupler! r! r"
sort_stats sF
zStats.sort_statsc C s | j r| j | S N)r@ reverser6 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 rb ) rF rZ func_strip_pathr3 r[ re rE rD r5 rC r@ r> )r6 ZoldstatsZnewstatsrC r\ r] r^ r_ r` ra 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> rF rZ )
r6 r> r\ r] r^ r_ r` ra 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#