o
6��f-r � @ s� d Z ddgZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlZdZG dd� d�Zd d
� Zd d � ZG d
d� d�Zdd� Zdd� Zddd�Zdd� ZG dd� d�Zdd� Ze dkrye� dS 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 @ s e Zd Zddd�Zdd� ZdS )�_IgnoreNc C s: |st � nt |�| _|sg ndd� |D �| _ddi| _d S )Nc S s g | ]}t j�|��qS � )�os�path�normpath��.0�dr r �,/opt/alt/python310/lib64/python3.10/trace.py�
<listcomp>H s �z$_Ignore.__init__.<locals>.<listcomp>z<string>� )�set�_mods�_dirs�_ignore)�self�modules�dirsr r r
�__init__F s
�z_Ignore.__init__c C s� || j v r
| j | S || jv rd| j |< dS | jD ]}|�|d �r*d| j |< dS q|d u r6d| j |< dS | jD ]}|�|tj �rKd| j |< dS q9d| j |< dS )Nr �.r )r r �
startswithr r �sep)r �filename�
modulename�modr r r r
�namesL s(
�
�
z
_Ignore.names�NN)�__name__�
__module__� __qualname__r r r r r r
r E s
r c C s t j�| �}t j�|�\}}|S )�,Return a plausible module name for the path.)r r �basename�splitext)r �baser �extr r r
�_modnamew s r( c C s� t j�| �}d}tjD ] }t j�|�}|�|�r+|t|� t jkr+t|�t|�kr+|}q |r9| t|�d d� }n| }t j�|�\}}|�t jd�}t j rT|�t j d�}t j�
|�\}}|� d�S )r# � r Nr ) r r �normcase�sysr �lenr �
splitdrive�replace�altsepr% �lstrip)r Z comparepathZlongest�dirr& Zdriver r' r r r
� _fullmodname~ s"
�
r2 c @ s> e Zd Z ddd�Zdd� Zdd� Zdd
d �Zdd d
�ZdS )r Nc
C s || _ | j d u r i | _ | j �� | _|| _| jd u r i | _| j�� | _|| _| jd u r-i | _| j�� | _|| _|| _| jr�z,t| jd��}t� |�\}}}W d � n1 sVw Y | �
| j |||d�� W d S t t
tfy� } ztd| j|f tjd� W Y d }~d S d }~ww d S )N�rb)�callerszSkipping counts file %r: %s��file)�counts�copyZcounter� calledfuncsr4 �infile�outfile�open�pickle�load�update� __class__�OSError�EOFError�
ValueError�printr+ �stderr)r r7 r9 r: r4 r; �f�errr r r
r � s:
� �����zCoverageResults.__init__c C s |� d�o |�d�S )z_Return True if the filename does not refer to a file
we want to have reported.
�<�>)r �endswith)r r r r r
�is_ignored_filename� s z#CoverageResults.is_ignored_filenamec C sn | j }| j}| j}|j }|j}|j}|D ]}|�|d�|| ||< q|D ]}d||< q%|D ]}d||< q.dS )z.Merge in the data from another CoverageResultsr r N)r7 r9 r4 �get) r �otherr7 r9 r4 Z other_countsZother_calledfuncsZ
other_callers�keyr r r
r? � s
�zCoverageResults.updateTFc "
C s | j r t� td� | j }t|�D ]\}}}td|||f � q| jrht� td� d }} t| j�D ]4\\}
} } \}
}}|
|krNt� td|
d� |
}d} |
|
kr]| |
kr]td|
� |
} td| | ||f � q3i }| jD ]\}}|�|i � }||< | j||f ||< qmi }|�� D ]�\}}| �|�r�q�|�d�r�|d d
� }|d u r�t j
� t j
� |��}t
|�}n|}t j
�|�s�t �|� t|�}|r�t|�}ni }t�|�}t j
�||d �}t|d ��}t�|j�\}}W d � n1 s�w Y | �|||||�\} } |�r| �rtd
| | �} | | ||f||< q�|�r<|�r<td� t|�D ]} || \} } }}td|| � �q(| j�r�z)t| jd��} t � | j| j | jf| d� W d � W d S 1 �sbw Y W d S t �y� }! ztd|! t j d� W Y d }!~!d S d }!~!ww d 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.coverr3 �d z lines cov% module (path)z%5d %3d%% %s (%s)�wbr z"Can't save counts files because %sr5 )!r9 rD �sortedr4 r7 rL �itemsrK rJ r r �dirname�abspathr( �exists�makedirsr2 �_find_executable_linenos� linecache�getlines�joinr< �tokenize�detect_encoding�readline�write_results_file�intr; r= �dumprA r+ rE )"r Z show_missing�summary�coverdirZcallsr r �funcnameZlastfileZ lastcfileZpfileZpmodZpfunc�cfileZcmodZcfuncZper_file�lineno� lines_hitZsums�countr1 �lnotab�sourceZ coverpath�fp�encoding�_�n_hits�n_linesZpercent�mrF rG r r r
�
write_results� s�
��
�� � �(� ���z CoverageResults.write_resultsc
C s z t |d|d�}W n ty' } ztd||f tjd� W Y d}~dS d}~ww d}d} |�J t|d�D ]9\}
} |
|v rN|�d ||
� | d7 } |d7 }n|
|v r`t| vr`|�d
� |d7 }n|�d � |�| �d �� q4W d � | |fS 1 s{w Y | |fS )
z'Return a coverage results file in path.�w�rl z3trace: Could not open %r for writing: %s - skippingr5 N)r r r r z%5d: z>>>>>> z � ) r< rA rD r+ rE � enumerate�write�PRAGMA_NOCOVER�
expandtabs) r r �linesri rg rl r; rG ro rn rf �liner r r
r_ ) s: ����
�
��z"CoverageResults.write_results_file)NNNNN)TFN�N)r r! r" r rK r? rq r_ r r r r
r � s
�
\c C s, i }t �| �D ] \}}||vrd||< q|S )z:Return dict where keys are lines in the line number table.r )�disZfindlinestarts)�code�strs�linenosrm rf r r r
�_find_lines_from_codeI s �r� c C s4 t | |�}| jD ]}t�|�r|�t||�� q|S )z<Return lineno dict for all code objects reachable from code.)r� � co_consts�inspectZiscoder? � _find_lines)r} r~ r �cr r r
r� S s
�r� c C s� i }t j}t| |d��;}t�|j�}|D ])\}}}} }
|t jkr;|t jkr;|\} } | \}
}t| |
d �D ]}d||< q4|}qW d � |S 1 sIw Y |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.
rs r N)�token�INDENTr< r\ �generate_tokensr^ �STRING�range)r rl r Z
prev_ttyperF �tokZttypeZtstr�start�endrz ZslineZscolZelineZecol�ir r r
�
_find_strings_ s"
�
�
�
r� c
C s� z t �| ��}|�� }|j}W d � n1 sw Y W n ty= } ztd| |f tjd� i W Y d}~S d}~ww t|| d�}t | |�}t
||�S )zAReturn dict where keys are line numbers in the line number table.Nz%Not printing coverage data for %r: %sr5 �exec) r\ r< �readrl rA rD r+ rE �compiler� r� )r rF �progrl rG r} r~ r r r
rX v s$ ������
rX c @ sv e Zd Z d dd�Zdd � Zd d
d �Zd d
� Zdd� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
d d � ZdS ) r r r r NFc
C s� || _ || _t||�| _i | _i | _d| _|| _i | _i | _ i | _
d| _ | r*t � | _ |r2| j
| _dS |r:| j| _dS |rH|rH| j| _| j| _dS |rT| j| _| j| _dS |r`| j| _| j| _dS d| _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
r Nr )r: r; r �ignorer7 Zpathtobasename� 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)
r rh r� �
countfuncs� countcallers�
ignoremods�
ignoredirsr: r; �timingr r r
r � s6
zTrace.__init__c C s dd l }|j}| �|||� d S )Nr )�__main__�__dict__�runctx)r �cmdr� �dictr r r
�run� s z Trace.runc C s� |d u ri }|d u r i }| j st�| j� t�| j� zt|||� W | j s2t�d � t�d � d S d S | j sBt�d � t�d � w w r{ )r� � threading�settracer� r+ r� )r r� �globals�localsr r r
r� � s
�
�z Trace.runctxc O sP d }| j s t�| j� z||i |��}W | j s t�d � |S | j s't�d � w w r{ )r� r+ r� r� )r �func�args�kw�resultr r r
�runfunc� s
� �z
Trace.runfuncc
C s� |j }|j}|r
t|�}nd }|j}d }|| jv r&| j| d ur%| j| }nCd | j|< dd� t�|�D �}t|�dkridd� t�|d �D �}t|�dkridd� t�|d �D �} t| �dkri| d j}|| j|< |d ursd||f }|||fS )Nc S s g | ] }t �|�r|�qS r )r� Z
isfunction)r rF r r r
r � �
�z1Trace.file_module_function_of.<locals>.<listcomp>r c S s g | ] }t |t�r|�qS r )�
isinstancer� r
r r r
r � r� r c S s g | ] }t |d �r|�qS )� __bases__)�hasattr)r r� r r r
r � r� z%s.%s) �f_code� co_filenamer( �co_namer� �gcZ
get_referrersr, r )
r �framer} r r rd ZclsnameZfuncsZdictsZclassesr r r
�file_module_function_of� s.
�
z Trace.file_module_function_ofc C s4 |dkr| � |�}| � |j�}d| j||f< dS dS )zkHandler for call events.
Adds information about who called who to the self._callers dict.
�callr N)r� �f_backr� )r r� �why�arg� this_funcZ parent_funcr r r
r� � s
�z Trace.globaltrace_trackcallersc C s$ |dkr| � |�}d| j|< dS dS )zoHandler for call events.
Adds (filename, modulename, funcname) to the self._calledfuncs dict.
r� r N)r� r� )r r� r� r� r� r r r
r� s
�z Trace.globaltrace_countfuncsc C sp |dkr6|j }|j�dd�}|r4t|�}|dur0| j�||�}|s2| jr-td||jf � | j S dS dS dS dS )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_globalsrL r( r� r r� rD r� r� )r r� r� r� r} r r Z ignore_itr r r
r� s" ���zTrace.globaltrace_ltc C s� |dkr?|j j}|j}||f}| j�|d�d | j|< | jr*tdt� | j dd� tj �
|�}td||t � ||�f dd� | j
S ) Nrz r r �%.2f� �r� �
%s(%d): %sr) )r� r� �f_linenor7 rL r� rD r� r r r$ rY �getliner� )r r� r� r� r rf rN �bnamer r r
r� ( s
��z Trace.localtrace_trace_and_countc C sd |dkr/|j j}|j}| jrtdt� | j dd� tj�|�}td||t �
||�f dd� | j S )Nrz r� r� r� r� r) ) r� r� r� r� rD r� r r r$ rY r� r� )r r� r� r� r rf r� r r r
r� 7 s
��zTrace.localtrace_tracec C s< |dkr|j j}|j}||f}| j�|d�d | j|< | jS )Nrz r r )r� r� r� r7 rL r� )r r� r� r� r rf rN r r r
r� D s zTrace.localtrace_countc C s t | j| j| j| j| jd�S )N)r: r; r9 r4 )r r7 r: r; r� r� )r r r r
�resultsL s
�z
Trace.results) r r r r r r NNFr )r r! r" r r� r� r� r� r� r� r� r� r� r� r� r r r r
r � s
�2
)
c s� dd l } | �� }|jdddd� |�dd�}|jdd d
d d � |jd
dd
dd � |jddd
dd � |jddd
dd � |�d�}|�� }|jddd
dd � |jddd
d d � |jd d d d � |jd!d"d#d � |jd$d%d
d&d � |jd'd(d
d)d � |jd*d+d
d,d � |�d-d.�}|jd/d0g d1d2� |jd3d0g d4d2� |jd5d
d6d7d2� |jd8d9d:d;� |jd<| jd=d;� |�� }|jr�t� d>��t� d?�� � �fd@dA��dBdC� |j
D �|_
�fdDdC�|jD �|_|j r�|j s�|�
dE� t|j |j dF�}|�|j|j|j�S t|j|j|j|jg��s
|�
dG� |j�r |j�s|j�r |�
dH� |j�r+|j�s+|�
dI� |jd u �r6|�
dJ� t|j|j|j|j|j
|j|j |j |jdK� }zk|j�rudd l }|j}|� |�\} }
} | j g|j �t _!dL| j |
j"|
j#|
d dM�} n:|jg|j �t _!t$j%�&|j�t j%d< t'�(|j��}
t)|
�*� |jdN�} W d � n 1 �s�w Y |jdLd d dO�} |�+| | | � W n( t,�y� } zt �-dPt j!d |f � W Y d }~nd }~w t.�y� Y nw |�/� }|j0�s�|�|j|j|j� d S d S )QNr z --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 <module>.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 <package>.<module> will be written to file <dir>/<package>/<module>.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 tracingZFiltersz Can 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
platstdlibc s4 t j�t j�| ��} | �d���d� �} t j�| �S )Nz$prefixz $exec_prefix)r r �
expanduser�
expandvarsr. r )�s)� _exec_prefix�_prefixr r
�parse_ignore_dir� s z main.<locals>.parse_ignore_dirc S s$ g | ]}|� d �D ]}|�� �q qS )�,)�split�strip)r r� r r r r
r � s
��zmain.<locals>.<listcomp>c s& g | ]}|� tj�D ]}� |��q
qS r )r� r �pathsep)r r� r� )r� r r
r � s
��z -r/--report requires -f/--file)r: r; 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� r: r; r� r� )r r� � __package__�
__loader__�__spec__�
__cached__r� )r� r r� r� z Cannot run file %r because: %s)1�argparse�ArgumentParser� add_argumentZadd_argument_groupZ add_mutually_exclusive_groupZ REMAINDER�
parse_argsZ
ignore_dir� sysconfigZget_pathZ
ignore_moduleZreportr6 �errorr rq Zmissingrb rc �anyr� rh Z listfuncsZ
trackcallsr� r r� �module�runpyZ_get_module_detailsr� r� r+ �argv�parent�loaderr r rT �io� open_coder� r� r� rA �exit�
SystemExitr� Z no_report)r� �parserZgrpZ_grpZoptsr� �tr� Z module_nameZmod_nameZmod_specr} Zglobsrk rG r )r� r� r� r
�mainR s� �
�
�
�
�
�
���
�
�
��
�
�
��
�
�
�
�� ��$���r� r� r{ ) �__doc__�__all__r� rY r r+ r� r� r\ r� r� r| r= �timer r� r� rw r r( r2 r r� r� r� rX r r� r r r r r
�<module> s@ 2 0
O
�