← Back
Editing: prof.cpython-312.pyc
� ���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
Save File
Cancel