File: //opt/alt/python37/lib/python3.7/site-packages/nose/plugins/__pycache__/prof.cpython-37.pyc
B
�fP� � @ s� d Z yddlZddlmZ W n ek
r8 d\ZZY nX ddlZddlZddlZddlZddlm Z ddl
m Z e� d�Z
G dd � d e �ZdS )
a/ 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
� N)�stats)NN)�Plugin)�tolistz nose.pluginsc @ s` e Zd ZdZdZdZdd� Zdd� Zee�Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� ZdS )�ProfilezC
Use this plugin to run tests using the hotshot profiler.
NFc C st | � � s dS t�| ||� |jddd|�dd�ddd � |jd
dd d |�d
�dd� |jdddd|�d�dd� dS )z&Register commandline options.
Nz--profile-sort�store� profile_sortZNOSE_PROFILE_SORTZ
cumulativeZSORTz"Set sort order for profiler output)�action�dest�default�metavar�helpz--profile-stats-file�profile_stats_fileZFILEZNOSE_PROFILE_STATS_FILEz;Profiler stats file; default is a new temp file on each run)r r r r
r z--profile-restrict�append�profile_restrictZRESTRICTZNOSE_PROFILE_RESTRICTz?Restrict profiler output. See help for pstats.Stats for details)� availabler �options�
add_option�get)�self�parser�env� r �B/opt/alt/python37/lib/python3.7/site-packages/nose/plugins/prof.pyr ! s"
zProfile.optionsc C s t d k S )N)�hotshot)�clsr r r r 8 s zProfile.availablec C s& | � � s dS | �� t�| j�| _dS )z5Create profile stats file and load profiler.
N)r �
_create_pfiler r �pfile�prof)r r r r �begin< s z
Profile.beginc C sf | � � sd| _dS t�| ||� || _|jr<|j| _d| _n d| _d| _d| _|j | _
t |j �| _
dS )zConfigure plugin.
FNT)r �enabledr � configure�confr
r �clean_stats_file�filenor �sortr r �restrict)r r r! r r r r D s zProfile.configurec s2 � � � s dS t�d| � � j|f� fdd� }|S )z6Wrap entire test run in :func:`prof.runcall`.
Nzpreparing test %sc s � � � |�|| � d S )N)r Zruncall)�resultr �test)r r r �run_and_profile\ s z,Profile.prepareTest.<locals>.run_and_profile)r �log�debugr )r r' r( r )r r � prepareTestV s
zProfile.prepareTestc C s� t �d� | j�� t�| j�}|�| j� t |d�}|rH|j
}||_
n t j }|t _ z.| j
rxt �d| j
� |j| j
� n|�� W d|r�||_
n|t _ X dS )z Output profiler report.
zprinting profiler report�streamz"setting profiler restriction to %sN)r) r* r �closer �loadr Z
sort_statsr$ �hasattrr, �sys�stdoutr% Z print_stats)r r, Z
prof_statsZ compat_25�tmpr r r �reporta s$
zProfile.reportc C s� | � � s dS y| j�� W n tk
r. Y nX | jr�| jrbyt�| j� W n tk
r` Y nX yt�| j � W n tk
r� Y nX dS )z5Clean up stats file, if configured to do so.
N)
r r r- �AttributeErrorr" r# �os�OSError�unlinkr )r r& r r r �finalize s" zProfile.finalizec C s | j s t�� \| _| _ d| _d S )NT)r �tempfile�mkstempr# r" )r r r r r � s zProfile._create_pfile)�__name__�
__module__� __qualname__�__doc__r r" r r � classmethodr r r+ r3 r8 r r r r r r s r )r> r r � ImportError�loggingr5 r0 r9 Znose.plugins.baser Z nose.utilr � getLoggerr) r r r r r �<module> s