Current 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-sortZstore� profile_sortZNOSE_PROFILE_SORTZ
cumulativeZSORTz"Set sort order for profiler output)�action�dest�default�metavar�helpz--profile-stats-file�profile_stats_file�FILEZNOSE_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 �optionsZ
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 Zenabledr � 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_25Ztmpr 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 �tempfileZmkstempr! r )r r r r r � s zProfile._create_pfile)�__name__�
__module__� __qualname__�__doc__r r r r � classmethodr r r) r0 r5 r r r r r r s r )r: r r � ImportErrorZloggingr2 r. r6 Znose.plugins.baser Z nose.utilr Z getLoggerr' r r r r r �<module> s
Mini Shell