File manager

3


 \�g�@sb
dd
lZdd
l
Z
dd
lZdd
lZdd
lZddlmZ
dg
ZGdd�d�ZGdd�d�Z	dd�Z
d	d
�Zdd�Zd
d�Z
dd�Zdd�Zdd�Zedk�
r^dd
lZydd
lZWnek
r�


YnXGdd�dej�Zeej�
dkr�ejdZnd
ZyTee�
Zx"ejdd
�D]Zeje�

�
qWedejd�
ej�
edejd�
Wne k
�
r\


YnXd
S)�N)
�
cmp_to_key�Statsc@s�eZ
dZd
d�
dd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�ZdAdEdIdMdPdSdVdYd\dadedhdldpd �Z	d!d"�Z
d#d$�Zd%d&�Zd'd(�Z
d)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Zdqd8d9�
Zd:d;�Zd<d=�Zd
S)rrN)
�streamc

GsF|
ptj
|_t|�
sd}n|d
}|dd�}|j|�

|j|�
dS)Nr
�
)�sys�stdoutr�len�init�add)�selfr�args�arg�r�/usr/lib64/python3.6/pstats.py�__init__As






zStats.__init__cCs�d|_g|_
d|_d
|_d
|_d
|_d
|_t�|_i|_	i|_
|j|
�

y|j�
Wn8t
k
r�


td|j
rx|j
dnd|jd�
�YnXdS)Nr
zInvalid timing data %sr�)
�file���)�all_callees�files�fcn_list�total_tt�total_calls�
prim_calls�max_name_len�set�	top_level�stats�
sort_arg_dict�
load_stats�get_top_level_stats�	Exception�printr)rr
rrrr	Ks"

















z
Stats.initcCs�|
dkri|_dSt
|
t�rxt|
d
��}tj|�
|_WdQRXy"tj|
�
}tj	|j
�
d|
}
Wn


YnX|
g
|_n t|
d�r�|
j
�
|
j|_i|
_|js�td|j|
f�
�
dS)N�rbz    �create_statsz.Cannot create or construct a %r object from %r)r�
isinstance�str�open�marshal�load�os�stat�timeZctime�st_mtimer�hasattrr$�	TypeError�	__class__)rr
�
fZ
file_statsrrrr^s(























zStats.load_statsc
Cs�x~|jj
�D]p\}
\}}}}}|j|7_|j|7_|j|7_d|kr\|jj|
�

tt|
�
�
|j	krtt|
�
�
|_	qWdS)N�jprofiler
�profiler)r2r
r3)
r�itemsrrrrr
r�func_std_stringr)r�func�cc�nc�tt�ct�callersrrrr ts







zStats.get_top_level_statsc
Gs�|
s|Sx�t|
�
D]�}t
|�
t
|�
kr.t|�
}|j|j7_|j|j7_|j|j7_|j|j7_x|jD]}|jj|�

qvW|j	|j	kr�|j	|_	d|_
xJ|jj�D]<\}}||jkr�|j|}nd
d
d
d
if}t
||�|j|<q�WqW|S)Nr
)�reversed�typerrrrrrr
rrrr4�add_func_stats)r�arg_list�itemr6r+Z
old_func_statrrrr
~s(
















z	Stats.addc
Cs(t|
d
��}t
j|j|�
WdQRXdS)N�wb)r'r(�dumpr)r�filenamer1rrr�
dump_stats�s
zStats.dump_statsr�
call count��cumulative time��	file name��line number��
function name�name/file/liner
�primitive call count��
standard name��
internal time)�callsZncallsZcumtime�
cumulativerrC�line�module�nameZnflZpcalls�stdnamer,Ztottimec
Cs�|jszi|_}
i}xP|j
j�D]B\}}|}x4|r`|s8P||
krJd
||<P||
|<|dd�}q.Wq Wx|D]
}|
|=qlW|jS)Nr
rr)r�sort_arg_dict_defaultr4)r�dictZbad_list�word�tupZfragmentrrr�get_sort_arg_defs�s 

















zStats.get_sort_arg_defsc
Gs"
|
sd
|_|St
|
�
dkrFt|
d
t�rFddd
dddddi|
d
g
}
|j�}f}d|_d}x:|
D]2}|||d
}|j|||d7_d	}qbWg}xB|jj�D]4\}\}}	}
}}|j||	|
|f|t	|�
|f�

q�W|j
tt|�
j
�
d
�

g|_}
x|D]}|
j|d�

�
qW|S)
Nr
rrYrTr,rRrUrz, )
�keyrr)rrr%�intr^�	sort_typerr4�appendr5�sortr�	TupleComp�compare)rZfieldZ
sort_arg_defsZ
sort_tupleZ	connectorr\Z
stats_listr6r7r8r9r:r;r�tuplerrr�
sort_stats�s2






















zStats.sort_statsc


Cs|jr|jj
�
|S)
N)r�reverse)
rrrr�
reverse_order�s



zStats.reverse_orderc
Cs�|j}
i|_}d
}x�|
j
�D]�\}\}}}}}	t|�
}
tt|
�
�
|krTtt|
�
�
}i}x |	j
�D]\}}
|
|t|�
<qbW|
|kr�t||
|||||f�||
<q|||||f||
<qW|j}t�|_}x|D]}|jt|�
�

q�W||_	d|_
d|_|S)Nr
)rr4�func_strip_pathrr5r>rrr
rrr)rZoldstatsZnewstatsrr6r7r8r9r:r;ZnewfuncZ
newcallers�func2�callerZold_topZnew_toprrr�
strip_dirs�s.




















zStats.strip_dirsc

Cs||jr
dSi|_}
xb|j
j�D]T\}\}}}}}||
krBi|
|<x0|j�D]$\}}	||
krdi|
|<|	|
||<qLWq WdS)
N)rrr4)
rrr6r7r8r9r:r;rkrlrrr�calc_callees
s











zStats.calc_calleesc
Cs
|}t|
t
�rpytj|
�
}Wn&tjk
rB


|d
|
7}||fSXg}x�|D]}|jt|�
�
rN|j|�

qNWnzt|�
}t|
t	�r�d|
k
o�dkn
r�t
||
d�
}|d|�}n2t|
t
�r�d|
k
o�|kn
r�|
}|d|�}t|�
t|�
k�
r|dt|�
t|�
|
f7}||fS)Nz#   <Invalid regular expression %r>
gg�?g�?r
z6   List reduced from %r to %r due to restriction <%r>
)r%r&�re�compile�error�searchr5rbr�floatr`)rZsel�list�msgZnew_listZrexr6�countrrr�eval_print_amount
s,














"


"




zStats.eval_print_amountcCs�|j}|j
r*|j
dd�}d
|jd}nt|jj��
}d}x|
D]}|j|||�\}}qBWt|�
}|spd|fSt||j	d�
|t|j�
kr�d}x(|D] }tt
|�
�
|kr�tt
|�
�
}q�W|d|fS)Nz   Ordered by: �

z!   Random listing order was used
r
)
rrR)rrrartr�keysrwrr"rr5)rZsel_list�widthZ	stat_listruZ	selectionrvr6rrr�get_print_list2
s$















zStats.get_print_listc
Gs�x|jD]}t
||jd
�
qW|jr0t
|jd
�

d}x"|jD]}t
|t|�
|jd
�
q<Wt
||jdd|jd�
|j|jkr�t
d|jd|jd�
t
d|j|jd
�
t
|jd
�

|j|
�
\}}|r�|j	�
x|D]}|j
|�

q�Wt
|jd
�

t
|jd
�

|S)	N)
r�
 �zfunction calls)�endrz(%d primitive calls)zin %.3f secondsz        )rr"rr�func_get_function_namerrrr{�print_title�
print_line)r�amountrC�indentr6rzrtrrr�print_statsJ
s(



















zStats.print_statsc
Gs~|j|
�
\}}|rz|j
�
|j|d
�
x:|D]2}||jkrP|j|||j|�
q,|j||i�
q,Wt|jd�

t|jd�

|S)Nz	called...)
r)r{rn�print_call_headingr�print_call_liner"r)rr�rzrtr6rrr�
print_calleesa
s










zStats.print_calleesc

Gsl|j|
�
\}}|rh|j
|d
�
x0|D](}|j|\}}}}}	|j|||	d�
q$Wt|jd�

t|jd�

|S)Nzwas called by...z<-)
r)r{r�rr�r"r)
rr�rzrtr6r7r8r9r:r;rrr�
print_callersp
s









zStats.print_callersc
Csxtd
j
|
�
||jd�
d}x<|jj�D].\}}}}}|r(tt|j��
�
}	t|	t�}Pq(W|rttd|
d|jd�
dS)Nz	Function )
rFr|z    ncalls  tottime  cumtime)	r"�ljustrr�values�next�iterr%rf)
r�	name_sizeZcolumn_titleZ	subheaderr7r8r9r:r;�valuerrrr�{
s








zStats.print_call_heading�->cCs
tt
|�
j|
�
|d
|jd�
|s2t|jd�

dSt|j��
}d}x�|D]�}t
|�
}||}	t|	t�r�|	\}
}}}
|
|kr�d|
|f}n
d|
f
}d|jdd	t	|�
�
t
|�
t
|
�
|f}|
d
}n$d||	t
|j|d�
f}|
d}t||||jd�
d
}qHWdS)
Nr|)r~r)
rrz%d/%dz%dz%s %s %s  %srPrRrz	%s(%r) %srF)r"r5r�r�sortedryr%rf�rjustr�f8r)rr��sourceZ	call_dictZarrowZclistr�r6rXr�r8r7r9r:ZsubstatsZ
left_widthrrrr��
s*






















zStats.print_call_linec

Cs"td
d|j
d�
td|j
d�
dS)Nz-   ncalls  tottime  percall  cumtime  percallr|)r~rzfilename:lineno(function))
r)r"r)
rrrrr��
s

zStats.print_titlecCs�|j|
\}}}}}t
|�
}||kr4|d
t
|�
}t|jd�
d|jd�
tt|�
d|jd�
|dkrxtdd|jd�
ntt||�
d|jd�
tt|�
d|jd�
|dkr�td	d|jd�
ntt||�
d|jd�
tt|
�
|jd�
dS)
N�
/�	r|)r~rr
r})
rz        z        )rr&r"r�rr�r5)rr6r7r8r9r:r;�
crrrr��
s











zStats.print_liner�rr�
r�)r�rEr�rr�
r�)r�rEr�rFr�
r�)r�rGr�rFr�
r�)r�rG�rHr�
r�)r�rI�rHr�
r�)r�rI�rJr�
r�)r�rK�rHr�
r�)r�rI�rLr�
r�)r�rM�rLr�rHr�rJr�r�r�r�)r�rNr�r
r�
r�)r�rO�rPr�
r�)r�rQr�rRr�
r�)r�rSr�rRr�
r�)r�rS)
r�)�__name__�
__module__�__qualname__rr	rr r
rDrZr^rgrirmrnrwr{r�r�r�r�r�r�r�rrrrr sD!














 
c@seZ
dZd
d�Zdd�ZdS)rdcCs
|
|_dS)
N)
�comp_select_list)rr�rrrr�
s
zTupleComp.__init__cCsBx<|jD]2\}}|
|}||}||kr.|S||kr|SqWd
S)Nr
)
r�)r�left�right�index�	direction�
l�
rrrrre�
s







zTupleComp.compareN)r�r�r�rrerrrrrd�
srdc
Cs|\}
}}tj
j|
�
||fS)
N)r*�path�basename)�	func_namerCrVrXrrrrj�
s


rjc

Cs|d
S)NrRr)
r6rrrr�
s
rc
CsN|dd
�d	krB|d
}
|
jd�
r<|
j
d�
r<d|
dd
�S|
Snd|SdS)NrR�
~r
�
<�
>z{%s}rz	%s:%d(%s))r�r
r)�
startswith�endswith)r�rXrrrr5�
s


r5cCs@|
\}}}}}|\}}}	}
}||||||	||
t||�fS)
N)
�add_callers)�targetr�r7r8r9r:r;Zt_ccZt_ncZt_ttZt_ctZ	t_callersrrrr>�
s


r>cCs�i}x|j�D]\}}|||<qWxb|
j�D]V\}}||krzt
|t�rhtd
d�t|||�D�
�
||<q�|||7<q,|||<q,W|S)Nc
Ssg|]}
|
d|
d
�qS)r
rr)�.0�
irrr�
<listcomp>�
szadd_callers.<locals>.<listcomp>)r4r%rf�zip)r�r�Znew_callersr6rlrrrr��
s







r�c
Cs"d
}
x|j�D]}|
|7}
qW|
S)Nr
)
r�)r;r8rTrrr�count_callss


r�c

Csd
|S)Nz%8.3fr)
�
xrrrr�s
r��__main__c@s�eZ
dZd6dd�
Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd
S)7�ProfileBrowserNcCs6tj
j|�

d
|_d|_tj|_|
dk	r2|j|
�

dS)Nz% )	�cmd�Cmdr�promptrrrr�do_read)r�profilerrrrs





zProfileBrowser.__init__cCs�|j�}g}x�|D]�}y|j
t|�
�

wWntk
r>


YnXy8t|�
}|d
ksZ|dkrjtd|jd�
w|j
|�

wWntk
r�


YnX|j
|�

qW|jr�t|j|
�|�
ntd|jd�
dS)Nrr
z#Fraction argument must be in [0, 1])
rzNo statistics object is loaded.)	�splitrbr`�
ValueErrorrsr"rr�getattr)r�fnrVrZ	processedZtermZfracrrr�generic"s,






















zProfileBrowser.genericc

CsXtd
|j
d�
td|j
d�
td|j
d�
td|j
d�
td|j
d�
td|j
d�
dS)NzArguments may be:)
rz0* 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)
rrrr�generic_help:s





zProfileBrowser.generic_helpcCsd|jrRy|jj
|
�

Wq`tk
rN
}
ztd
|
|f|jd�
WYdd}~Xq`Xntd|jd�
dS)Nz$Failed to load statistics for %s: %s)
rzNo statistics object is loaded.r
)rr
�IOErrorr"r)rrV�
errr�do_addBs




*
zProfileBrowser.do_addc

Cstd
|j
d�
dS)Nz>Add profile info from given file to current statistics object.)
r)r"r)
rrrr�help_addKs
zProfileBrowser.help_addcCs|jd
|
�S)Nr�)
r�)rrVrrr�
do_calleesNs
zProfileBrowser.do_calleesc

Cstd
|j
d�
|j�
dS)Nz6Print callees statistics from the current stat object.)
r)r"rr�)
rrrr�help_calleesPs

zProfileBrowser.help_calleescCs|jd
|
�S)Nr�)
r�)rrVrrr�
do_callersTs
zProfileBrowser.do_callersc

Cstd
|j
d�
|j�
dS)Nz6Print callers statistics from the current stat object.)
r)r"rr�)
rrrr�help_callersVs

zProfileBrowser.help_callerscCstd
|j
d�
dS)Nr)
rr)r"r)rrVrrr�do_EOFZs

zProfileBrowser.do_EOFc

Cstd
|j
d�
dS)NzLeave the profile brower.)
r)r"r)
rrrr�help_EOF]s
zProfileBrowser.help_EOFc
Csd
S)Nrr)rrVrrr�do_quit`s
zProfileBrowser.do_quitc

Cstd
|j
d�
dS)NzLeave the profile brower.)
r)r"r)
rrrr�	help_quitbs
zProfileBrowser.help_quitcCs�|
r�yt|
�
|_
Wnjtk
rF
}
zt|jd
|jd�
dSd}~Xn8tk
r|
}
zt|jjd||jd�
dSd}~XnX|
d|_	n6t
|j	�
dkr�|j	dd�}
|j|
�

ntd|jd�
dS)	Nr)
r�
:z% rRz1No statistics object is current -- cannot reload.r
���)rr�OSErrorr"rrr!r0r�r�rr�)rrV�errrrrr�es













zProfileBrowser.do_readc

Cs td
|j
d�
td|j
d�
dS)Nz+Read in profile data from a specified file.)
rz*Without argument, reload the current file.)r"r)
rrrr�	help_readvs

zProfileBrowser.help_readcCs$|jr|jj
�
ntd
|jd�
dS)NzNo statistics object is loaded.)
rr
)rrir"r)rrVrrr�
do_reversezs


zProfileBrowser.do_reversec

Cstd
|j
d�
dS)Nz/Reverse the sort order of the profiling report.)
r)r"r)
rrrr�help_reverse�s
zProfileBrowser.help_reversecs�|jst
d
|jd�
dS|jj��|
rRt�f
dd�|
j�D�
�
rR|jj|
j��
n@t
d|jd�
x0tjj	�D]"\}}t
d||df|jd�
qlWdS)	NzNo statistics object is loaded.)
rc
3s|]}
|
�kV
qdS)
Nr)r�r�)
�abbrevsrr�	<genexpr>�sz)ProfileBrowser.do_sort.<locals>.<genexpr>z/Valid sort keys (unique prefixes are accepted):z%s -- %srr
)
rr"rr^�allr�rgrrZr4)rrVr_r�r)
r�r�do_sort�s









zProfileBrowser.do_sortc

Cs td
|j
d�
td|j
d�
dS)Nz.Sort profile data according to specified keys.)
rz3(Typing `sort' without arguments lists valid keys.))r"r)
rrrr�	help_sort�s

zProfileBrowser.help_sortcs�f
d
d�tj
D�
S)Nc
sg|]}
|
j��
r|
�qSr)
r�)r��
a)
�textrrr��sz0ProfileBrowser.complete_sort.<locals>.<listcomp>)rrZ)rr�rr)
r�r�
complete_sort�s
zProfileBrowser.complete_sortcCs|jd
|
�S)Nr�)
r�)rrVrrr�do_stats�s
zProfileBrowser.do_statsc

Cstd
|j
d�
|j�
dS)Nz.Print statistics from the current stat object.)
r)r"rr�)
rrrr�
help_stats�s

zProfileBrowser.help_statscCs$|jr|jj
�
ntd
|jd�
dS)NzNo statistics object is loaded.)
r)rrmr"r)rrVrrr�do_strip�s

zProfileBrowser.do_stripc

Cstd
|j
d�
dS)Nz<Strip leading path information from filenames in the report.)
r)r"r)
rrrr�
help_strip�s
zProfileBrowser.help_stripc

Cstd
|j
d�
dS)NzShow help for a given command.)
r)r"r)
rrrr�	help_help�s
zProfileBrowser.help_helpc
Cs|
r|
SdS)
Nr)r�stoprVrrr�postcmd�s


zProfileBrowser.postcmd)
N)r�r�r�rr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrr�s4

	r�rrRz*Welcome to the profile statistics browser.)
rzGoodbye.)!rr*r,r(ro�	functoolsr�__all__rrdrjrr5r>r�r�r�r�r��readline�ImportErrorr�r�r�argvZinitprofileZbrowserr�r�r"rZcmdloop�KeyboardInterruptrrrr�<module>sL