3
\=p @ s, d Z ddgZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ y ddlZW n" ek
r ejZdd ZY nX dd Zd
d Zd ZG d d
d
Zdd Zdd ZG dd dZdd Zdd Zd ddZ dd Z G dd dZ d d Z e d kr(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")
TraceCoverageResults N) monotonicc C s t jd d S )N)syssettrace r r /usr/lib64/python3.6/trace.py _unsettraceE s r c C s t j| tj| d S )N) threadingr r )funcr r r _settraceH s
r c C s t jd tjd d S )N)r r r
r r r r r L s
z#pragma NO COVERc @ s e Zd ZdddZdd ZdS )_IgnoreNc C s: |s
t nt || _|s g n dd |D | _ddi| _d S )Nc S s g | ]}t jj|qS r )ospathnormpath).0dr r r
U s z$_Ignore.__init__..z )set_mods_dirs_ignore)selfmodulesdirsr r r __init__S s
z_Ignore.__init__c C s || j kr| j | S || jkr,d| j |< dS x*| jD ] }|j|d r4d| j |< dS q4W |d krnd| j |< dS x,| jD ]"}|j|tj rvd| j |< dS qvW d| j |< dS )Nr .r )r r
startswithr r sep)r filename
modulenamemodr r r r namesY s$
z
_Ignore.names)NN)__name__
__module__ __qualname__r r# r r r r r
R s
r
c C s t jj| }t jj|\}}|S )z-Return a plausible module name for the patch.)r r basenamesplitext)r baser extr r r _modname s r+ c C s t jj| }d}xJtjD ]@}t jj|}|j|r|t| t jkrt|t|kr|}qW |rv| t|d d }n| }t jj|\}}|jt jd}t j r|jt j d}t jj
|\}}|j dS )z,Return a plausible module name for the path. r Nr ) r r normcaser r lenr
splitdrivereplacealtsepr( lstrip)r Z comparepathZlongestdirr) Zdriver r* r r r _fullmodname s r4 c @ s: e Zd ZdddZdd Zdd Zdd
d Zdd d
ZdS )r Nc C s || _ | j d kri | _ | j j | _|| _| jd kr8i | _| jj | _|| _| jd krZi | _| jj | _|| _|| _| jry@t| jd}tj |\}}}W d Q R X | j
| j ||| W n@ t t
tfk
r } z td| j|f tjd W Y d d }~X nX d S )NrbzSkipping counts file %r: %s)file)countscopyZcounter calledfuncscallersinfileoutfileopenpickleloadupdate __class__OSErrorEOFError
ValueErrorprintr stderr)r r7 r9 r; r: r< ferrr r r r s,
zCoverageResults.__init__c C s |j do|jdS )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 sz | j }| j}| j}|j }|j}|j}x$|D ] }|j|d|| ||< q*W x|D ] }d||< qPW x|D ] }d||< qfW dS )z.Merge in the data from another CoverageResultsr r N)r7 r9 r: get) r otherr7 r9 r: Z other_countsZother_calledfuncsZ
other_callerskeyr r r r@ s
zCoverageResults.updateTFc ! C s | j rDt td | j }x(t|D ] \}}}td|||f q$W | jrt td d }} xvt| jD ]h\\}
} } \}
}}|
|krt td|
d |
}d} |
|
kr| |
krtd|
|
} td| | ||f qlW i }x8| jD ].\}}|j|i }||< | j||f ||< qW i }x$|j D ]\}}| j|r@q&|jdrX|d d }|d krt j
j t j
j |}t
|}n$|}t j
j|st j| t|}|rt|}ni }tj|}t j
j||d }t|d }tj|j\}}W d Q R X | j|||||\} } |r&| r&td
| | } | | ||f||< q&W |r|rtd x2t|D ]&} || \} } }}td|| q`W | jry&t j | j| j | jft| jdd
W n6 t k
r } ztd| t j d W Y d d } ~ X nX 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.pycNr z.coverr5 d z lines cov% module (path)z%5d %3d%% %s (%s)wbz"Can't save counts files because %s)r6 )!r9 rE sortedr: r7 rM itemsrL rK r r dirnameabspathr+ existsmakedirsr4 _find_executable_linenos linecachegetlinesjoinr= tokenizedetect_encodingreadlinewrite_results_fileintr< r> dumprB r rF )!r Z show_missingsummarycoverdirZcallsr r! funcnameZlastfileZ lastcfileZpfileZpmodZpfunccfileZcmodZcfuncZper_filelineno lines_hitZsumscountr3 lnotabsourceZ coverpathfpencoding_n_hitsn_linesZpercentmrH r r r
write_results s|
z CoverageResults.write_resultsc C s yt |d|d}W n6 tk
rH } ztd||f tjd d S d}~X nX d}d} | xt|dD ]r\}
} |
|kr|jd||
| d7 } |d7 }n.|
|krt| kr|jd |d7 }n
|jd
|j| jd qdW W dQ R X | |fS )
z'Return a coverage results file in path.w)rm z3trace: Could not open %r for writing: %s - skipping)r6 r Nr z%5d: z>>>>>> z )r r ) r= rB rE r rF enumeratewritePRAGMA_NOCOVER
expandtabs) r r linesrj rh rm r< rH rp ro rg liner r r r` 5 s(
z"CoverageResults.write_results_file)NNNNN)TFN)N)r$ r% r& r rL r@ rr r` r r r r r s
[c C s0 i }x&t j| D ]\}}||krd||< qW |S )z:Return dict where keys are lines in the line number table.r )disZfindlinestarts)codestrslinenosrn rg r r r _find_lines_from_codeU s
r c C s8 t | |}x(| jD ] }tj|r|jt|| qW |S )z.r c S s g | ]}t |tr|qS r )
isinstancer )r r r r r r s r c S s g | ]}t |d r|qS ) __bases__)hasattr)r r r r r r s z%s.%s) f_code co_filenamer+ co_namer gcZ
get_referrersr. r$ )
r framer| r r! re ZclsnameZfuncsZdictsclassesr r r file_module_function_of s,
z Trace.file_module_function_ofc C s0 |dkr,| j |}| j |j}d| j||f< dS )zkHandler for call events.
Adds information about who called who to the self._callers dict.
callr N)r f_backr )r r whyarg this_funcZ parent_funcr r r r s
z Trace.globaltrace_trackcallersc C s |dkr | j |}d| j|< 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 sj |dkrf|j }|jjdd}|rbt|}|dk rf| jj||}|sf| jrZtd||jf | j S ndS dS )zHandler 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_globalsrM r+ r r# r rE 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}| jj|dd | j|< | jrTtdt | j dd tj j
|}td||t j ||f dd | j
S ) Nrz r r z%.2f )r z
%s(%d): %sr, )r r f_linenor7 rM r rE r r r r' rZ getliner )r r r r r rg rO bnamer r r r 2 s z Trace.localtrace_trace_and_countc C sd |dkr^|j j}|j}| jr4tdt | j dd tjj|}td||t j
||f dd | j S )Nrz z%.2fr )r z
%s(%d): %sr, ) r r r r rE r r r r' rZ r r )r r r r r rg r r r r r A s zTrace.localtrace_tracec C s< |dkr6|j j}|j}||f}| jj|dd | j|< | jS )Nrz r r )r r r r7 rM r )r r r r r rg rO r r r r N s zTrace.localtrace_countc C s t | j| j| j| j| jdS )N)r; r< r9 r: )r r7 r; r< r r )r r r r resultsV s
z
Trace.results)NN)r$ r% r& r r r r r r r r r r r r r r r r r s 0
)
c s t j } | jdddd | jdd}|jddd d
d |jd d
d dd |jddd dd |jddd dd | jd}|j }|jddd dd |jddd dd |jd d d d |jd d!d"d |jd#d$d d%d |jd&d'd d(d |jd)d*d d+d | jd,d-}|jd.d/g d0d1 |jd2d/g d3d1 | jd4d5d6d7 | jd8t jd9d7 | j }|jrd:d;jt j
f}t j j
t jf| t j j
t jf| fdd? |jD |_fd@d?|jD |_|jr|js| jdA t|j|jdB}|j|j|j|jS t|j|j|j |j gs*| jdC |j rL|jsB|jrL| jdD |jrh|j rh| jdE |j d kr~| jdF |j f|j t _ t j j!|j t j dG< t"|j|j|j |j |j|j|j|j|j#dH }yJt$|j }t%|j& |j dI}W d Q R X |j dJd d dK} |j'|| | W nP t(k
rT }
z t j)dLt j dG |
f W Y d d }
~
X n t*k
rh Y nX |j+ }|j,s|j|j|j|j d S )MNz --versionversionz trace 2.0)actionr z Main optionsz(One of these (or --report) must be givenz-cz--count
store_truezCount 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 --listfuncszKeep 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--reportzGenerate 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
--coverdirzDirectory 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 tracingZFiltersz Can be specified multiple timesz--ignore-moduleappendzqIgnore 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 programlibzpython{0.major}.{0.minor}c s4 t jjt jj| } | jdjd } t jj| S )Nz$prefixz $exec_prefix)r r
expanduser
expandvarsr0 r )s) _exec_prefix_prefixr r parse_ignore_dir s z main..parse_ignore_dirc S s$ g | ] }|j d D ] }|j qqS ),)splitstrip)r r r" r r r r s zmain..c s& g | ] }|j tjD ] } |qqS 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 --reportz3filename is missing: required with the main optionsr )r r r r r; r< r r r )r r$ __package__
__cached__z Cannot run file %r because: %s)-argparseArgumentParser add_argumentZadd_argument_groupZ add_mutually_exclusive_groupZ REMAINDER
parse_argsZ
ignore_dirformatr version_infor r r\ base_prefixbase_exec_prefixZ
ignore_moduleZreportr6 errorr rr Zmissingrc rd anyr ri Z listfuncsZ
trackcallsr r argvrU r r r= r r r rB exit
SystemExitr Z no_report) parserZgrpZ_grpZoptsZrel_pathr trl r| ZglobsrH r )r r r r main\ s
(r r )N)!__doc____all__r rZ r rer r r] r r r{ r> Ztimer r r
ImportErrorr r r rw r
r+ r4 r r r r rY r r r$ r r r r 1 sH 2 /
M