HEX

File: //lib64/python3.9/__pycache__/trace.cpython-39.opt-1.pyc
a

�DOgr�@s�dZddgZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddl
mZddlZdZGdd�d�Zd	d
�Zdd�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|�|_|sgndd�|D�|_ddi|_dS)NcSsg|]}tj�|��qS�)�os�path�normpath��.0�drr�/usr/lib64/python3.9/trace.py�
<listcomp>Hs�z$_Ignore.__init__.<locals>.<listcomp>z<string>�)�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�modrrrr
�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
�_modnamewsr&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)rZcomparepathZlongest�dirr$�driverr%rrr
�_fullmodname~s 
r1c@s:eZdZddd�Zdd�Zdd�Zdd
d�Zddd
�ZdS)rNc
Cs||_|jduri|_|j��|_||_|jdur8i|_|j��|_||_|jdurZi|_|j��|_||_||_|j�rzTt|jd�� }t�	|�\}}}Wd�n1s�0Y|�
|�|||��WnDtt
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�otherr5r7r8Zother_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)"rZshow_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<Return lineno dict for all code objects reachable from code.)r�	co_consts�inspectZiscoder>�_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)rrkrZ
prev_ttyperE�tokZttypeZtstr�start�endrzZslineZscolZelineZecol�irrr
�
_find_strings_s


$r�c
Cs�z<t�|��}|��}|j}Wd�n1s00YWn>tyz}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�Zdd
�Zdd�Zdd�Zdd�Z	dd�Z
dd�Zdd�Zdd�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�|duri}|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
�
zTrace.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)rrErrr
r�s
�z1Trace.file_module_function_of.<locals>.<listcomp>rcSsg|]}t|t�r|�qSr)�
isinstancer�r
rrr
r�s
�rcSsg|]}t|d�r|�qS)�	__bases__)�hasattr)rr�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_funcZparent_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�|jS)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�DszTrace.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

)	
cs�ddl}|��}|jdddd�|�dd�}|jdd	d
dd�|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|jdF�}|�|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�zMain 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 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.<locals>.parse_ignore_dircSs$g|]}|�d�D]}|���qqS)�,)�split�strip)rr�rrrr
r�s�zmain.<locals>.<listcomp>cs&g|]}|�tj�D]}�|��qqSr)r�r�pathsep)rr�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
�<module>s<20

O