Edit File: 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