B

��{a~o�@s�dZd
dgZ
ddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlm
Z
ddlZdZGdd�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@seZ
dZddd�
Zdd�Zd
S)�_IgnoreNcCs:|
s
t�nt|
�
|_
|sgnd
d�|D�
|_ddi
|_dS)Nc
Ssg|]}
tj
�|
�
�qS�)�os�path�normpath)�.0�
drr�/usr/lib/python3.7/trace.py�
<listcomp>Gs
z$_Ignore.__init__.<locals>.<listcomp>z<string>�
)�set�_mods�_dirs�_ignore)�self�modules�dirsrrr�__init__Es




z_Ignore.__init__cCs�||jkr|j|S||j
kr,d
|j|<d
Sx*|j
D] }|�|d�
r4d
|j|<d
Sq4W|
dkrnd
|j|<d
Sx,|jD]"}|
�|tj�
rvd
|j|<d
SqvWd|j|<dS)Nr�
.r)rr�
startswithrr�sep)r�filename�
modulename�modrrrr�namesKs$

















z
_Ignore.names)NN)�__name__�
__module__�__qualname__rrrrrrrDs

rc
Cs tj
�|�
}
tj
�|
�
\}}|S)
z-Return a plausible module name for the patch.)rr�basename�splitext)r�baser�extrrr�_modnamevs

r%c
Cs�tj
�|�
}
d
}xJtj
D]@}tj
�|�
}|
�|�
r|
t|�
tjkrt|�
t|�
kr|}qW|rv|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#Zdriverr$rrr�_fullmodname}s 











r/c@s:eZ
dZddd�
Zdd�Zdd�Zdd
d�
Zddd
�
Zd
S)rNc
Cs�|
|_|jdkri|_|j�
�|_||_|jdkr8i|_|j�
�|_||_|jdkrZi|_|j�
�|_||_||_|jr�y@t|jd
��}t�	|�
\}
}}WdQRX|�
|�|
||��

Wn@tt
tfk
r�
}
ztd|j|ftjd�
Wdd}~XYnXdS)N�rbzSkipping counts file %r: %s)
�file)�counts�copyZcounter�calledfuncs�callers�infile�outfile�open�pickle�load�update�	__class__�OSError�EOFError�
ValueError�printr(�stderr)rr2r4r6r5r7�
f�errrrrr�s,






















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	Csz|j}|j
}|j}|
j}|
j
}|
j}x$|D]}|�|d
�||||<q*Wx|D]}d||<qPWx|D]}d||<qfWdS)z.Merge in the data from another CoverageResultsrrN)r2r4r5�get)	r�otherr2r4r5Zother_countsZother_calledfuncsZ
other_callers�keyrrrr;�s










zCoverageResults.updateTFc!
Cs�|jrDt
�
t
d
�

|j}x(t|�
D]\}}}t
d|||f�

q$W|jr�t
�
t
d�

d}}	xvt|j�
D]h\\}
}}\}
}}|
|kr�t
�
t
d|
d�
|
}d}	|
|
kr�|	|
kr�t
d|
�
|
}	t
d||||f�

qlWi}x8|jD].\}}|�|i�}||<|j||f||<q�Wi}�
x$|��D�
]\}}|�|�
�
r@�
q&|�d�
�
rX|d	d
�}|d	k�
r�t	j
�t	j
�|�
�
}t
|�
}n$|}t	j
�|�
�
s�t	�|�

t|�
}|
�
r�t|�
}ni}t�|�
}t	j
�||d�}t|d��}t�|j�
\}}Wd	QRX|�|||||�\}}|�
r&|�
r&td
||�
}||||f||<�
q&W|�r�|�r�t
d�

x2t|�
D]&}||\}}}}t
d||�

�q`W|j�r�y&t�|j|j|jft|jd�d�
Wn6tk
�r�
} 
zt
d| tj d�
Wd	d	} ~ XYnXd	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.coverr0�dzlines   cov%   module   (path)z%5d   %3d%%   %s   (%s)�wbrz"Can't save counts files because %s)
r1)!r4r@�sortedr5r2rH�itemsrGrFrr�dirname�abspathr%�exists�makedirsr/�_find_executable_linenos�	linecache�getlines�joinr8�tokenize�detect_encoding�readline�write_results_file�intr7r9�dumpr=r(rA)!rZshow_missing�summary�coverdirZcallsrr�funcnameZlastfileZ	lastcfileZpfileZpmodZpfunc�cfileZcmodZcfuncZper_file�lineno�	lines_hitZsums�countr.�lnotab�sourceZ	coverpath�fp�encoding�
_�n_hits�n_linesZpercent�
mrCrrr�
write_results�s|























































zCoverageResults.write_resultsc
Cs�yt|
d
|d�}Wn8t
k
rJ
}
ztd|
|ftjd�
dSd}~XYnXd}d}	|��
x�t|d�D]r\}
}|
|kr�|�d	||
�

|	d7}	|d7}n.|
|kr�t|kr�|�d
�

|d7}n
|�d�

|�|�d�
�

qfWWdQRX|	|fS)
z'Return a coverage results file in path.�
w)
rhz3trace: Could not open %r for writing: %s - skipping)
r1)rrNrrz%5d: z>>>>>> z       �)	r8r=r@r(rA�	enumerate�write�PRAGMA_NOCOVER�
expandtabs)rr�linesrercrhr7rCrkrjrb�linerrrr['
s(

















z"CoverageResults.write_results_file)NNNNN)TFN)
N)rrr rrGr;rmr[rrrrr�s



[cCs0i}x&t�
|�
D]\}}||
krd
||<qW|S)z:Return dict where keys are lines in the line number table.r)�disZfindlinestarts)�code�strs�linenosrirbrrr�_find_lines_from_codeG
s


rzcCs8t||
�}x(|j
D]}t�|�
r|�t||
��

qW|S)
z<Return lineno dict for all code objects reachable from code.)rz�	co_consts�inspectZiscoder;�_find_lines)rwrxry�
crrrr}Q
s




r}c	Cs�i}tj
}t||
d
��r}t�|j�
}x^|D]V\}}}}	}
|tjkr||tj
kr||\}}|	\}
}xt||
d�D]}d||<qlW|}q*WWdQRX|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.
    )
rhrN)�token�INDENTr8rX�generate_tokensrZ�STRING�range)rrhrZ
prev_ttyperB�tokZttypeZtstr�start�endruZslineZscolZelineZecol�
irrr�
_find_strings]
s












r�c

Cs�y(t�
|�
�}
|
��}|
j}Wd
QRXWn8tk
r`
}
ztd||ftjd�
iSd
}~XYnXt||d�}t	||�}t
||�S)zAReturn dict where keys are line numbers in the line number table.Nz%Not printing coverage data for %r: %s)
r1�exec)rXr8�readrhr=r@r(rA�compiler�r})rrB�progrhrCrwrxrrrrTt
s










rTc	@speZ
dZddd�
Zdd	�Zdd
d�
Zdd
�Zdd�Zdd�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�ZdS) r
rrrNFc

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)r6r7r�ignorer2Zpathtobasename�	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)
rrdr��
countfuncs�countcallers�
ignoremods�
ignoredirsr6r7�timingrrrr�
s6





























zTrace.__init__cCs d
dl}|j
}|�|
||�
dS)Nr)�__main__�__dict__�runctx)r�cmdr��dictrrr�run�
s


z	Trace.runc	Csh|dkri}|dkri}|js6t
�|j�

t�|j�

zt|
||�
Wd|jsbt�d�

t
�d�

XdS)
N)r��	threading�settracer�r(r�)rr��globals�localsrrrr��
s









zTrace.runctxc	Os<d}|jst
�|j�

z|
||�
}Wd|js6t
�d�

X|S)
N)r�r(r�r�)r�func�args�kw�resultrrr�runfunc�
s






z
Trace.runfuncc
Cs�|
j}|j
}|rt|�
}nd}|j}d}||jkrL|j|dk	r�|j|}n�d|j|<d
d�t�|�
D�
}t|�
dkr�dd�t�|d�
D�
}t|�
dkr�dd�t�|d�
D�
}	t|	�
dkr�|	dj}||j|<|dk	r�d||f}|||fS)Nc
Ssg|]}
t�
|
�
r|
�qSr)r|Z
isfunction)r
rBrrrr
�
s
z1Trace.file_module_function_of.<locals>.<listcomp>rc
Ssg|]}
t|
t
�r|
�qSr)�
isinstancer�)r
rrrrr
�
s
rc
Ssg|]}
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�framerwrrr`ZclsnameZfuncsZdicts�classesrrr�file_module_function_of�
s,
















zTrace.file_module_function_ofcCs0|d
kr,|�|
�
}|�|
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 |d
kr|�|
�
}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|d
krf|
j}|
j
�dd�}|rbt|�
}|dk	rf|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_globalsrHr%r�rr�r@r�r�)rr�r�r�rwrrZ	ignore_itrrrr�s









zTrace.globaltrace_ltcCs�|d
kr~|
jj
}|
j}||f}|j�|d�d|j|<|jrTtdt�|jdd�
tj	�
|�
}td||t�||�fdd�
|j
S)	Nrurrz%.2f�
 )
r�z
%s(%d): %sr&)r�r��f_linenor2rHr�r@r�rrr!rU�getliner�)rr�r�r�rrbrJ�bnamerrrr�&s








z Trace.localtrace_trace_and_countcCsd|d
kr^|
jj
}|
j}|jr4tdt�|jdd�
tj�|�
}td||t	�
||�fdd�
|jS)Nruz%.2fr�)
r�z
%s(%d): %sr&)r�r�r�r�r@r�rrr!rUr�r�)rr�r�r�rrbr�rrrr�5s






zTrace.localtrace_tracecCs<|d
kr6|
jj
}|
j}||f}|j�|d�d|j|<|jS)Nrurr)r�r�r�r2rHr�)rr�r�r�rrbrJrrrr�Bs





zTrace.localtrace_countc

Cst|j
|j|j|j|jd
�S)N)r6r7r4r5)rr2r6r7r�r�)
rrrr�resultsJs




z
Trace.results)	rrrrrrNNF)NN)rrr rr�r�r�r�r�r�r�r�r�r�r�rrrrr
�
s



0

)	
cs�d
dl}|�
�}
|
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�
|
jd5d6d7d8�
|
jd9|jd:d8�
|
��}|j�
r�d;d<�t	j
�
f}tjj
t	jf
|���
tjj
t	jf
|�����
fd=d>��d?d@�|jD�
|_�f
dAd@�|jD�
|_|j�r|j�
s�|
�dB�

t|j|jdC�}|�|j|j|j�St|j|j|j|jg�
�s2|
�dD�

|j�rT|j�sJ|j�rT|
�dE�

|j�rn|j�sn|
�dF�

|jdk�r�|
�dG�

|jf
|j�t	_ tj�!|j�
t	jd
<t"|j|j|j|j|j|j|j|j|j#dH�	}yJt$|j�
�}t%|�&�|jdI�}	WdQRX|jdJdddK�}
|�'|	|
|
�
WnPt(k
�rZ
}
zt	�)dLt	j d
|f�

Wdd}~XYnt*k
�rn


YnX|�+�}|j,�s�|�|j|j|j�
dS)MNrz	--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 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).r�
?zfile to run as main program)�nargsr��	argumentszarguments to the programZlibzpython{0.major}.{0.minor}c

s4tj
�tj
�|�
�
}|�d
�
��d��}tj
�|�
S)Nz$prefixz$exec_prefix)rr�
expanduser�
expandvarsr+r	)
�
s)�_exec_prefix�_prefixrr�parse_ignore_dir�s


zmain.<locals>.parse_ignore_dirc
Ss$g|]}
|
�d�
D]}|�
��qqS)
�
,)�split�strip)r
r�rrrrr
�s
zmain.<locals>.<listcomp>c
s&g|]}
|
�t
j�
D]}�|�
�qqSr)r�r�pathsep)r
r�r�)
r�rrr
�s
z-r/--report requires -f/--file)r6r7zLmust 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 --reportz3filename is missing: required with the main options)r�r�r�r�r6r7r�r�r�)r�r�__package__�
__cached__zCannot run file %r because: %s)-�argparse�ArgumentParser�add_argumentZadd_argument_groupZadd_mutually_exclusive_groupZ	REMAINDER�
parse_argsZ
ignore_dir�formatr(�version_inforrrW�base_prefix�base_exec_prefixZ
ignore_moduleZreportr1�errorrrmZmissingr^r_�anyr�rdZ	listfuncsZ
trackcallsrr��argvrPr
r�r8r�r�r�r=�exit�
SystemExitr�Z	no_report)r��parserZgrpZ_grpZoptsZrel_pathr��
trgrwZglobsrCr)r�r�r�r�mainPs�




































































(


r�r�)
N)�__doc__�__all__rUr�rer(rrXr|r�rvr9Ztimerr�r�rrrr%r/rrzr}r�rTr
r�rrrrr�<module>1s:










2/

O