�9�c@s�dZy d
dl
Z
d
dl
mZ
Wnek
rE


d	\Z
Zn
Xd
dlZd
dlZd
dlZd
dlZd
dl	m
Z

d
dlmZ
ej
d�Zde
f
d��YZdS(
s/This plugin will run tests using the hotshot profiler, which is part
of the standard library. To turn it on, use the ``--with-profile`` option
or set the NOSE_WITH_PROFILE environment variable. Profiler output can be
controlled with the ``--profile-sort`` and ``--profile-restrict`` options,
and the profiler output file may be changed with ``--profile-stats-file``.

See the `hotshot documentation`_ in the standard library documentation for
more details on the various output options.

.. _hotshot documentation: http://docs.python.org/library/hotshot.html
i�N(
tstats(
tPlugin(
ttolistsnose.pluginstProfilecBsneZ
dZd	ZeZd
�Zd�Ze	e�Zd�Z
d�Zd�Zd�Z
d�Zd�ZRS(
sC
    Use this plugin to run tests using the hotshot profiler. 
    cCs�|j�sd
St
j||
|�
|
jdddddd|jdd	�d
ddd
�
|
jdddddd
dd|jd�dd�
|
jdddddd
dd|jd�dd�
d
S(s&Register commandline options.
        Ns--profile-sorttactiontstoretdesttprofile_sorttdefaulttNOSE_PROFILE_SORTt
cumulativetmetavartSORTthelps"Set sort order for profiler outputs--profile-stats-filetprofile_stats_filetFILEtNOSE_PROFILE_STATS_FILEs;Profiler stats file; default is a new temp file on each runs--profile-restricttappendtprofile_restricttRESTRICTtNOSE_PROFILE_RESTRICTs?Restrict profiler output. See help for pstats.Stats for details(t	availableR
toptionst
add_optiontget(tselftparsertenv((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pyR!s"














c

Cs
tdk	S(
N(thotshottNone(
tcls((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pyR8s
c

Cs3|j�sd
S|j
�
tj|j�|_d
S(s5Create profile stats file and load profiler.
        N(Rt
_create_pfileRRtpfiletprof(
R((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pytbegin<s



cCs�|j�st
|_d
Stj||
|�
||_|
jrV|
j|_t
|_nd
|_t
|_d
|_|
j|_
t|
j�|_d
S(sConfigure plugin.
        N(RtFalsetenabledR
t	configuretconfRR tclean_stats_fileRtTruetfilenoRtsortRRtrestrict(RRR&((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pyR%Ds
	


	
	

	
	
	

cs=�j�sd
St
jd|
�
�j|
�f
d�}|S(s6Wrap entire test run in :func:`prof.runcall`.
        Nspreparing test %scs�j�
|
j
||�
dS(
N(Rtruncall(tresultR!ttest(
R(s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pytrun_and_profile\s


(RtlogtdebugR!(RR.R/((
Rs5/sys/lib/python2.7/site-packages/nose/plugins/prof.pytprepareTestVs



cCs�tj
d
�
|jj�
tj|j�}|j|j�
t	|d�}|rf|j
}|
|_
ntj}|
t_z=|j
r�tj
d|j
�
|j|j
�
n
|j�
Wd|r�||_
n	|t_XdS(s Output profiler report.
        sprinting profiler reporttstreams"setting profiler restriction to %sN(R0R1R!tcloseRtloadR t
sort_statsR*thasattrR3tsyststdoutR+tprint_stats(RR3t
prof_statst	compat_25ttmp((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pytreportas$






	
	
	

	


cCs�|j�sd
Sy|j
j�
Wntk
r4


n
X|jr�|jrrytj|j�
Wqrtk
rn


qrXnytj|j	�
Wq�tk
r�


q�Xnd
S(s5Clean up stats file, if configured to do so.
        N(RR!R4tAttributeErrorR'R)tostOSErrortunlinkR R(RR-((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pytfinalizes"





	
	










c

Cs1|js-t
j�\|_|_t|_ndS(
N(R ttempfiletmkstempR)R(R'(
R((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pyR�s
	

N(t__name__t
__module__t__doc__RR R#R'RRtclassmethodR"R%R2R>RCR(((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pyRs


							(NN(RHRRtImportErrorRtloggingR@R8RDtnose.plugins.baseR
t	nose.utilRt	getLoggerR0R(((s5/sys/lib/python2.7/site-packages/nose/plugins/prof.pyt<module>s