Your IP : 216.73.216.213
�fP� � @ s� d Z y d d l Z d d l m Z Wn e k
rF d \ Z Z Yn Xd d l Z d d l Z d d l Z d d l Z d d l m Z d d l
m Z e j d � Z
Gd d � d e � Z d S)
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)�Plugin)�tolistz nose.pluginsc @ s� e Z d Z d Z d Z d Z d d � Z d d � Z e e � Z d d � Z d
d � Z
d d
� Z d d � Z d d � Z
d d � Z d S)�ProfilezC
Use this plugin to run tests using the hotshot profiler.
NFc C s� | j � s d St j | | | � | j d d d d d d | j d d � d
d d d
�| j d d d d d d
d d | j d � d d �| j d d d d d d
d d | j d � d d �d S)z&Register commandline options.
Nz--profile-sort�action�store�dest� profile_sort�defaultZNOSE_PROFILE_SORTZ
cumulative�metavarZSORT�helpz"Set sort order for profiler outputz--profile-stats-file�profile_stats_fileZFILEZNOSE_PROFILE_STATS_FILEz;Profiler stats file; default is a new temp file on each runz--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/python35/lib/python3.5/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 s3 | j � s d S| j � t j | j � | _ d S)z5Create profile stats file and load profiler.
N)r �
_create_pfiler r �pfile�prof)r r r r �begin< s
z
Profile.beginc C s� | j � s d | _ d St j | | | � | | _ | j rV | j | _ d | _ n d | _ d | _ d | _ | j | _
t | j � | _
d S)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 s@ � j � s d St j d | � � j | � f d d � } | S)z6Wrap entire test run in :func:`prof.runcall`.
Nzpreparing test %sc s � j � | j | | � 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 j d � | j j � t j | j � } | j | j � t | d � } | rf | j
} | | _
n t j } | t _ z= | j
r� t j d | j
� | j | j
� n
| j � Wd | r� | | _
n | t _ Xd S)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� | j � s d Sy | j j � Wn t k
r5 Yn X| j r� | j rq y t j | j � Wn t k
rp Yn Xy t j | j � Wn t k
r� Yn Xd S)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 j � \ | _ | _ 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 )NN)r> r r � ImportError�loggingr5 r0 r9 Znose.plugins.baser Z nose.utilr � getLoggerr) r r r r r �<module> s