관리-도구
편집 파일: prof.cpython-38.pyc
U )�e� � @ s� d Z zddlZddlmZ 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_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 �options� add_option�get)�self�parser�env� r �?/opt/hc_python/lib/python3.8/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 r )r Zruncall)�resultr �testr 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 5 |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 z| j�� W n tk r. Y nX | jr�| jrbzt�| j� W n tk r` Y nX zt�| j � W n tk r� Y nX dS )z5Clean up stats file, if configured to do so. N) r r r0 �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. r6 r; r r r r r r s r )rA r r � ImportError�loggingr8 r3 r< Znose.plugins.baser Z nose.utilr � getLoggerr, r r r r r �<module> s