XylotrechusZ
�
���g� � � � d Z ddlZddlmZ ddlZddlZddlZddlZddlm Z ddl
m Z ej d� Z
G d� d e � Zy# e$ r d\ ZZY �Fw xY w)
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 �X � e Zd ZdZdZdZd� Zd� Z ee� Zd� Z d� Z
d� Z d � Z d
� Z
d � Zy) �ProfilezC
Use this plugin to run tests using the hotshot profiler.
NFc �: � | j � syt j | ||� |j ddd|j dd� dd� � |j d
dd d |j d
� d�� |j dddd|j d� d�� y)z&Register commandline options.
Nz--profile-sort�store� profile_sort�NOSE_PROFILE_SORT�
cumulative�SORTz"Set sort order for profiler output)�action�dest�default�metavar�helpz--profile-stats-file�profile_stats_file�FILE�NOSE_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_restrict�RESTRICT�NOSE_PROFILE_RESTRICTz?Restrict profiler output. See help for pstats.Stats for details)� availabler �options�
add_option�get)�self�parser�envs �@/opt/hc_python/lib/python3.12/site-packages/nose/plugins/prof.pyr zProfile.options! s� � � �~�~� � ����t�V�S�)����*�7� �"%�'�'�*=�|�"L�"(� C� � E� ���0�� 3�"(�"%�'�'�*C�"D� 2� � 3� ���.�x� 1�",�"%�'�'�*A�"B� 5� � 6� c � � t d uS �N)�hotshot)�clss r! r zProfile.available8 s
� ��d�"�"r" c � � | j � sy| j � t j | j � | _ y)z5Create profile stats file and load profiler.
N)r �
_create_pfiler% r �pfile�prof�r s r! �beginz
Profile.begin< s1 � � �~�~� � � ��� ��O�O�D�J�J�/�� r" c �: � | j � sd| _ yt j | ||� || _ |j
r|j
| _ d| _ nd| _ d| _ d| _ |j | _
t |j � | _
y)zConfigure plugin.
FNT)r �enabledr � configure�confr r) �clean_stats_file�filenor
�sortr r �restrict)r r r0 s r! r/ zProfile.configureD s� � � �~�~� � �D�L� �����w��-��� � � %� %� �3�3�D�J�$)�D� !� �D�J�$(�D� !��� ��(�(�� � �w� 7� 7�8��
r" c �~ � � � j � syt j d|z � � j |f� fd� }|S )z6Wrap entire test run in :func:`prof.runcall`.
Nzpreparing test %sc �J �� �j � |j || � y r$ )r( �runcall)�resultr* �testr s �r! �run_and_profilez,Profile.prepareTest.<locals>.run_and_profile\ s �� � � � � � �L�L��v� &r" )r �log�debugr* )r r9 r: s ` r! � prepareTestzProfile.prepareTestV s: �� � �~�~� � � � � �%��,�-�)-���� '� � r" c �\ � t j d� | j j � t j
| j � }|j | j � t |d� }|r|j }||_
nt j }|t _ | j r:t j d| j � |j | j � n|j � |r||_
y|t _ y# |r||_
w |t _ w xY w)z Output profiler report.
zprinting profiler report�streamz"setting profiler restriction to %sN)r; r< r* �closer �loadr) �
sort_statsr3 �hasattrr? �sys�stdoutr4 � print_stats)r r? �
prof_stats� compat_25�tmps r! �reportzProfile.reporta s� � � � � �,�-� � � �����Z�Z��
�
�+�
�� � �d�i�i�(�
�J��1� � � �#�#�C� &�J� ��*�*�C� �C�J�
!��}�}�� � �>��
�
�N�&�
�&�&��
�
�6��&�&�(��$'�
�!� ��
�� �$'�
�!� ��
�s
�(AD �D+c �h � | j � sy | j j � | j rM| j
r t
j | j
� t
j | j � yy# t $ r Y �ew xY w# t $ r Y �<w xY w# t $ r Y yw xY w)z5Clean up stats file, if configured to do so.
N)
r r* r@ �AttributeErrorr1 r2 �os�OSError�unlinkr) )r r8 s r! �finalizezProfile.finalize s� � � �~�~� � � � �I�I�O�O� � � � ��{�{���H�H�T�[�[�)�
�� � �$�*�*�%� �� � �
� �� � ���� �
���
�s5 �B � B �& B% � B�B� B"�!B"�% B1�0B1c �n � | j s)t j � \ | _ | _ d| _ y y )NT)r) �tempfile�mkstempr2 r1 r+ s r! r( zProfile._create_pfile� s- � ��z�z�&.�&6�&6�&8� #�D�K���$(�D� !� r" )�__name__�
__module__� __qualname__�__doc__r) r1 r r � classmethodr, r/ r= rJ rP r( � r" r! r r sF � ��
�E� ��6�.#��I�&�I�0�9�$ � !�<�0)r" r )rW r% r � ImportError�loggingrM rD rR �nose.plugins.baser � nose.utilr � getLoggerr; r rY r" r! �<module>r_ sc ��
� �� � � �
� � $� ��g����'��)�f� )�� � � �N�G�U� �s �
A �
A�A