a �DOgr�@s�dZddgZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddlZdZGdd�d�Zd d �Zd d �ZGd d�d�Zdd�Zdd�Zddd�Zdd�ZGdd�d�Zdd�Zedkr�e�dS)a�program/module to trace Python program or function execution Sample use, command line: trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs trace.py -t --ignore-dir '$prefix' spam.py eggs trace.py --trackcalls spam.py eggs Sample use, programmatically import sys # create a Trace object, telling it what to ignore, and whether to # do tracing or line-counting or both. tracer = trace.Trace(ignoredirs=[sys.base_prefix, sys.base_exec_prefix,], trace=0, count=1) # run the new command using the given tracer tracer.run('main()') # make a report, placing output in /tmp r = tracer.results() r.write_results(show_missing=True, coverdir="/tmp") �Trace�CoverageResults�N)� monotonicz#pragma NO COVERc@seZdZddd�Zdd�ZdS)�_IgnoreNcCs:|s t�nt|�|_|sgn dd�|D�|_ddi|_dS)NcSsg|]}tj�|��qS�)�os�path�normpath��.0�drr�/usr/lib64/python3.9/trace.py� Hs�z$_Ignore.__init__..z�)�set�_mods�_dirs�_ignore)�self�modules�dirsrrr �__init__Fs �z_Ignore.__init__cCs�||jvr|j|S||jvr,d|j|<dS|jD]"}|�|d�r2d|j|<dSq2|durld|j|<dS|jD]$}|�|tj�rrd|j|<dSqrd|j|<dS)Nr�.r)rr� startswithrr�sep)r�filename� modulename�modr rrr �namesLs$          z _Ignore.names)NN)�__name__� __module__� __qualname__rrrrrr rEs rcCs tj�|�}tj�|�\}}|S)z-Return a plausible module name for the patch.)rr�basename�splitext)r�baser�extrrr �_modnamews r&cCs�tj�|�}d}tjD]@}tj�|�}|�|�r|t|�tjkrt|�t|�kr|}q|rr|t|�dd�}n|}tj�|�\}}|�tjd�}tj r�|�tj d�}tj� |�\}}|� d�S)z,Return a plausible module name for the path.�rNr) rr�normcase�sysr�lenr� splitdrive�replace�altsepr#�lstrip)rZ comparepathZlongest�dirr$�driverr%rrr � _fullmodname~s    r1c@s:eZdZddd�Zdd�Zdd�Zdd d �Zdd d �ZdS)rNc Cs||_|jduri|_|j��|_||_|jdur8i|_|j��|_||_|jdurZi|_|j��|_||_||_|j�rzTt|jd�� }t� |�\}}}Wd�n1s�0Y|� |� |||��WnDt t tf�y}z$td|j|ftjd�WYd}~n d}~00dS)N�rbzSkipping counts file %r: %s��file)�counts�copyZcounter� calledfuncs�callers�infile�outfile�open�pickle�load�update� __class__�OSError�EOFError� ValueError�printr)�stderr)rr5r7r9r8r:�f�errrrr r�s2      .��zCoverageResults.__init__cCs|�d�o|�d�S)z_Return True if the filename does not refer to a file we want to have reported. �<�>)r�endswith)rrrrr �is_ignored_filename�sz#CoverageResults.is_ignored_filenamec Csn|j}|j}|j}|j}|j}|j}|D]}|�|d�||||<q(|D] }d||<qJ|D] }d||<q\dS)z.Merge in the data from another CoverageResultsrrN)r5r7r8�get) r�otherr5r7r8Z other_countsZother_calledfuncsZ other_callers�keyrrr r>�s zCoverageResults.updateTFc" Cs|jr@t�td�|j}t|�D]\}}}td|||f�q"|jr�t�td�d}} t|j�D]h\\} } } \} }}| |kr�t�td| d�| }d} | | kr�| | kr�td| �| } td| | ||f�qfi}|jD].\}}|�|i�}||<|j||f||<q�i}|��D�],\}}|�|��r0�q|�d��rH|d d �}|d u�rpt j � t j � |��}t |�}n$|}t j �|��s�t �|�t|�}|�r�t|�}ni}t�|�}t j �||d �}t|d �� }t�|j�\}}Wd �n1�s�0Y|�|||||�\}}|�r|�rtd ||�}||||f||<�q|�r�|�r�td�t|�D]&}||\}}}}td||��qb|j�rzLt|jd��*} t�|j|j|jf| d�Wd �n1�s�0YWn8t�y}!ztd|!tj d�WYd }!~!n d }!~!00d S)af Write the coverage results. :param show_missing: Show lines that had no hits. :param summary: Include coverage summary per module. :param coverdir: If None, the results of each module are placed in its directory, otherwise it is included in the directory specified. zfunctions called:z*filename: %s, modulename: %s, funcname: %szcalling relationships:r'z***z -->z %s.%s -> %s.%sz.pycN�����z.coverr2�dzlines cov% module (path)z%5d %3d%% %s (%s)�wbrz"Can't save counts files because %sr3)!r7rC�sortedr8r5rK�itemsrJrIrr�dirname�abspathr&�exists�makedirsr1�_find_executable_linenos� linecache�getlines�joinr;�tokenize�detect_encoding�readline�write_results_file�intr:r<�dumpr@r)rD)"rZ show_missing�summary�coverdirZcallsrr�funcnameZlastfileZ lastcfileZpfileZpmodZpfunc�cfileZcmodZcfuncZper_file�lineno� lines_hitZsums�countr/�lnotab�sourceZ coverpath�fp�encoding�_�n_hits�n_lines�percent�mrErFrrr � write_results�s� ��           0�   �(zCoverageResults.write_resultsc Cszt|d|d�}Wn<tyN}z$td||ftjd�WYd}~dSd}~00d}d} |��t|d�D]r\} } | |vr�|�d || �| d7} |d7}n.| |vr�t| vr�|�d �|d7}n |�d �|�| �d ��qhWd�n1s�0Y| |fS) z'Return a coverage results file in path.�w�rkz3trace: Could not open %r for writing: %s - skippingr3N)rrrrz%5d: z>>>>>> z �) r;r@rCr)rD� enumerate�write�PRAGMA_NOCOVER� expandtabs) rr�linesrhrfrkr:rFrnrmre�linerrr r^)s.��    0z"CoverageResults.write_results_file)NNNNN)TFN)N)rr r!rrJr>rqr^rrrr r�s�  \cCs,i}t�|�D]\}}||vrd||<q|S)z:Return dict where keys are lines in the line number table.r)�disZfindlinestarts)�code�strs�linenosrlrerrr �_find_lines_from_codeIs  rcCs4t||�}|jD]}t�|�r|�t||��q|S)z� _find_lines)r|r}r~�crrr r�Ss    r�cCs�i}tj}t||d��t}t�|j�}|D]R\}}}} } |tjkrv|tjkrv|\} } | \} }t| | d�D] }d||<qh|}q(Wd�n1s�0Y|S)z�Return a dict of possible docstring positions. The dict maps line numbers to strings. There is an entry for line that contains only a string or a part of a triple-quoted string. rsrN)�token�INDENTr;r[�generate_tokensr]�STRING�range)rrkr Z prev_ttyperE�tokZttypeZtstr�start�endrzZslineZscolZelineZecol�irrr � _find_strings_s    $r�c Cs�ztyz}z&td||ftjd�iWYd}~Sd}~00t||d�}t ||�}t ||�S)zAReturn dict where keys are line numbers in the line number table.Nz%Not printing coverage data for %r: %sr3�exec) r[r;�readrkr@rCr)rD�compiler�r�)rrE�progrkrFr|r}rrr rWvs (��  rWc @speZdZddd�Zdd �Zdd d �Zd d �Zdd�Zdd�Zdd�Z dd�Z dd�Z dd�Z dd�Z dd�ZdS) rrrrNFc Cs�||_||_t||�|_i|_i|_d|_||_i|_i|_ i|_ d|_ | rTt �|_ |rb|j |_nZ|rp|j|_nL|r�|r�|j|_|j|_n2|r�|j|_|j|_n|r�|j|_|j|_nd|_dS)ax @param count true iff it should count number of times each line is executed @param trace true iff it should print out each line that is being counted @param countfuncs true iff it should just output a list of (filename, modulename, funcname,) for functions that were called at least once; This overrides `count' and `trace' @param ignoremods a list of the names of modules to ignore @param ignoredirs a list of the names of directories to ignore all of the (recursive) contents of @param infile file from which to read stored counts to be added into the results @param outfile file in which to write the results @param timing true iff timing information be displayed rNr)r9r:r�ignorer5Zpathtobasename� donothing�trace� _calledfuncs�_callers� _caller_cache� start_time�_time�globaltrace_trackcallers� globaltrace�globaltrace_countfuncs�globaltrace_lt�localtrace_trace_and_count� localtrace�localtrace_trace�localtrace_count) rrgr�� countfuncs� countcallers� ignoremods� ignoredirsr9r:�timingrrr r�s6      zTrace.__init__cCs ddl}|j}|�|||�dS)Nr)�__main__�__dict__�runctx)r�cmdr��dictrrr �run�sz Trace.runc Cs�|dur i}|duri}|js6t�|j�t�|j�z*t|||�W|js~t�d�t�d�n|js|t�d�t�d�0dS�N)r�� threading�settracer�r)r�)rr��globals�localsrrr r��s      � z Trace.runctxc OsPd}|jst�|j�z"||i|��}W|jsLt�d�n|jsJt�d�0|Sr�)r�r)r�r�)r�func�args�kw�resultrrr �runfunc�s  � z Trace.runfuncc Cs�|j}|j}|rt|�}nd}|j}d}||jvrL|j|dur�|j|}n�d|j|<dd�t�|�D�}t|�dkr�dd�t�|d�D�}t|�dkr�dd�t�|d�D�} t| �dkr�| dj}||j|<|dur�d||f}|||fS)NcSsg|]}t�|�r|�qSr)r�Z isfunction)r rErrr r�s �z1Trace.file_module_function_of..rcSsg|]}t|t�r|�qSr)� isinstancer�r rrr r�s �rcSsg|]}t|d�r|�qS)� __bases__)�hasattr)r r�rrr r�s �z%s.%s) �f_code� co_filenamer&�co_namer��gcZ get_referrersr*r) r�framer|rrrcZclsnameZfuncsZdicts�classesrrr �file_module_function_of�s,          zTrace.file_module_function_ofcCs0|dkr,|�|�}|�|j�}d|j||f<dS)zkHandler for call events. Adds information about who called who to the self._callers dict. �callrN)r��f_backr�)rr��why�arg� this_funcZ parent_funcrrr r��s  zTrace.globaltrace_trackcallerscCs |dkr|�|�}d|j|<dS)zoHandler for call events. Adds (filename, modulename, funcname) to the self._calledfuncs dict. r�rN)r�r�)rr�r�r�r�rrr r�s zTrace.globaltrace_countfuncscCsj|dkrf|j}|j�dd�}|rbt|�}|durf|j�||�}|sf|jrZtd||jf�|j SndSdS)z�Handler for call events. If the code block being entered is to be ignored, returns `None', else returns self.localtrace. r��__file__Nz! --- modulename: %s, funcname: %s) r�� f_globalsrKr&r�rr�rCr�r�)rr�r�r�r|rrZ ignore_itrrr r�s�zTrace.globaltrace_ltcCs�|dkr~|jj}|j}||f}|j�|d�d|j|<|jrTtdt�|jdd�tj � |�}td||t � ||�fdd�|j S) Nrzrr�%.2f� �r�� %s(%d): %sr')r�r��f_linenor5rKr�rCr�rrr"rX�getliner�)rr�r�r�rrerM�bnamerrr r�(s  ��z Trace.localtrace_trace_and_countcCsd|dkr^|jj}|j}|jr4tdt�|jdd�tj�|�}td||t � ||�fdd�|j S)Nrzr�r�r�r�r') r�r�r�r�rCr�rrr"rXr�r�)rr�r�r�rrer�rrr r�7s  ��zTrace.localtrace_tracecCs<|dkr6|jj}|j}||f}|j�|d�d|j|<|jS)Nrzrr)r�r�r�r5rKr�)rr�r�r�rrerMrrr r�Ds zTrace.localtrace_countcCst|j|j|j|j|jd�S)N)r9r:r7r8)rr5r9r:r�r�)rrrr �resultsLs  �z Trace.results) rrrrrrNNF)NN)rr r!rr�r�r�r�r�r�r�r�r�r�r�rrrr r�s� 2  )   c s�ddl}|��}|jdddd�|�dd�}|jdd d d d �|jd dd dd �|jddd dd �|jddd dd �|�d�}|��}|jddd dd �|jddd dd �|jdddd �|jd!d"d#d �|jd$d%d d&d �|jd'd(d d)d �|jd*d+d d,d �|�d-d.�}|jd/d0gd1d2�|jd3d0gd4d2�|jd5d d6d7d2�|jd8d9d:d;�|jd<|jd=d;�|��}|j�r�t� d>��t� d?����fd@dA��dBdC�|j D�|_ �fdDdC�|jD�|_|j �r�|j �s�|� dE�t|j |j dF�}|�|j|j|j�St|j|j|j|jg��s |� dG�|j�rB|j�s8|j�rB|� dH�|j�r\|j�s\|� dI�|jdu�rr|� dJ�t|j|j|j|j|j |j|j |j |jdK� }z�|j�r�ddl}|j}|�|�\} } } | jg|j�t _!dL| j| j"| j#| ddM�} nt|jg|j�t _!t$j%�&|j�t j%d<t'�(|j��"} t)| �*�|jdN�} Wd�n1�sJ0Y|jdLdddO�} |�+| | | �WnPt,�y�}z$t �-dPt j!d|f�WYd}~nd}~0t.�y�Yn0|�/�}|j0�s�|�|j|j|j�dS)QNrz --version�versionz trace 2.0)�actionr�z Main optionsz(One of these (or --report) must be givenz-cz--count� store_truez�Count the number of times each line is executed and write the counts to .cover for each module executed, in the module's directory. See also --coverdir, --file, --no-report below.)r��helpz-tz--tracez3Print each line to sys.stdout before it is executedz-lz --listfuncsz�Keep track of which functions are executed at least once and write the results to sys.stdout after the program exits. Cannot be specified alongside --trace or --count.z-Tz --trackcallsz^Keep track of caller/called pairs and write the results to sys.stdout after the program exits.Z Modifiersz-rz--reportz�Generate a report from a counts file; does not execute any code. --file must specify the results file to read, which must have been created in a previous run with --count --file=FILEz-Rz --no-reportz^Do not generate the coverage report files. Useful if you want to accumulate over several runs.z-fz--filez+File to accumulate counts over several runs)r�z-Cz --coverdirz�Directory where the report files go. The coverage report for . will be written to file //.coverz-mz --missingz?Annotate executable lines that were not executed with ">>>>>> "z-sz --summaryz\Write a brief summary for each file to sys.stdout. Can only be used with --count or --reportz-gz--timingzQPrefix each line with the time since the program started. Only used while tracingZFilterszCan be specified multiple timesz--ignore-module�appendzqIgnore the given module(s) and its submodules (if it is a package). Accepts comma separated list of module names.)r��defaultr�z --ignore-dirzWIgnore files in the given directory (multiple directories can be joined by os.pathsep).z--moduleFzTrace a module. �progname�?zfile to run as main program)�nargsr�� argumentszarguments to the programZstdlibZ platstdlibcs4tj�tj�|��}|�d���d��}tj�|�S)Nz$prefixz $exec_prefix)rr� expanduser� expandvarsr,r )�s)� _exec_prefix�_prefixrr �parse_ignore_dir�szmain..parse_ignore_dircSs$g|]}|�d�D] }|���qqS)�,)�split�strip)r r�rrrr r�s�zmain..cs&g|]}|�tj�D] }�|��qqSr)r�r�pathsep)r r�r�)r�rr r�s�z-r/--report requires -f/--file)r9r:zLmust specify one of --trace, --count, --report, --listfuncs, or --trackcallsz8cannot specify both --listfuncs and (--trace or --count)z3--summary can only be used with --count or --reportz3progname is missing: required with the main options)r�r�r�r�r9r:r�r�)rr�� __package__� __loader__�__spec__� __cached__r�)r�rr�r�zCannot run file %r because: %s)1�argparse�ArgumentParser� add_argument�add_argument_group�add_mutually_exclusive_group� REMAINDER� parse_argsZ ignore_dir� sysconfigZget_pathZ ignore_module�reportr4�errorrrq�missingrarb�anyr�rgZ listfuncsZ trackcallsr�rr��module�runpy�_get_module_detailsr�r�r)�argv�parent�loaderrrrS�io� open_coder�r�r�r@�exit� SystemExitr�Z no_report)r��parser�grpZ_grpZoptsr��tr�� module_name�mod_name�mod_specr|ZglobsrjrFr)r�r�r�r �mainRs�� � � � �  � ��� � � �� � � �� �  � �      �� 2�,rr�)N)�__doc__�__all__r�rXrr)r�r�r[r�r�r{r<�timerr�r�rwrr&r1rrr�r�rWrrrrrrr �s< 20  O